2120: 数颜色
Time Limit: 6 Sec Memory Limit: 259 MBSubmit: 1394 Solved: 516
[Submit][Status]
Description
墨墨购买了一套N支彩色画笔(其中有些颜色可能相同),摆成一排,你需要回答墨墨的提问。墨墨会像你发布如下指令: 1、 Q L R代表询问你从第L支画笔到第R支画笔中共有几种不同颜色的画笔。 2、 R P Col 把第P支画笔替换为颜色Col。为了满足墨墨的要求,你知道你需要干什么了吗?
Input
第1行两个整数N,M,分别代表初始画笔的数量以及墨墨会做的事情的个数。第2行N个整数,分别代表初始画笔排中第i支画笔的颜色。第3行到第2+M行,每行分别代表墨墨会做的一件事情,格式见题干部分。
Output
对于每一个Query的询问,你需要在对应的行中给出一个数字,代表第L支画笔到第R支画笔中共有几种不同颜色的画笔。
Sample Input
6 5
1 2 3 4 5 5
Q 1 4
Q 2 6
R 1 2
Q 1 4
Q 2 6
1 2 3 4 5 5
Q 1 4
Q 2 6
R 1 2
Q 1 4
Q 2 6
Sample Output
4
4
3
4
4
3
4
HINT
对于100%的数据,N≤10000,M≤10000,修改操作不多于1000次,所有的输入数据中出现的所有整数均大于等于1且不超过10^6。
Source
题解:
同维护队列
代码:
1 #include<cstdio> 2 #include<cstdlib> 3 #include<cmath> 4 #include<cstring> 5 #include<algorithm> 6 #include<iostream> 7 #include<vector> 8 #include<map> 9 #include<set> 10 #include<queue> 11 #include<string> 12 #define inf 1000000000 13 #define maxn 10000+1000 14 #define maxm 1000000+1000 15 #define eps 1e-10 16 #define ll long long 17 #define pa pair<int,int> 18 using namespace std; 19 inline int read() 20 { 21 int x=0,f=1;char ch=getchar(); 22 while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} 23 while(ch>='0'&&ch<='9'){x=10*x+ch-'0';ch=getchar();} 24 return x*f; 25 } 26 int n,m,block,b[maxn],c[maxn],pre[maxn],pos[maxn],last[maxm]; 27 void reset(int x) 28 { 29 int l=(x-1)*block+1,r=min(x*block,n); 30 for(int i=l;i<=r;i++)pre[i]=b[i]; 31 sort(pre+l,pre+r+1); 32 } 33 int find(int x,int y) 34 { 35 int l=(x-1)*block+1,r=x*block,mid; 36 while(l<=r) 37 { 38 mid=(l+r)>>1; 39 if(pre[mid]>=y)r=mid-1;else l=mid+1; 40 } 41 return l-(x-1)*block-1; 42 } 43 int query(int x,int y) 44 { 45 int sum=0,bx=pos[x],by=pos[y]; 46 if(by-bx<=1) 47 { 48 for(int i=x;i<=y;i++)if(b[i]<x)sum++; 49 } 50 else 51 { 52 for(int i=x;i<=bx*block;i++)if(b[i]<x)sum++; 53 for(int i=(by-1)*block+1;i<=y;i++)if(b[i]<x)sum++; 54 } 55 for(int i=bx+1;i<by;i++)sum+=find(i,x); 56 return sum; 57 } 58 void change(int x,int y) 59 { 60 for(int i=1;i<=n;i++)last[c[i]]=0; 61 c[x]=y; 62 for(int i=1;i<=n;i++) 63 { 64 int t=b[i]; 65 b[i]=last[c[i]]; 66 last[c[i]]=i; 67 if(t!=b[i])reset(pos[i]); 68 } 69 } 70 int main() 71 { 72 freopen("input.txt","r",stdin); 73 freopen("output.txt","w",stdout); 74 n=read();m=read(); 75 block=floor(sqrt(n)); 76 for(int i=1;i<=n;i++) 77 { 78 c[i]=read(); 79 pos[i]=(i-1)/block+1; 80 b[i]=last[c[i]]; 81 last[c[i]]=i; 82 } 83 for(int i=1;i<=pos[n];i++)reset(i); 84 char ch;int x,y; 85 while(m--) 86 { 87 ch=' '; 88 while(ch!='Q'&&ch!='R')ch=getchar();x=read();y=read(); 89 if(ch=='R')change(x,y);else printf("%d ",query(x,y)); 90 } 91 return 0; 92 }