제출 #160994

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
160994		achibasadzishvili	휴가 (IOI14_holiday)	C++14	47 / 100	167 ms	65540 KiB

holiday

#include<bits/stdc++.h>
#define ll long long
#define f first
#define s second
#define pb push_back
#define N 100005
using namespace std;
int n,d,s,a[N],lst[3 * N];
long long ans,co,cur,z,kk,sum1,sum2,of;
int t,raod,pp,mid,ind;
vector<long long>s1[3*N + 100];
vector<ll>v[3*N + 100],b[3*N + 100],g;
void build(int x,int dep){
    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]);
    sort(g.begin() , g.end());
    if(g[0] == g[g.size() - 1]){
        lst[x] = 1;
        return;
    }
    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);s1[2*x].pb(0);
    v[2*x+1].pb(0);b[2*x+1].pb(0);s1[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);
        s1[x].pb(sum2);
    }
    if(!pp){lst[x] = 1;return;}
    build(2 * x,dep + 1);
    build(2 * x + 1,dep + 1);
}

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

void buildwavelet(){
    v[1].pb(0);
    b[1].pb(0);
    s1[1].pb(0);
    for(int i=1; i<=n; i++)
        v[1].pb(a[i]);
    build(1 , 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(){
    ifstream cin("input.in");
    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: In function 'void build(int, int)':
holiday.cpp:19:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i=1; i<v[x].size(); i++)
                  ~^~~~~~~~~~~~
holiday.cpp:35:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i=1; i<v[x].size(); i++){
                  ~^~~~~~~~~~~~

• Subtask 1: 7 / 7
• Subtask 2: 23 / 23
• Subtask 3: 17 / 17
• Subtask 4: 0 / 53