【Codeforces】 CF992E. Nastya and King-Shamans

题目链接

Codeforces方向

Luogu方向

题目解法

如果 a[i] = s[i-1]，那么 s[i] - s[i-1] = s[i-1]

说明 s[i] = 2*s[i-1]，s[i] 会翻一倍

这说明出现 a[i] = s[i-1] 的点不会超过 个

同理 a[i] - s[i-1] >= 0 的点不会超过 个

由于题目中有大量的区间操作，于是考虑线段树

可以用线段树维护 a[i] - s[i-1] 的最大值

每次查找时只递归查找最大的 a[i] - s[i-1]  >=0 的区间

这样可以保证时间复杂度是

修改时就是单点修改+区间修改

这道题启发我们一些暴力做法有时会因为特殊的性质，复杂度大大下降，从而可以用数据结构维护

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int N(200100);
int n,m,a[N],s[N];
struct SegmentTree{
	struct Node{
		int maxi,tag;
	}seg[N<<2];
	void build(int l,int r,int x){
		if(l==r){
			seg[x].maxi=a[l]-s[l-1];
			return;
		}
		int mid=(l+r)>>1;
		build(l,mid,x<<1),build(mid+1,r,x<<1^1);
		seg[x].maxi=max(seg[x<<1].maxi,seg[x<<1^1].maxi); 
	}
	void pushdown(int x){
		seg[x<<1].maxi+=seg[x].tag,seg[x<<1].tag+=seg[x].tag;
		seg[x<<1^1].maxi+=seg[x].tag,seg[x<<1^1].tag+=seg[x].tag;
		seg[x].tag=0;
	}
	void modify(int l,int r,int x,int L,int R,int k){
		if(L<=l&&r<=R){
			seg[x].maxi+=k,seg[x].tag+=k;
//			cout<<l<<" : "<<seg[x].maxi<<'\n';
			return;
		}
		pushdown(x);
		int mid=(l+r)>>1;
		if(mid>=L)
			modify(l,mid,x<<1,L,R,k);
		if(mid<R)
			modify(mid+1,r,x<<1^1,L,R,k);
		seg[x].maxi=max(seg[x<<1].maxi,seg[x<<1^1].maxi); 
//		cout<<l<<' '<<r<<" : "<<seg[x].maxi<<'\n'; 
	}
	int query(int l,int r,int x){
//		cout<<l<<' '<<r<<" : "<<seg[x].maxi<<'\n';
		if(seg[x].maxi<0)
			return -1;
		if(l==r){
			if(seg[x].maxi==0)
				return l;
			return -1;
		}
		pushdown(x);
		int mid=(l+r)>>1,lc=query(l,mid,x<<1);
		if(lc!=-1)
			return lc;
		return query(mid+1,r,x<<1^1);
	}
}SGT;
inline int read(){
	int FF=0,RR=1;
	char ch=getchar();
	for(;!isdigit(ch);ch=getchar())
		if(ch=='-')
			RR=-1;
	for(;isdigit(ch);ch=getchar())
		FF=(FF<<1)+(FF<<3)+ch-48;
	return FF*RR;
}
signed main(){
	n=read(),m=read();
	for(int i=1;i<=n;i++)
		a[i]=read(),s[i]=s[i-1]+a[i];
//	for(int i=1;i<=n;i++)
//		cout<<a[i]-s[i-1]<<' ';
//	cout<<'\n';
	SGT.build(1,n,1);
	for(int i=1,p,x;i<=m;i++){
		p=read(),x=read();
//		cout<<x-a[p]<<'\n';
		SGT.modify(1,n,1,p,p,x-a[p]);
		if(p<n)
			SGT.modify(1,n,1,p+1,n,a[p]-x);
		a[p]=x;
		printf("%lld\n",SGT.query(1,n,1));
	}
	return 0;
}