cf D. Vessels

http://codeforces.com/contest/371/problem/D

第一遍写的超时了，然后看了别人的代码，思路都是找一个点的根，在往里面加水的时候碗中的水满的时候建立联系。查询的时候直接查询就行。

#include <cstdio>
#include <cstring>
#include <algorithm>
#define LL __int64
#define maxn 500000
using namespace std;

int f[maxn];
LL a[maxn],cap[maxn];

int find1(int x)
{
    if(f[x]==x)
    {
        return x;
    }
    return f[x]=find1(f[x]);
}

void deal(int pos,LL y)
{
    while(y>0)
    {
        int c=find1(pos);
        a[c]+=y;
        y=a[c]-cap[c];
        if(y<=0) return;
        a[c]=cap[c];
        if(a[c]==cap[c]) f[c]=f[c+1];
    }
}

int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1; i<=n; i++)
    {
        scanf("%I64d",&cap[i]);
        f[i]=i;
    }
    f[n+1]=n+1;
    cap[n+1]=200000000000000LL;
    int m;
    scanf("%d",&m);
    while(m--)
    {
        int xx,p;
        LL x;
        scanf("%d",&xx);
        if(xx==1)
        {
            scanf("%d%I64d",&p,&x);
            deal(p,x);
        }
        else
        {
            scanf("%d
",&p);
            printf("%I64d
",a[p]);
        }
    }
    return 0;
}