Submission #372673

#	Time	Username	Problem	Language	Result	Execution time	Memory
372673		BJoozz	케이크 (JOI13_cake)	C++14	10 / 100	1551 ms	63792 KiB

cake

#include <bits/stdc++.h>

using namespace std;

#define X first
#define Y second
#define pb push_back
#define int long long
const int MAX = 300000+3,M=(1<<25)+1,mod=1e9+7;
void maxx(int &a,int b) {if(b>a) a=b;}
void minn(int &a,int b) {if(b<a) a=b;}
using ii = pair < int ,int > ;
using ll = long long;
int a[MAX],B[MAX][2];
int ST[MAX][20];
int minu(int i,int j){
    if(a[i]<a[j]) return i;
    else return j;
}
int n;
int LG[MAX];
int id(int u){
    return (u+n-1)%n+1;
}
int get(int l,int r){
        int u=r-l+1;
        if(r<l)u+=n;
        u=LG[u];
        return minu(ST[l][u],ST[id(r-(1<<u)+1)][u]);
}
bool tt;
int getsum(bool x,int l,int r){
    if(r>=l) return B[r][x]-B[l-1][x];
    else return B[n][x]-B[l-1][x]+B[r][x^tt];
}
int getsum2(bool x,int l,int r){
    if(r>=l) return B[r][x]-B[l-1][x];
    else return B[n][x^tt]-B[l-1][x^tt]+B[r][x];
}
int ans[MAX];


int H;
int getdown(int u,int val){
    //if(a[u]<val) return u;
    for(int z=H;z>=0;z--)
        if(a[ ST[id(u-(1<<z))][z] ]>=val ) {
            u=id(u-(1<<z));
        }
    return u;
}
int getup(int u,int val){
    for(int z=H;z>=0;z--)
        if(a[ ST[id(u+1)][z] ]>=val ) {
            u=id(u+(1<<z));
        }
    return u;
}
int dis(int L,int R){
    if(R>=L) return R-L+1;
    else return R-L+1+n;
}
int cos(int z,int L,int R){
    int res=a[z];
        int i=id(z-1),j=id(z+1);
        bool ok=1;

        while(i!=L && j!=R){
            //cout<<i<<' '<<j<<'\n';
            //if(a[j]<a[i]) {if(!ok) res+=a[i];i=id(i-1);ok^=1;continue;}
            if(a[i]<a[j]){
                int m=getup(j,a[i]);
                res+=getsum((ok^j)&1,j,m);
                ok^=dis(j,m)&1;
                j=id(m+1);
            }
            else{
                int m=getdown(i,a[j]);
                res+=getsum2((ok^i)&1,m,i);
                ok^=dis(m,i)&1;
                i=id(m-1);
            }
        }
        while(j!=R){
            if(!ok) res+=a[j];
            ok^=1;
            j=id(j+1);
        }
        while(i!=L){
            if(!ok) res+=a[i];
            ok^=1;
            i=id(i-1);
        }
    return res;
}
void GODOWN(int L,int R,int sum){
    if(R==id(L+1))return;
    int z=get(id(L+1),id(R-1));
    ans[z]=cos(z,L,R)+sum;
    GODOWN(L,z,sum+getsum( (z^tt^((L-z+1+n)%n))&1,z,id(R-1) ));
    GODOWN(z,R,sum+getsum2( (z^tt^((z-R+1+n)%n))&1,id(L+1),z ));
}

signed main() {
    ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    //freopen("A.inp", "r", stdin);freopen("A.out", "w", stdout);
    LG[1]=0;
    for(int i=2;i<MAX;i++) LG[i]=LG[i/2]+1;
    cin>>n;
    H=LG[n];
    if(H>=17) H--;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        ST[i][0]=i;
        B[i][0]=B[i-1][0];
        B[i][1]=B[i-1][1];
        B[i][i%2]+=a[i];
    }
    for(int j=1;j<20;j++){
        for(int i=1;i<=n;i++){
            ST[i][j]=minu(ST[i][j-1],ST[id(i+(1<<j-1))][j-1]);
        }
    }
    int u1=ST[1][19];
    int u2=get(id(u1+1),id(u1-1));
    int &res = ans[u1];

    tt=n&1;
    res=cos(u1,u2,u2);//return 0;
    if(tt) res+=a[u2];
    if(tt) GODOWN(u1,u1,a[u1]);
    else GODOWN(u1,u1,0);
    for(int i=1;i<=n;i++) cout<<ans[i]<<'\n';
}

Compilation message (stderr)

cake.cpp: In function 'int main()':
cake.cpp:121:51: warning: suggest parentheses around '-' inside '<<' [-Wparentheses]
  121 |             ST[i][j]=minu(ST[i][j-1],ST[id(i+(1<<j-1))][j-1]);
      |                                                  ~^~

• Subtask 1: 10 / 10
• Subtask 2: 0 / 90