题解：洛谷 P4879 ycz的妹子

最新推荐文章于 2025-12-31 21:25:26 发布

原创最新推荐文章于 2025-12-31 21:25:26 发布 · 621 阅读

9 ·

CC 4.0 BY-SA版权

文章标签：

#算法

题目https://www.luogu.com.cn/problem/P4879感觉还比较简单的线段树。

首先我们先建立一棵线段树（范围： $[1,5\times 10^5]$ ）。

void build(int k,int l,int r){
	tr[k]={l,r};
	if(l==r){
		Tree[k]=a[l],c[k]=(l<=n);
		return;
	}
	int mid=(l+r)>>1ll;
	build(k<<1ll,l,mid);
	build((k<<1ll)|1ll,mid+1,r);
	push_up(k);
}

对于操作一

线段树单点修改，我们缩小区间范围，直到找到 $x$ 所在的叶子节点，然后将叶子节点减去 $y$ 。

对于操作二

先找到 $x$ 所在的叶子节点，然后将数字顶替掉。

操作一 + 操作二实现：

void modify(int k,int x,int val,bool flag){
	if(tr[k].l==tr[k].r&&tr[k].l==x){
		c[k]=(!flag?1:c[k]);
		Tree[k]=(flag?Tree[k]+val:val);
		return;
	}
	int mid=(tr[k].l+tr[k].r)>>1ll;
	if(mid>=x){
		modify(k<<1ll,x,val,flag);
	}else{
		modify((k<<1ll)|1ll,x,val,flag);
	}
	push_up(k);
}

对于操作三

最难的一个点了好吧（其实也没那么难）。

我们维护区间中的妹子数量（ $c$ 数组的用处就是这个）。

当然，如果你更新的是右区间，找的不是右边区间的第 $x$ 个妹子，而是 $x$ 减去左边区间的妹子数量。

void Delete(int k,int x){
	if(tr[k].l==tr[k].r&&c[k]==x){
		Tree[k]=0,c[k]--;
		return;
	}
	if(c[k<<1ll]>=x){
		Delete(k<<1ll,x);
	}else{
		Delete((k<<1ll)|1ll,x-c[k<<1ll]);
	}
	push_up(k);
}

对于操作四

嗯……直接输出根节点的值就好了（就是这么简单）。

实现

#include<bits/stdc++.h>
using namespace std;
#define int long long
int n,m,a[500005],Tree[2000005],c[2000005];
struct node{
	int l,r;
}tr[2000005];
void push_up(int k){
	Tree[k]=Tree[k<<1ll]+Tree[(k<<1ll)|1ll];
	c[k]=c[k<<1ll]+c[(k<<1ll)|1ll];
}
void build(int k,int l,int r){
	tr[k]={l,r};
	if(l==r){
		Tree[k]=a[l],c[k]=(l<=n);
		return;
	}
	int mid=(l+r)>>1ll;
	build(k<<1ll,l,mid);
	build((k<<1ll)|1ll,mid+1,r);
	push_up(k);
}
void modify(int k,int x,int val,bool flag){
	if(tr[k].l==tr[k].r&&tr[k].l==x){
		c[k]=(!flag?1:c[k]);
		Tree[k]=(flag?Tree[k]+val:val);
		return;
	}
	int mid=(tr[k].l+tr[k].r)>>1ll;
	if(mid>=x){
		modify(k<<1ll,x,val,flag);
	}else{
		modify((k<<1ll)|1ll,x,val,flag);
	}
	push_up(k);
}
void Delete(int k,int x){
	if(tr[k].l==tr[k].r&&c[k]==x){
		Tree[k]=0,c[k]--;
		return;
	}
	if(c[k<<1ll]>=x){
		Delete(k<<1ll,x);
	}else{
		Delete((k<<1ll)|1ll,x-c[k<<1ll]);
	}
	push_up(k);
}
signed main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	build(1,1,500000);
	int x,y;
	char ID;	
	while(m--){
		cin>>ID;
		if(ID=='C'){
			cin>>x>>y;
			modify(1,x,-y,true);
		}else if(ID=='I'){
			cin>>x>>y;
			modify(1,x,y,false);
		}else if(ID=='D'){
			cin>>x;
			Delete(1,x);
		}else{
			cout<<Tree[1]<<'\n';
		}
	}
	return 0;
}