Submission #1297421

#TimeUsernameProblemLanguageResultExecution timeMemory
1297421trandangquangCake 3 (JOI19_cake3)C++20
24 / 100
386 ms327680 KiB
#include<bits/stdc++.h>
using namespace std;

#define foru(i,a,b) for(int i=(a); i<=(b); ++i)
#define ford(i,a,b) for(int i=(a); i>=(b); --i)
#define rep(i,a) for(int i=0; i<(a); ++i)
#define sz(a) (int)(a).size()
#define all(a) (a).begin(),(a).end()
#define bit(s,i) (((s)>>(i))&1)
#define ii pair<int,int>
#define vi vector<int>
#define vii vector<ii>
#define fi first
#define se second
#define ll long long
#define eb emplace_back
#define pb push_back
#define __builtin_popcount __builtin_popcountll
#define _ << " " <<

template <class X, class Y> bool maxi(X &x, Y y){return x<y?x=y,true:false;}
template <class X, class Y> bool mini(X &x, Y y){return x>y?x=y,true:false;}

const int N=202020;

int n,m; ll res=-1e18;
ii cake[N];

struct Node{
    Node *lc,*rc;
    ll sum; int cnt;

    Node(){
        lc=rc=nullptr;
        sum=0; cnt=0;
    }
};
Node *ver[N];

ll walk(int ql, int qr){
    Node *idL=ver[ql-1],*idR=ver[qr];

    int l=1, r=1e9; ll res=0; int lt=m;
    while(l<r){
        int mid=(l+r)>>1;

        int cntR=0; ll sumR=0;
        if(idR!=0 && idR->rc!=0){
            cntR+=idR->rc->cnt;
            sumR+=idR->rc->sum;
        }
        if(idL!=0 && idL->rc!=0){
            cntR-=idL->rc->cnt;
            sumR-=idL->rc->sum;
        }

//        cout<<l _ mid _ r _ lt _ cntR _ sumR<<'\n';
        if(lt>cntR){
            lt-=cntR;
            res+=sumR;
            if(idR!=0) idR=idR->lc;
            if(idL!=0) idL=idL->lc;
            r=mid;
        } else{
            if(idR!=0) idR=idR->rc;
            if(idL!=0) idL=idL->rc;
            l=mid+1;
        }
    }

//    cout<<res _ lt _ l<<'\n';
    return res+1LL*lt*l;
}

ll cost(int i, int j){
    if(j-i+1<m) return -1e18;
//    cerr<<i _ j _ walk(i,j)<<'\n';
    return walk(i,j)-2*(cake[j].se-cake[i].se);
}

void calc(int L, int R, int optL, int optR){
    if(L>R) return;
    int M=(L+R)>>1;

    ll pos=0, ans=-1e18;
    foru(opt,optL,min(optR,M)){
        if(maxi(ans,cost(opt,M))){
            pos=opt;
        }
    }
    maxi(res,ans);

    calc(L,M-1,optL,pos);
    calc(M+1,R,pos,optR);
}

Node *update(int x, Node *old, int l, int r){
    if(l==r){
        Node *cur=new Node();
        *cur=*old;
        cur->cnt++;
        cur->sum+=x;
        return cur;
    }

    Node *cur=new Node();

    int mid=(l+r)>>1;
    if(x<=mid){
        if(old->lc==0) old->lc=new Node();
        cur->lc=update(x,old->lc,l,mid);
        cur->rc=old->rc;
    } else{
        cur->lc=old->lc;
        if(old->rc==0) old->rc=new Node();
        cur->rc=update(x,old->rc,mid+1,r);
    }

    if(cur->lc!=0){
        cur->sum+=cur->lc->sum;
        cur->cnt+=cur->lc->cnt;
    }
    if(cur->rc!=0){
        cur->sum+=cur->rc->sum;
        cur->cnt+=cur->rc->cnt;
    }
    return cur;
}

void solve(){
    cin>>n>>m;
    foru(i,1,n) cin>>cake[i].fi>>cake[i].se;
    sort(cake+1, cake+1+n, [](ii x, ii y){return x.se<y.se;});

//    foru(i,1,n) cout<<cake[i].fi<<'\n';
    ver[0]=new Node();
    foru(i,1,n){
        ver[i]=update(cake[i].fi,ver[i-1],1,1e9);
    }

//    cout<<walk(1,5)<<'\n';
    calc(m,n,1,n);
    cout<<res<<'\n';
}

int32_t main(){
    #define task "test"
    if(fopen(task".inp", "r")){
        freopen(task".inp", "r", stdin);
        freopen(task".out", "w", stdout);
    }
    cin.tie(0)->sync_with_stdio(0);

    int tc=1; //cin>>tc;
    foru(i,1,tc){
        solve();
    }
}

Compilation message (stderr)

cake3.cpp: In function 'int32_t main()':
cake3.cpp:149:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  149 |         freopen(task".inp", "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
cake3.cpp:150:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  150 |         freopen(task".out", "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...