[编程题]香槟塔

[编程题]香槟塔

节日到啦，牛牛和妞妞邀请了好多客人来家里做客。
他们摆出了一座高高的香槟塔，牛牛负责听妞妞指挥，往香槟塔里倒香槟。
香槟塔有个很优雅的视觉效果就是如果这一层的香槟满了，就会从边缘处往下一层流去。
妞妞会发出两种指令，指令一是往第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;
}