0
1
mirror of https://git.sb/baoshuo/OI-codes.git synced 2024-11-08 15:38:47 +00:00

P10334 [UESTCPC 2024] 饮料

https://www.luogu.com.cn/record/167874750
This commit is contained in:
Baoshuo Ren 2024-07-22 21:14:57 +08:00
parent 723456f05e
commit cb44458fea
Signed by: baoshuo
GPG Key ID: 00CB9680AB29F51A

72
Luogu/P10334/P10334.cpp Normal file
View File

@ -0,0 +1,72 @@
#include <iostream>
#include <stack>
using std::cin;
using std::cout;
const char endl = '\n';
const int N = 2e5 + 5;
int n, t[N], a[N];
std::stack<int> st;
long long ans;
int main() {
std::ios::sync_with_stdio(false);
cin.tie(nullptr);
cin >> n;
for (int i = 1; i <= n; i++) {
cin >> t[i];
}
for (int i = 1; i <= n; i++) {
cin >> a[i];
}
for (int i = 1; i <= n; i++) {
if (t[i] < i) {
cout << -1 << endl; // 无解
exit(0); // 退出程序
}
}
int now = n; // 记录做到的杯号
// 枚举每一个时间点
for (int i = t[n]; i; i--) {
if (t[now] == i) { // 当前时间有人来取饮料,先给他制作
if (!st.empty()) {
// 注意当前所制作饮料不能小于后面提前制作的饮料(因为要保证提前制作的饮料不会被抢走)
st.emplace(std::max(st.top(), a[now]));
} else {
st.emplace(a[now]);
}
// 准备制作下一杯饮料(编号靠前的)
// 需要在前面找时间做这杯饮料
now--, i++;
} else if (t[now] < i) {
// 这个人在此之前取饮料,所以不用管它
// 当前时间就可以用来做后面需要提前制作的饮料
if (!st.empty()) {
ans += st.top();
st.pop();
}
} else { // t[now] > i
// 这个人取饮料的时间在此之后,也就是说他需要提前制作饮料
now--, i++;
}
}
while (!st.empty()) { // 把剩余的做完
ans += st.top();
st.pop();
}
cout << ans << endl;
return 0;
}