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

题目描述

分析

设 \(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