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

                rep(i, 1, cnt){
                    h[d[i]] = x;
                    f[d[i]] = x;
                }
                
                cap = a[x] - b[x];
                if (cap < p){
                    b[x] = a[x];
                    c[x] = 1;
                    p -= cap;
                }
                else b[x] += p, p = 0;
                k = x;
            }
        }
        else{
            scanf("%lld", &op);
            printf("%lld\n", b[op]);
        }
    }

    return 0;

}