[编程题]香槟塔算法和数据结构

[编程题]香槟塔节日到啦，牛牛和妞妞邀请了好多客人来家里做客。
他们摆出了一座高高的香槟塔，牛牛负责听妞妞指挥，往香槟塔里倒香槟。
香槟塔有个很优雅的视觉效果就是如果这一层的香槟满了，就会从边缘处往下一层流去。
妞妞会发出两种指令，指令一是往第x层塔内倒体积为v的香槟，指令二是询问第k层塔香槟的体积为多少。
告诉你香槟塔每层香槟塔的初始容量，你能帮牛牛快速回答妞妞的询问吗？
输入描述:

第一行为两个整数n，m。表示香槟塔的总层数和指令条数。
第二行为n个整数ai，表示每层香槟塔的初始容量。
第三行到第2+m行有两种输入，一种输入是“2 x v”表示往第x层倒入体积为v的香槟；另一种输入是“1 k”表示询问第k层当前有多少香槟。1 <= n, m <= 1000。
1 <= n ,m <= 200000，1 <= ai ,v <= 1000000000。

输出描述:

对于每个询问，输出一个整数，表示第k层香槟的容量。

输入例子1:

1 2 8 2 1 9 1 1

输出例子1:

8

输入例子2:

5 4 1 2 2 10 1 1 3 2 2 5 2 4 3 1 4

输出例子2:

0 4

代码

#include using namespace std; int main() { int n, m; cin>> n>> m; vector fVec(n+1), rVec(n+1); for (int i = 1; i <= n; ++i) cin>> fVec[i]; int com, x, v, k; for (int i = 0; i < m; ++i) { cin>> com; if (com == 1) { cin>> k; cout<< rVec[k]<< endl; } else { cin>> x>> v; while (v > 0 && x <= n){ if(fVec[x] - rVec[x] < v) { v -= fVec[x] - rVec[x]; // 一开始逻辑写反了，查找了20分钟 rVec[x] = fVec[x]; //先赋值了，汗 x++; } else { rVec[x] += v; v = 0; } } } } return 0; }