Codeforces 371D Vessels (模拟)

题目链接　Vessels

这道题我做得有点稀里糊涂啊＝＝TLE了几发之后改了一行就A了。

具体思路就是记fi为若第i个容器已经盛不下水了，那么接下来盛水的那个容器。

hi为若现在要给i号容器加水，当前真正被加水的那个容器。

这样就大大降低了时间复杂度。

 #include <bits/stdc++.h>

 using namespace std;
 #define rep(i,a,b) for(int i(a); i <= (b); ++i)

 const int N = ;
 long long a[N], b[N], f[N], h[N], c[N], d[N], op, x, cap, p;
 int n, q, k;

 int main(){

     scanf("%d", &n);
     rep(i, , n) scanf("%lld", a + i);
     a[n + ] = 1e18; f[n + ] = n + ; h[n + ] = n + ;
     rep(i, , n) b[i] = , f[i] = i + , h[i] = i, c[i] = ;
     scanf("%d", &q);
     while (q--){
         scanf("%lld", &op);
         if (op == ){
             scanf("%d%lld", &k, &p);
             while (p > ){
                 x = h[k]; int l = x, cnt = ;
                 while (c[x]){
                     d[++cnt] = x;
                     x = f[x];
                 }

                 rep(i, , cnt){
                     h[d[i]] = x;
                     f[d[i]] = x;
                 }

                 cap = a[x] - b[x];
                 if (cap < p){
                     b[x] = a[x];
                     c[x] = ;
                     p -= cap;
                 }
                 else b[x] += p, p = ;
                 k = x;
             }
         }
         else{
             scanf("%lld", &op);
             printf("%lld\n", b[op]);
         }
     }

     return ;

 }