香槟塔
内存限制: 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;
}