ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

noip晚间小测5 A. 统计 线段树

2020-11-27 17:01:12  阅读:173  来源: 互联网

标签:ch 小测 noip int 线段 da rg return tr2


题目描述


分析

设 \(f[i]\) 为在 \(i\) 右边且比 \(a[i]\) 大的数的个数

则初始的答案为 \(\sum_{i=1}^nf[i]\)

如果操作的位置是 \(m\)

就相当于把在 \(m\) 右边且小于等于 \(a[m]\) 的位置上的 \(f\) 置为 \(0\)

因为每一个点只会对答案贡献一次,暴力枚举会浪费很多时间

可以用线段树去加速

如果一段区间内没有比 \(a[m]\) 小的数,直接吧整个区间跳过

如果一个位置已经被访问过,就把访问过的地方置为无穷大

时间复杂度 \(nlogn\)

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#define rg register
inline int read(){
	rg int x=0,fh=1;
	rg char ch=getchar();
	while(ch<'0' || ch>'9'){
		if(ch=='-') fh=-1;
		ch=getchar();
	}
	while(ch>='0' && ch<='9'){
		x=(x<<1)+(x<<3)+(ch^48);
		ch=getchar();
	}
	return x*fh;
}
const int maxn=1e6+5;
int n,m,tr[maxn],f[maxn],a[maxn];
int Min(int aa,int bb){
	return aa<bb?aa:bb;
}
long long ans=0;
struct asd{
	int id,val;
}b[maxn];
bool cmp(asd aa,asd bb){
	if(aa.val==bb.val) return aa.id<bb.id;
	return aa.val<bb.val;
}
inline int lb(int xx){
	return xx&-xx;
}
int cxsum(int wz){
	rg int nans=0;
	for(rg int i=wz;i>0;i-=lb(i)){
		nans+=tr[i];
	}
	return nans;
}
void ad(int wz,int val){
	for(rg int i=wz;i<maxn;i+=lb(i)){
		tr[i]+=val;
	}
}
int cxqj(int l,int r){
	if(l>r) return 0;
	return cxsum(r)-cxsum(l-1);
}
struct trr{
	int l,r,val;
}tr2[maxn<<2];
void push_up(int da){
	tr2[da].val=Min(tr2[da<<1].val,tr2[da<<1|1].val);
}
void build(int da,int l,int r){
	tr2[da].l=l,tr2[da].r=r;
	if(tr2[da].l==tr2[da].r){
		tr2[da].val=a[l];
		return;
	}
	rg int mids=(l+r)>>1;
	build(da<<1,l,mids);
	build(da<<1|1,mids+1,r);
	push_up(da);
}
void solve(int da,int l,int r,int val){
	if(tr2[da].val>val) return;
	if(tr2[da].l==tr2[da].r){
		ans-=f[tr2[da].l];
		tr2[da].val=0x3f3f3f3f;
		return;
	}
	rg int mids=(tr2[da].l+tr2[da].r)>>1;
	if(l<=mids) solve(da<<1,l,r,val);
	if(r>mids) solve(da<<1|1,l,r,val);
	push_up(da);
}
int main(){
	n=read(),m=read();
	for(rg int i=1;i<=n;i++){
		b[i].val=read();
		b[i].id=i;
		a[i]=b[i].val;
	}
	std::sort(b+1,b+1+n,cmp);
	for(rg int i=1;i<=n;i++){
		ad(b[i].id,1);
		f[b[i].id]=cxqj(b[i].id+1,n);
	}
	for(rg int i=1;i<=n;i++){
		ans+=f[i];
	}
	build(1,1,n);
	printf("%lld ",ans);
	rg int aa;
	for(rg int i=1;i<=m;i++){
		aa=read();
		solve(1,aa,n,a[aa]);
		printf("%lld ",ans);
	}
	printf("\n");
	return 0;
}

标签:ch,小测,noip,int,线段,da,rg,return,tr2
来源: https://www.cnblogs.com/liuchanglc/p/14049095.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有