QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#768365#9528. New Energy VehiclefruianTL 1ms3600kbC++143.1kb2024-11-21 09:48:372024-11-21 09:48:38

Judging History

This is the latest submission verdict.

  • [2024-11-21 09:48:38]
  • Judged
  • Verdict: TL
  • Time: 1ms
  • Memory: 3600kb
  • [2024-11-21 09:48:37]
  • Submitted

answer

#include <bits/stdc++.h>
using namespace std;
using i64 = int64_t;
#define all(x) (x).begin(), (x).end()
#define x first
#define y second
// #define endl "\n" 
#define int long long
// #define sz(x) ((int)(x).size())
#define lowbit(x) ((x) & -(x))
#define IOS ios::sync_with_stdio(0);cin.tie(0);cout.tie(0)
#define rep(i,x,y) for(int i = x;i<y;i++)
#define reb(i,x,y) for(int i = x;i<=y;i++)
typedef long long ll;
typedef pair<int, int> pii;
typedef double db;
typedef long double ld;
const int N = 1e5+10, M = 2e7+10, mod = 998244353 , inf = 0x7fffffff;
const ld eps = 1e-12;
ll gcd(ll a, ll b) { return b ? gcd(b, a % b) : a; }
ll lcm(ll a, ll b) { return a / gcd(a, b) * b; }
ll powmod(ll a,ll b) {ll res=1;a%=mod; assert(b>=0); for(;b;b>>=1){if(b&1)res=res*a%mod;a=a*a%mod;}return res;}

struct fenwick {
    vector<int> tree;
    int sz;

    void add(int x, int d){
        if(!x)return;
        for(int i = x; i < sz; i += (i & -i)) tree[i] += d;
    }

    int sum(int x) {
        if (!x) return 0;
        if (x > sz) return 0;
        int res = 0;
        for (int i = x; i; i -= (i & -i)) res += tree[i];
        return res;
    }

    int sum(int x, int y){
        if(!x)return 0;
        return sum(y) - sum(x-1);
    }

    fenwick(int n): sz(n+1), tree(vector<int>(n+1)) {}
};


inline void solve() {
    int n, m;
    cin>>n>>m;
    vector<int>a(n+1);
    int s = 0;
    rep(i,1,n+1){
        cin>>a[i];
        s += a[i];
    }
    vector<pii>tar(m+1);
    int end = 0;
    rep(i,1,m+1){
        cin>>tar[i].x>>tar[i].y;
        end = tar[i].x;
    }

    map<int, int> rem;
    fenwick ud_tree(m+1);
    map<int,int> ud_x_to_idx;
    int ud_s = 0;
    int now = 0;
    for(int i = 1;i<=m;i++){
        int d = tar[i].x - now;
        if(d + ud_s > s)break;

        int ud_sum = ud_tree.sum(rem[tar[i].y]+1, i);
        int f = min({d+ud_sum, a[tar[i].y], tar[i].x-rem[tar[i].y]});
        
        int fx = d-f;
        // cout<<i<<" "<<fx<<" "<<ud_sum<<endl;
        ud_x_to_idx[tar[i].x] = i;
        if(fx > 0){
            ud_s += fx;
            ud_tree.add(i, fx);
        } else if(fx < 0){
            ud_s += fx;
            fx = abs(fx);
            int idx = ud_x_to_idx[rem[tar[i].y]]+1;
            int hc = ud_tree.sum(idx, idx);
            // cout<<idx<<endl;
            while(hc <= fx && fx){
                if(hc==0){
                    idx++;
                    hc = ud_tree.sum(idx, idx);
                    continue;
                }
                ud_tree.add(idx, -hc);
                idx++;
                fx -= hc;
                hc = ud_tree.sum(idx, idx);
            }
            if(fx){
                ud_tree.add(idx, -fx);
                fx = 0;
            }
        }

        rem[tar[i].y] = tar[i].x;
        now = tar[i].x;
    }

    // cout<<ud_s<<endl;
    cout<<now + s - ud_s<<endl;
}

signed main(){
    IOS;
    cout << fixed << setprecision(10);
    int __ = 1;
    cin>>__;
    while(__--) {
        solve();
        // cout<<endl;
    }
    return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 0ms
memory: 3568kb

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: 1ms
memory: 3600kb

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
Time Limit Exceeded

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:

1543020
1578939
1526016
1527381
1547404
1564631

result: