cf D. Vessels

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

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

 1 #include <cstdio>
 2 #include <cstring>
 3 #include <algorithm>
 4 #define LL __int64
 5 #define maxn 500000
 6 using namespace std;
 7 
 8 int f[maxn];
 9 LL a[maxn],cap[maxn];
10 
11 int find1(int x)
12 {
13     if(f[x]==x)
14     {
15         return x;
16     }
17     return f[x]=find1(f[x]);
18 }
19 
20 void deal(int pos,LL y)
21 {
22     while(y>0)
23     {
24         int c=find1(pos);
25         a[c]+=y;
26         y=a[c]-cap[c];
27         if(y<=0) return;
28         a[c]=cap[c];
29         if(a[c]==cap[c]) f[c]=f[c+1];
30     }
31 }
32 
33 int main()
34 {
35     int n;
36     scanf("%d",&n);
37     for(int i=1; i<=n; i++)
38     {
39         scanf("%I64d",&cap[i]);
40         f[i]=i;
41     }
42     f[n+1]=n+1;
43     cap[n+1]=200000000000000LL;
44     int m;
45     scanf("%d",&m);
46     while(m--)
47     {
48         int xx,p;
49         LL x;
50         scanf("%d",&xx);
51         if(xx==1)
52         {
53             scanf("%d%I64d",&p,&x);
54             deal(p,x);
55         }
56         else
57         {
58             scanf("%d\n",&p);
59             printf("%I64d\n",a[p]);
60         }
61     }
62     return 0;
63 }
View Code

 

posted @ 2014-11-02 23:16  null1019  阅读(208)  评论(0)  编辑  收藏  举报