洛谷P1972
AtCoder Beginner Contest 174-F
树状数组的离线处理
首先对询问进行读入,把询问右区间进行排序处理,再逐步向后更新树状数组
树状数组储存值为从1-j的不同数字的个数
最终答案就是FIND( r ) - FIND( l - 1 )
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <map>
#include <queue>
#include <functional>
#include <vector>
#include <stack>
#include <set>
using namespace std;
typedef long long ll;
const int maxn=1e6+50;
const int inf=0x3f3f3f3f;
const int MOD=1e9+7;
const int HASH=131;
int c[maxn];
int jungle[maxn];
int lowbit(int x)
{
return x&(-x);
}
void updata(int x,int y,int n)
{
for(int i=x;i<=n;i+=lowbit(i))
c[i]+=y;
}
int FIND(int x)
{
int ans=0;
for(int i=x;i;i-=lowbit(i))
ans+=c[i];
return ans;
}
struct node
{
int l,r;
int id;
friend bool operator<(node a,node b)
{
return a.r<b.r;
}
}ques[maxn];
int a[maxn];
int ans[maxn];
int main()
{
int n;
cin>>n;
int k;//两道题就这里不相同
cin>>k;
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
}
for(int i=1;i<=k;i++)
{
scanf("%d %d",&ques[i].l,&ques[i].r);
ques[i].id=i;//记录编号,后期离线处理重新排序
}
sort(ques+1,ques+1+k);//快排
int next=1;从1开始遍历
for(int i=1;i<=k;i++)
{
for(int j=next;j<=ques[i].r;j++)//往后遍历,一直到询问值右区间
//到右区间是防止后面出现重复数字覆盖前面导致答案出错
{
if(jungle[a[j]])//判断是否出现过,出现过就对区间值-1
{
updata(jungle[a[j]],-1,n);
}
updata(j,1,n);
jungle[a[j]]=j;//无论是否出现过,用新地址覆盖原地址
}
next=ques[i].r+1;//下次从r+1开始更新,碰到相同r在循环内也直接跳出
ans[ques[i].id]=FIND(ques[i].r)-FIND(ques[i].l-1);
}
for(int i=1;i<=k;i++)
printf("%d\n",ans[i]);
return 0;
}