[HEOI2012]采花(树状数组维护贡献之和)

[HEOI2012]采花

https://ac.nowcoder.com/acm/contest/19684/C

参考博客

分析:此题要出现了两个及以上的数量才算贡献,{1,2,2,3,1}可以看作{1,1,0,0,0},当一种颜色出些两个以上时,对其出现的倒数第二个位置计算贡献;树状数组维护的是贡献之和

第一次抖过去

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define low(x) (x&(-x))
int const N=2e6+7;
struct L{
	int l,r,p;
}a[N];
inline bool const cmp(const L& a,const L& b){
	return a.r<b.r;
}
int n,c,m;
int tr[N<<2];
inline void add(int p,int const w){
	for(;p<=n;p+=low(p)){
		tr[p]+=w;
	}
}
inline int ask(int p){
	int res=0;
	for(;p>0;p-=low(p)){
		res+=tr[p];
	}
	return res;
}
int col[N],pre[N],pre2[N],ans[N];  //pre[i]表示颜色i的上一个位置在哪,pre2[i]则表示上上次在哪
int main(){
	cin >> n >> c >> m;
	for(int i=1;i<=n;++i) cin >> col[i];
	for(int i=1;i<=m;++i){
		cin >> a[i].l >> a[i].r;
		a[i].p=i;
	}
	sort(a+1,a+m+1,cmp);
	int j=1;
	for(int i=1;i<=m;++i){
		for(;j<=a[i].r;++j){
			if(pre2[ col[j] ]){
				add(pre2[ col[j] ],-1);
			}
			pre2[col[j]]=pre[col[j]];pre[col[j]]=j;
			if(pre2[col[j]]) add(pre2[col[j]],1);
		}
		ans[a[i].p]=ask(a[i].r)-ask(a[i].l-1);
	}
	for(int i=1;i<=m;++i) cout << ans[i] << "\n";
	return 0;
}
全部评论

相关推荐

想按时下班的大菠萝在...:隔壁学校的,加油多投, 实在不好找可以下个学期开学找,把算法八股准备好,项目有空再换换
投了多少份简历才上岸
点赞 评论 收藏
分享
06-02 15:17
门头沟学院 Java
心爱的idea:怎么会呢 应该是打招呼有问题 问就说实习6个月全国可飞随时到岗
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务