2120: 数颜色
Time Limit: 6 Sec Memory Limit: 259 MB
Submit: 3327 Solved: 1287
[Submit][Status][Discuss]
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
Sample Output
4
4
3
4
HINT
对于100%的数据,N≤10000,M≤10000,修改操作不多于1000次,所有的输入数据中出现的所有整数均大于等于1且不超过10^6。
2016.3.2新加数据两组by Nano_Ape
【分析】
树套数啥的真不会啊…还是分块大法好。
这道题的分块是需要一定技巧的…用b[i]表示第i个数之前与它相距最近的相同的数的位置,比如 1,2,3,2 中,b[4]=2。
那么对于 l~r 的询问中,暴力的方法就是扫一遍,如果b[i] < l ,那么ans++,但这显然会超时,于是分块优化。
我们对于每一个块维护一个pre数组,pre数组中存放的是该块中b排序后的结果,具体在块中查询时,二分找出块中第一个使pre[i] < l的i。
修改操作会影响到b[i]的值,那么我们暴力重新跑一遍,如果发现b[i]发生了变动,那么将i所在块重新进行pre排序。
【代码】
//【bzoj2120】数颜色
#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define M(a) memset(a,0,sizeof a)
#define fo(i,j,k) for(i=j;i<=k;i++)
using namespace std;
int n,m,q;
int c[10005],be[10005],b[10005],lef[105],rig[105];
int last[1000001],pre[10005];
char ch[3];
inline int read()
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
inline void reset(int x)
{
int i,j;
fo(i,lef[x],rig[x]) pre[i]=b[i];
sort(pre+lef[x],pre+rig[x]+1);
}
inline void change(int x,int v)
{
int i,j;
fo(i,1,n) last[c[i]]=0;
c[x]=v;
fo(i,1,n)
{
int t=b[i];
b[i]=last[c[i]];
if(t!=b[i]) reset(be[i]);
last[c[i]]=i;
}
}
inline int ask(int l,int r)
{
int ans=0,i,j;
if(be[l]==be[r]) fo(i,l,r) {if(b[i]<l) ans++;}
else
{
fo(i,l,rig[be[l]]) if(b[i]<l) ans++;
fo(i,lef[be[r]],r) if(b[i]<l) ans++;
}
fo(i,be[l]+1,be[r]-1)
{
int L=lef[i],R=rig[i];
while(L<R)
{
int mid=(L+R)/2+1;
if(pre[mid]>=l) R=mid-1;
else L=mid;
}
ans+=L-lef[i]+1;
}
return ans;
}
int main()
{
int i,j,x,y;
n=read();q=read();
m=sqrt(n);
fo(i,1,n)
c[i]=read();
fo(i,1,n)
{
b[i]=last[c[i]];
last[c[i]]=i;
}
fo(i,1,m)
lef[i]=rig[i-1]+1,rig[i]=i*m;
rig[m]=n;
fo(i,1,m)
fo(j,lef[i],rig[i])
be[j]=i;
fo(i,1,m) reset(i);
while(q--)
{
scanf("%s",ch);
x=read();y=read();
if(ch[0]=='Q') printf("%d\n",ask(x,y));
else change(x,y);
}
return 0;
}