上海计算机学会2023年8月月赛C++乙组T3香槟塔

香槟塔

内存限制: 256 Mb时间限制: 1000 ms

题目描述

有一个高度为 n 层的香槟塔，最高层记为第一层，最底层记为第 n 层，每一层有最大容量，其中第 i 层香槟杯的容量为 wi​。

现给定 q 次操作，每次操作为倒香槟或查询中的一个：

• 当 op=A 时，表示在第 x 层的香槟杯中，倒入 y 个单位的香槟。
• 当 op=Q 时，表示查询当前第 x 层的香槟杯中目前有多少单位香槟。

第 i 层香槟杯倒满后，多余的香槟会溢出流向第 i+1 层香槟杯，若第 i+1 层香槟杯也溢出，则会流向第 i+2 层香槟杯，以此类推，最底层香槟杯溢出的香槟不计入询问范围。

输入格式

输入第一行，一个正整数 n
输入第二行，n 个正整数 w1​,w2​,...,wn​
输入第三行，一个正整数 q
接下来 q 行，每行表示一次操作，以 A x y 或 Q x 的形式给出。

输出格式

对于每个询问，输出对应答案，以换行为分割标志。

数据范围

• 对于 30% 的数据， 1≤n,q≤10
• 对于 60% 的数据， 1≤n,q≤10^3
• 对于 100% 的数据， 1≤n,q≤10^5,1≤x≤n,1≤wi​,y≤10^9

样例数据

输入:
2
2 5
4
A 1 1
A 2 7
Q 1
Q 2
输出:
1
5
 

解析：

此题采用模拟法：模拟法得了70分

详见代码：

#include<bits/stdc++.h>
using namespace std;
string s;
int n,q;
char op;
int mi;//变动里的最小值
int w[100005];
int a[100005];
int main() {
    cin>>n;
    for (int i=1;i<=n;i++){
        cin>>w[i];
    }
    cin>>q;
    mi=n;
    while(q--){
        cin>>op;
        if (op=='A'){//处理A操作
            int x,y;
            cin>>x>>y;
            a[x]+=y;
            mi=min(mi,x);//找到最小变动的层
        }else if(op=='Q') {//处理Q操作
            int x;
            cin>>x;
            for (int i=mi;i<=n;i++){//从最小变动的层处理到最后一层
                if (a[i]>w[i]){
                    a[i+1]+=a[i]-w[i];
                    a[i]=w[i];
                }
            }
            mi=n;
            cout<<a[x]<<endl;
        }
    }
    return 0;
}

正解代码是采用查并集，给每一层倒酒时忽略它下边满了的层，直接对第一个没满的层倒酒。

详见代码：

#include<bits/stdc++.h>
using namespace std;
int father[100005];
long long w[100005];
long long a[100005];
int n;
int q;
char op;

int findf(int x){//找最底层
    if (father[x]==x) return x;
    father[x]=findf(father[x]);
    return father[x];
}
void Union(int x,int y){//合并
    int xx=findf(x);
    int yy=findf(y);
    if (xx!=yy){
        father[xx]=father[yy];
    }
}

void daojiu(int x,int y){//向x层到y杯酒
    if (x>n) return;//如果超出n层，返回
    if (a[x]+y<=w[x]){//如果没倒满
        a[x]+=y;//全倒了
        return;//返回
    }else{//如果有剩余
        y=a[x]+y-w[x];//剩余y杯酒
        a[x]=w[x];//x层倒满
        if (findf(x)==x){//如果该层没倒满过
            Union(x,x+1);//加入下一层的队伍
        }
        daojiu(findf(x),y);//向队伍的最底层倒酒
    }
}
int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin>>n;
    for (int i=1;i<=n;i++){
        cin>>w[i];
        father[i]=i;//初始化每层的最底层为自己
    }
    father[n+1]=n+1;//设置第你+层哨兵
    cin>>q;
    while(q--){
        cin>>op;
        if (op=='A'){//如果倒酒
            int x,y;
            cin>>x>>y;
            daojiu(x,y);//就倒酒
        }else if(op=='Q') {//如果查询
            int x;
            cin>>x;
            cout<<a[x]<<endl;//就返回
        }
    }
    return 0;
}