【BZOJ2453】维护队列
Description
Input
Output
Sample Input
1 2
Q 1 2
R 1 2
Q 1 2
Sample Output
1
HINT
对于100%的数据,有1 ≤ N ≤ 10000, 1 ≤ M ≤ 10000,小朋友A不会修改超过1000次,所有颜色均用1到10^6的整数表示。
题解:听说这题卡树套树(没写过)
我们还是考虑分块
一个经典的分块思路:我们用pre[i]表示i之前第一个与i颜色相同的弹珠,这样在询问时,在[l,r]中pre[i]小于l的i的个数就是答案。发现这个pre数组有一个很好的性质,即单调性,因此我们想到在教主的魔法那里用的办法,新建一个数组,块内按pre排序,这样就能二分查找出一个块内pre[i]<l的i的个数,然后就很容易了
现在我们考虑怎么修改,题目里说保证修改次数不多于1000,因此网上很多大犇都用的是暴力修改,但毕竟我们是读书人,怎能用暴力呢?
其实不用暴力修改也很(ju)容(ma)易(fan)
我们新开一个数组s[i][j]表示第i个块里,颜色为j个弹珠的个数。然后修改时我们要找到这个点:前面+后面,与原来的颜色相同的+与修改后颜色相同的,最近的点是哪个,(说白了就是维护pre数组),那我们就先在自己的块里找,然后在一个一个块扫,直接看s数组就可以知道块里有没有要找的数,然后在进入块里一个一个扫。时间复杂度:扫块sqrt(n),扫块内sqrt(n),所以加起来还是sqrt(n),比起那些O(n)来说简直有莫大的优越感
现在问题来了,你们想看压行的代码还是不压行的代码?反正我压了
#include <cstdio>
#include <cstring>
#include <iostream>
#include <cmath>
#include <algorithm>
using namespace std;
const int maxn=11010;
int n,m,siz,ans,tot;
char str[5];
int p[maxn],q[maxn],ref[1000010],pre[maxn],s[110][maxn];
void rebuild(int x)
{
for(int i=x*siz;i<x*siz+siz;i++) q[i]=pre[i];
sort(q+x*siz,q+x*siz+siz);
}
int main()
{
scanf("%d%d",&n,&m);
siz=int(sqrt(1.0*n));
int i,j,k,a,b,l,r,mid,flag;
memset(ref,-1,sizeof(ref));
for(i=0;i<n;i++)
{
scanf("%d",&p[i]);
if(ref[p[i]]==-1) ref[p[i]]=++tot;
p[i]=ref[p[i]],s[i/siz][p[i]]++,pre[i]=-1,flag=0;
if(s[i/siz][p[i]]>1) for(k=i-1;k>=i/siz*siz&&!flag;k--) if(p[k]==p[i]) pre[i]=k,flag=1;
for(j=i/siz-1;j>=0&&!flag;j--) if(s[j][p[i]]) for(k=j*siz+siz-1;k>=j*siz&&!flag;k--) if(p[k]==p[i]) pre[i]=k,flag=1;
}
for(i=n;i<(n-1)/siz*siz+siz;i++) p[i]=pre[i]=1<<30;
for(i=0;i*siz<=n;i++) rebuild(i);
for(i=1;i<=m;i++)
{
scanf("%s%d%d",str,&a,&b);
if(str[0]=='Q')
{
a--,b--,ans=0;
if(a/siz==b/siz)
{
for(j=a;j<=b;j++) ans+=pre[j]<a;
printf("%d\n",ans);
continue;
}
for(j=a;j<a/siz*siz+siz;j++) ans+=pre[j]<a;
for(j=b/siz*siz;j<=b;j++) ans+=pre[j]<a;
for(j=a/siz*siz+siz;j<b/siz*siz;j+=siz)
{
l=j,r=j+siz;
while(l<r)
{
mid=l+r>>1;
if(q[mid]<a) l=mid+1;
else r=mid;
}
ans+=l-j;
}
printf("%d\n",ans);
}
else
{
flag=0,a--;
if(ref[b]==-1) ref[b]=++tot;
b=ref[b];
if(s[a/siz][p[a]]>1) for(k=a+1;k<a/siz*siz+siz&&!flag;k++) if(pre[k]==a) pre[k]=pre[a],flag=1;
for(j=a/siz+1;j*siz<n&&!flag;j++) if(s[j][p[a]]) for(k=j*siz;k<j*siz+siz&&!flag;k++) if(pre[k]==a) pre[k]=pre[a],flag=1,rebuild(j);
s[a/siz][p[a]]--,flag=0;
if(s[a/siz][b]) for(k=a-1;k>=a/siz*siz&&!flag;k--) if(p[k]==b) pre[a]=k,flag=1;
for(j=a/siz-1;j>=0&&!flag;j--) if(s[j][b]) for(k=j*siz+siz-1;k>=j*siz&&!flag;k--) if(p[k]==b) pre[a]=k,flag=1,rebuild(j);
if(!flag) pre[a]=-1;
flag=0;
if(s[a/siz][b]) for(k=a+1;k<a/siz*siz+siz&&!flag;k++) if(p[k]==b) pre[k]=a,flag=1;
for(j=a/siz+1;j*siz<n&&!flag;j++) if(s[j][b]) for(k=j*siz;k<j*siz+siz&&!flag;k++) if(p[k]==b) pre[k]=a,flag=1,rebuild(j);
s[a/siz][b]++,p[a]=b;
rebuild(a/siz);
}
}
return 0;
}