本文主要是介绍HH的项链(树状数组)区间内不同的数量,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
M行,每行一个整数,依次表示询问对应的答案。
6 1 2 3 4 3 5 3 1 2 3 5 2 6
2 2 4
题意:
有一串项链,每点有个颜色,给你个区间。查询这个区间内有几种不同的颜色?
思路:
预处理出上一个与i颜色相同的点pre[i],将询问按右端点排序,然后一边往树状数组进入点一边处理询,注意其中一些小技巧。
代码:
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
int n,m,sum;
const int maxn=1000010;
int pre[maxn],s[maxn],p[maxn],head[maxn],ans[maxn];
struct Q
{int l,r,id;
}q[maxn];
bool cmp(Q a,Q b)
{return a.r<b.r;
}
int lowbit(int i){return i&-i;
}
void updata(int x,int v) //单点更新
{if(!x) return ;sum+=v; //记录的更新的次数// cout<<sum<<"......"<<endl;while(x<=n){s[x]+=v;x+=lowbit(x);
}}
int query(int x) //查询0-x和
{int i,res=0;while(x>0){res+=s[x];x-=lowbit(x);
}return res;
}
int main()
{scanf("%d",&n);int i,j;for(i=1;i<=n;i++){scanf("%d",&p[i]);pre[i]=head[p[i]]; //仔细理解head[p[i]]=i;}scanf("%d",&m);for(i=1;i<=m;i++) scanf("%d%d",&q[i].l,&q[i].r),q[i].id=i;sort(q+1,q+m+1,cmp);for(i=j=1;i<=m;i++){for(;j<=q[i].r;j++) updata(pre[j],-1),updata(j,1); //核心处ans[q[i].id]=sum-query(q[i].l-1);}for(i=1;i<=m;i++) printf("%d\n",ans[i]);return 0;
}
多看 多学 多想
这篇关于HH的项链(树状数组)区间内不同的数量的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!