QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#713387#9528. New Energy Vehiclenahida_qaqWA 98ms13380kbC++231.9kb2024-11-05 19:11:582024-11-05 19:11:58

Judging History

This is the latest submission verdict.

  • [2024-11-05 19:11:58]
  • Judged
  • Verdict: WA
  • Time: 98ms
  • Memory: 13380kb
  • [2024-11-05 19:11:58]
  • Submitted

answer

#include <bits/stdc++.h>
#define int long long
#define io ios::sync_with_stdio(0),cin.tie(0),cout.tie(0)
#define pb push_back
using namespace std;
const int N=1e6+5,mod1=1e9+7,mod2=998244353,inf=1e18;
typedef pair<int,int> pi;
void solve()
{
    
    int n,m;
    cin>>n>>m;
    vector<int>a(100005),b(100005),x(100005),y(100005);
    for(int i=1;i<=n;i++)cin>>a[i],b[i]=a[i];
    priority_queue<pi,vector<pi>,greater<pi>>q;
    vector<int>s[n+5];
    for(int i=1;i<=m;i++)
    {
        cin>>x[i]>>y[i];
        s[y[i]].pb(i);
    }
    for(int i=1;i<=n;i++)if(s[i].size())sort(s[i].begin(),s[i].end(),greater());
    for(int i=1;i<=n;i++)
    {
        if(s[i].empty())q.push({mod1,i});
        else
        {
            q.push({s[i].back(),i});
            s[i].pop_back();
        }
    }
    int ans=0;
    int l=1;
    while(q.size())
    {
        pi t=q.top();
        q.pop();
        if(l>m)
        {
            ans+=a[t.second];
            a[t.second]=0;
        }
        else if(ans+a[t.second]>=x[l])
        {
            int w=x[l]-ans;
            ans=x[l];
            a[t.second]-=w;
            a[y[l]]=b[y[l]];
            if(y[l]!=t.second&&a[t.second]!=0)
            {
                if(!s[t.second].empty())
                {
                    q.push({s[t.second].back(),t.second});
                    s[t.second].pop_back();
                }
                else q.push({mod1,t.second});
            }
            if(!s[y[l]].empty())
            {
                q.push({s[y[l]].back(),y[l]});
                s[y[l]].pop_back();
            }
            else q.push({mod1,y[l]});
            l++;
        }
        else
        {
            ans+=a[t.second];
            a[t.second]=0;
        }
    }
    cout<<ans<<'\n';
}
signed main()
{
    io;
    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: 6908kb

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: 0
Accepted
time: 3ms
memory: 6688kb

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:

9
11
4
11
999999999
2000000000

result:

ok 6 lines

Test #3:

score: -100
Wrong Answer
time: 98ms
memory: 13380kb

input:

10
230 8042
599 1039 69 1011 1366 824 14117 1523 806 5002 332 55 3769 996 359 1040 255 1135 3454 3609 6358 2509 3695 8785 3890 1304 3394 14611 33 89 2245 508 22 1043 10411 628 1279 714 903 585 7413 5099 845 148 4689 2110 8683 1613 143 3263 2599 110 244 3297 4742 1571 425 1822 15692 572 9397 328 1691...

output:

1531889
1572162
1339347
1519343
1501335
1553572
900568
1543669
997453
932319

result:

wrong answer 1st lines differ - expected: '1543020', found: '1531889'