CodeForces - 371D 线段树+二分

线段树+二分水题。。

线段树维护第i个容器还能装多少水

倒水d到第x容器，先判断水是否会流到地上，流到，就把所有容器的容量赋值0.

否则，二分找到，x---pos 容量和第一个大于d的容器。把x --- pos-1  的容器容量赋值0，第pos个容器的水量为x---pos区间和减去d。所以容量为tmp-水量即可。

查询就单点查询。

注意tg可能为0.

//KX
#include <bits/stdc++.h>
using namespace std;
template<typename T>inline void rd(T&x){
    x=0;int f=1;char ch=getchar();
    while(ch<'0' ||ch>'9'){ if(ch=='-')f=-1; ch=getchar(); }
    while(ch>='0' && ch<='9'){ x=x*10+ch-'0'; ch=getchar(); }
    x*=f;
}
typedef long long ll;
typedef double db;
const int M = 2e5+7;
#define ls (o<<1)
#define rs (o<<1|1)
#define m ((l+r)>>1)
#define pb push_back
ll tr[M<<2],tg[M<<2];
void pd(int o,int l,int r)
{
	if(tg[o]==-1)return ;
	tg[ls]=tg[rs]=tg[o];
	tr[ls]=tg[o]*(m-l+1);
	tr[rs]=tg[o]*(r-m);
	tg[o]=-1;
}
void up(int o,int l,int r,int x,int y,ll d)
{
//	printf("                            %d %d %d   %d  %d  %lld\n",l,r,m,x,y,d);
	if(x<=l&&r<=y)
	{
		tg[o]=d;
		tr[o]=(r-l+1)*d;
		return ;
	}
	pd(o,l,r);
	if(x<=m)up(ls,l,m,x,y,d);
	if(y>m)up(rs,m+1,r,x,y,d);
	tr[o]=tr[ls]+tr[rs];
}
ll qu(int o,int l,int r,int x,int y)
{
	if(x<=l&&r<=y)
	{
		return tr[o];
	}
	ll ans=0;
	pd(o,l,r);
	if(x<=m)ans+=qu(ls,l,m,x,y);
	if(y>m)ans+=qu(rs,m+1,r,x,y);
	tr[o]=tr[ls]+tr[rs];
	return ans;
}
ll a[M];
int main()
{
	ll n,q;
	memset(tg,-1,sizeof(tg));
	rd(n);
	for(int i=1;i<=n;i++)
  	rd(a[i]),up(1,1,n,i,i,a[i]);
  	rd(q);
  	while(q--)
  	{
  		ll op,x,d;
  		rd(op);
  		if(op==1)
  		{
  			rd(x),rd(d);
  			if(qu(1,1,n,x,n)<d)
  			{
  				up(1,1,n,x,n,0);
  				continue;
			}
  			
  			int l=x,r=n;
			int pos=l,cz=0;
  			while(l<=r)
  			{
  				int mid=(l+r)/2;
  				ll tmp=qu(1,1,n,x,mid);
  				if(tmp>d)
  				{ 	
  					r=mid-1;
  					pos=mid;
  					cz=tmp-d;
				}
				else l=mid+1;
			}
			if(pos>x)
				up(1,1,n,x,pos-1,0);
			ll tmp=qu(1,1,n,pos,pos);
			up(1,1,n,pos,pos,cz);
		}
		else
		{
			rd(x);
			ll ans=qu(1,1,n,x,x);
			printf("%lld\n",a[x]-ans);
		}
	  }
  	return 0;
}