求区间有多少不同的数
可以莫队,离线树状数组,主席树做
先写一个莫队的做法
代码如下:
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
int n,m,l,r,visit[1000005],pos[50005],a[50005],sz,ans,sum[200005];
struct Node
{
int l,r,id;
}q[200005];
bool cmp(Node a,Node b)
{
if(pos[a.l]==pos[b.l]) return a.r<b.r;
else return a.l<b.l;
}
bool cmp_id(Node a,Node b)
{
return a.id<b.id;
}
void add(int p)
{
if(!visit[p]) ans++;
visit[p]++;
}
void del(int p)
{
visit[p]--;
if(!visit[p]) ans--;
}
int main()
{
scanf("%d",&n); sz=sqrt(n);
for(int i=1;i<=n;i++) {
scanf("%d",&a[i]);
pos[i]=(i-1)/sz+1;
}
scanf("%d",&m);
for(int i=1;i<=m;i++) {
scanf("%d%d",&q[i].l,&q[i].r);
q[i].id=i;
}
sort(q+1,q+1+m,cmp);
int l=1,r=0;
for(int i=1;i<=m;i++) {
while(l<q[i].l) {del(a[l++]);}
while(l>q[i].l) {add(a[--l]);}
while(r<q[i].r) {add(a[++r]);}
while(r>q[i].r) {del(a[r--]);}
sum[q[i].id]=ans;
}
for(int i=1;i<=m;i++) {
printf("%d\n",sum[i]);
}
}
主席树
一直WA,一直没找到错在哪里,最后发现last开小了 QAQ 啊啊
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=60000;
int n,m,last[1000005],root[N],a[N],cnt,temp,x,y;
struct Node
{
int l,r,sum;
}p[N*40];
void Update(int l,int r,int &x,int y,int pos,int v)
{
p[++cnt]=p[y]; p[cnt].sum+=v; x=cnt;
if(l==r) {
return ;
}
int mid=l+r>>1;
if(pos<=mid) {
Update(l,mid,p[x].l,p[y].l,pos,v);
}
else {
Update(mid+1,r,p[x].r,p[y].r,pos,v);
}
}
int Query(int l,int r,int pos,int x)
{
// printf("%d %d\n",pos,x);
if(l==r) {
return p[x].sum;
}
int mid=l+r>>1;
if(pos<=mid) {
return p[p[x].r].sum+Query(l,mid,pos,p[x].l);
}
else{
return Query(mid+1,r,pos,p[x].r);
}
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++) {
scanf("%d",&a[i]);
}
for(int i=1;i<=n;i++) {
int v=a[i];
if(last[v]>0) {
Update(1,n,temp,root[i-1],last[v],-1);
Update(1,n,root[i],temp,i,1);
}
else{
Update(1,n,root[i],root[i-1],i,1);
}
last[v]=i;
}
scanf("%d",&m);
for(int i=1;i<=m;i++) {
scanf("%d%d",&x,&y);
printf("%d\n",Query(1,n,x,root[y]));
}
}