题目连接 : https://www.nowcoder.com/acm/contest/139/J
题意:给出长为n的数组a1,a2...an,有q次查询,每次查询给出Li,Ri,让求数组中去掉开区间(aLi,aRi)后剩余部分中不同元素的个数的和。
思路 : 树状数组有求区间不同数的做法,现在分开求2个区间,所以考虑拼接一个区间,加在后边就可以树状数组解决问题了。
#include<bits/stdc++.h>
using namespace std;
int a[200005];
struct node{
int l;
int r;
int id;
}no[100005];
int ans[200005];
int Ans[200005];
map<int,int>M;
int n,m;
int lowbit(int x)
{
return x&(-x);
}
void upda(int x,int val)
{
while(x<=2*n)
{
ans[x]+=val;
x=x+lowbit(x);
}
}
int sum(int x)
{
int ANS=0;
while(x>0)
{
ANS+=ans[x];
x=x-lowbit(x);
}
return ANS;
}
int cmp(node a,node b)
{
if(a.r == b.r)
return a.l < b.l;
return a.r < b.r;
}
int main()
{
while(scanf("%d%d",&n,&m)!=EOF)
{
memset(ans,0,sizeof(ans));
M.clear();
for(int i = 1;i <= n;i++)
{
scanf("%d",&a[i]);
a[i+n] = a[i];
}
for(int i = 0;i < m; i++)
{
scanf("%d%d",&no[i].l,&no[i].r);
no[i].id = i;
int temp = no[i].r;
no[i].r = n + no[i].l;
no[i].l = temp;
}
sort(no,no+m,cmp);
int l=1;
for(int i = 0;i < m;i++)
{
for(int j = l ;j <= no[i].r;j++)
{
if(M.find(a[j])!=M.end())
{
upda(M[a[j]],-1);
}
upda(j,1);
M[a[j]]=j;
}
l=no[i].r+1;
Ans[no[i].id]=sum(no[i].r)-sum(no[i].l-1);
}
for(int i = 0 ;i < m ;i++)
printf("%d\n",Ans[i]);
}
return 0;
}
莫队也可以过,但是我T到怀疑人生,后来加了输入挂 更改了分块的大小 才过。但是是当时1000ms过了,后来交了1400ms。
#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5+5;
struct node{
int l,r,id;
}que[maxn];
int ans[maxn];
int pos[maxn];
int a[maxn];
int vis[maxn];
int sum;
inline bool scan_d(int &num)
{
char in;bool IsN=false;
in=getchar();
if(in==EOF) return false;
while(in!='-'&&(in<'0'||in>'9')) in=getchar();
if(in=='-'){ IsN=true;num=0;}
else num=in-'0';
while(in=getchar(),in>='0'&&in<='9'){
num*=10,num+=in-'0';
}
if(IsN) num=-num;
return true;
}
int cmp(node a,node b)//右端点
{
if(pos[a.l] == pos[b.l])
return a.r<b.r;
return pos[a.l] < pos[b.l];
}
void add(int x)
{
if(vis[a[x]] == 0)
sum++;
vis[a[x]]++;
}
void del(int x)
{
if(vis[a[x]] == 1)
sum--;
vis[a[x]]--;
}
int main()
{
int n,m;
while(scanf("%d%d",&n,&m)!=EOF)
{
sum=0;
int block = (int)pow(n,0.57);
for(int i=1;i<=n;i++)
{
vis[i]=0;
ans[i]=0;
}
for(int i=1;i<=n;i++)
{
scan_d(a[i]);
pos[i]=i/block;
vis[a[i]]++;
if(vis[a[i]] == 1)
sum++;
}
for(int i=1;i<=m;i++)
{
scan_d(que[i].l);
scan_d(que[i].r);
que[i].id=i;
if(que[i].r - que[i].l <= 1)//挨着
ans[i] = sum;
else
{
que[i].l++;
que[i].r--;
}
}
sort(que+1,que+m+1,cmp);
int l=1,r=0;
for(int i=1;i<=m;i++)
{
if(ans[que[i].id]) continue;
while(l<que[i].l) add(l++);
while(l>que[i].l) del(--l);
while(r>que[i].r) add(r--);
while(r<que[i].r) del(++r);
ans[que[i].id] = sum;
}
for(int i=1;i<=m;i++) printf("%d\n",ans[i]);
}
return 0;
}