QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#715244 | #9528. New Energy Vehicle | ljljljlj# | WA | 0ms | 3528kb | C++20 | 2.6kb | 2024-11-06 11:02:51 | 2024-11-06 11:02:58 |
Judging History
answer
#include <bits/stdc++.h>
#define int long long
#define endl '\n'
using namespace std;
using i64 = long long;
const int inf = 1e18;
void solve()
{
int n, k;
cin >> n >> k;
int sum = 0;
vector<int> val(n + 1, 0);
vector<int> a(n + 1, 0);
vector<pair<int, int>> b(k + 2);
vector<int> fa(n + 10, 0);
vector<int> vis(max(n, k) + 10, 0);
map<int, int> mp;
priority_queue<int, vector<int>, greater<int>> q;
for (int i = 1; i <= n; i++)
{
cin >> val[i];
a[i] = val[i];
}
for (int i = 1; i <= k; i++)
{
cin >> b[i].first >> b[i].second;
}
b[k + 1] = {1e18, 0};
for (int i = k + 1; i; i--)
{
fa[i] = mp[b[i].second];
mp[b[i].second] = i;
q.push(i);
vis[b[i].second] = 1;
}
for (int i = 1; i <= n; i++)
{
if (!vis[i])
sum += val[i];
}
int ans = 0;
// cout << sum;
for (int i = 1; i <= k + 1; i++)
{
int dis = b[i].first - b[i - 1].first;
while (q.size())
{
int x = q.top();
q.pop();
if (a[b[x].second] < dis)
{
dis -= a[b[x].second];
ans += a[b[x].second];
a[b[x].second] = 0;
}
else
{
ans += dis;
a[b[x].second] -= dis;
if (x > i)
q.push(x);
break;
}
}
if (q.size())
{
// cout << i << '*' << endl;
if (fa[i])
q.push(fa[i]);
else
sum += val[b[i].second];
a[b[i].second] = val[b[i].second];
// cout << a[b[i].second] << '*' << endl;
}
else
{
// cout << "*" << dis << endl;
sum -= dis;
if (sum >= 0)
{
ans += dis;
if (fa[i])
q.push(fa[i]);
else
sum += val[b[i].second];
a[b[i].second] = val[b[i].second];
}
else
{
ans += (sum + dis);
cout << ans << '\n';
return;
}
}
// cout << ans << ' ' << a[1] << " " << a[2] << endl;
}
}
signed main()
{
ios::sync_with_stdio(false), cin.tie(0);
int T = 1;
cin >> T;
while (T--)
solve();
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 3520kb
input:
2 3 1 3 3 3 8 1 2 2 5 2 1 2 2 1
output:
12 9
result:
ok 2 lines
Test #2:
score: -100
Wrong Answer
time: 0ms
memory: 3528kb
input:
6 3 2 2 2 2 6 1 7 1 2 2 3 3 2 1 6 2 2 3 2 2 5 1 7 2 9 1 2 2 3 3 2 1 6 2 1 1 999999999 1000000000 1 1 1 1000000000 1000000000 1
output:
7 11 4 11 999999999 2000000000
result:
wrong answer 1st lines differ - expected: '9', found: '7'