QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#715244#9528. New Energy Vehicleljljljlj#WA 0ms3528kbC++202.6kb2024-11-06 11:02:512024-11-06 11:02:58

Judging History

This is the latest submission verdict.

  • [2024-11-06 11:02:58]
  • Judged
  • Verdict: WA
  • Time: 0ms
  • Memory: 3528kb
  • [2024-11-06 11:02:51]
  • Submitted

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'