제출 #160988

#제출 시각아이디문제언어결과실행 시간메모리
160988achibasadzishvili휴가 (IOI14_holiday)C++14
24 / 100
132 ms65536 KiB
#include<bits/stdc++.h>
#pragma GCC diagnostic warning "-std=c++11"
#define ll long long
#define f first
#define s second
#define pb push_back
#define N 100005
using namespace std;
long long n,d,s,a[N],lst[N];
long long ans,co,cur,z,kk,sum1,sum2,of;
int t,raod,pp,mid,ind;
vector<long long>s2[3*N];
vector<int>v[3*N],b[3*N],g;
void build(int x){
    if((int)v[x].size() == 1){
        lst[x] = 1;
        return;
    }
    g.clear();
    for(int i=1; i<v[x].size(); i++)
        g.pb(v[x][i]);
    /*
    if(x >= 100000){
        cout << x << '\n';
        for(auto t : v[x])cout << t << " ";
        cout << '\n';
        exit(0);
    }
    if(x >= 200000)exit(0);
    */
    sort(g.begin() , g.end());
    of=0;
    mid = g[((int)g.size() - 1) / 2];
    for(int i=((int)g.size() - 1) / 2; i>=0; i--){
        if(g[i] == mid)of++;
    }
    raod = 0,pp = 0;
    v[2*x].pb(0);b[2*x].pb(0);s2[2*x].pb(0);
    v[2*x+1].pb(0);b[2*x+1].pb(0);s2[2*x+1].pb(0);
    kk = v[x][1] , sum1 = 0 , sum2 = 0;
    for(int i=1; i<v[x].size(); i++){
        t = v[x][i];
        if(t == mid)of--;
        if(t != kk)pp = 1;
        if(t <= mid && !(t == mid && of < 0)){
            v[2 * x].pb(t);
            sum1 += t;
            raod++;
        }
        else{
            v[2 * x + 1].pb(t);
            sum2 += t;
        }
        b[x].pb(raod);
        s2[x].pb(sum2);
    }
    if(!pp){lst[x] = 1;return;}
    build(2 * x);
    build(2 * x + 1);
}

void count(int x,int L,int R,int raod){
    //if(x >= 100000)exit(0);
    if(lst[x] == 1){
        if((int)v[x].size() <= 1)return;
        co += min((ll)raod , (ll)((int)v[x].size() - 1)) * v[x][1];
        return;
    }
    if(R - L + 1 - b[x][R] + b[x][L - 1] >= raod)
        count(2 * x + 1 , L - b[x][L - 1] , R - b[x][R] , raod);
    if(R - L + 1 - b[x][R] + b[x][L - 1] < raod){
        co += s2[x][R] - s2[x][L - 1];
        count(2 * x, b[x][L - 1] + 1 , b[x][R] , raod - (R - L + 1 - b[x][R] + b[x][L - 1]));
    }
}

void buildwavelet(){
    v[1].pb(0);
    b[1].pb(0);
    s2[1].pb(0);
    for(int i=1; i<=n; i++)
        v[1].pb(a[i]);
    build(1);
}
ll get(int x,int y,int k){
    co = 0;
    count(1 , x , y , k);
    return co;
}
void solve(int x,int y,int p,int q){
    if(y < x || q < 1 || p > n)return;
    mid = (x + y) / 2,ind = q; cur = -100000000000000;
    for(int i=p; i<=q; i++){
        z = get(i , mid , d - 2 * (mid - s) - (s - i));
        if(z > cur){
            cur = z;
            if(z > ans)ans = z;
            ind = i;
        }
    }
    solve(x , mid - 1 , p , ind);
    solve(mid + 1 , y , ind , q);
}
void solverev(int x,int y,int p,int q){
    if(y < x || q < 1 || p > n)return;
    mid = (x + y) / 2 , ind = p; cur = -100000000000000;
    for(int i=p; i<=q; i++){
        z = get(mid , i , d - 2 * (s - mid) - (i - s));
        if(z > cur){
            cur = z;
            ind = i;
            if(z > ans)ans = z;
        }
    }
    solverev(x , mid - 1 , p , ind);
    solverev(mid + 1 , y , ind , q);
}

long long int findMaxAttraction(int nnn,int sss,int ddd, int aaa[]){
    n = nnn;
    s = sss + 1;
    d = ddd;
    for(int i=1; i<=n; i++)a[i] = aaa[i - 1];
    ans = 0;
    buildwavelet();
    solve(s , n , 1 , s - 1);
    for(int i=s; i<=n; i++){
        z = get(s , i , d - (i - s));
        if(z > ans)
            ans = z;
    }
    for(int i=s; i>=1; i--){
        z = get(i , s , d - (s - i));
        if(z > ans)
            ans = z;
    }
    solverev(1 , s , s + 1 , n);
    return ans;
}
/*
main(){
    int nn,aa[200005],ss,dd;
    cin >> nn >> ss >> dd;
    for(int i=0; i<nn; i++){
        cin >> aa[i];
    }
    
    cout << findMaxAttraction(nn , ss , dd , aa) << endl;
    
    
    
    
    return 0;
}
*/

컴파일 시 표준 에러 (stderr) 메시지

holiday.cpp:2:32: warning: '-std=c++11' is not an option that controls warnings [-Wpragmas]
 #pragma GCC diagnostic warning "-std=c++11"
                                ^~~~~~~~~~~~
holiday.cpp: In function 'void build(int)':
holiday.cpp:20:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i=1; i<v[x].size(); i++)
                  ~^~~~~~~~~~~~
holiday.cpp:41:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i=1; i<v[x].size(); i++){
                  ~^~~~~~~~~~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...