Submission #934330

#	Time	Username	Problem	Language	Result	Execution time	Memory
934330		alexander707070	Candies (JOI18_candies)	C++14	0 / 100	14 ms	2528 KiB

candies

#include<bits/stdc++.h>
#define MAXN 500007
using namespace std;

const long long inf=1e17;

int n,best;
long long a[MAXN],sum,maxs,curr;
bool used[MAXN];

long long calc(int x){
    long long res=a[x];

    for(int i=x-1;used[i] and i>=1;i-=2){
        res-=a[i]; res+=a[i-1];
    }
    for(int i=x+1;used[i] and i<=n;i+=2){
        res-=a[i]; res+=a[i+1];
    }

    return res;
}

void mark(int x){
    used[x]=true;

    for(int i=x-1;used[i] and i>=1;i-=2){
        used[i]=false; 
        used[i-1]=true;
    }
    for(int i=x+1;used[i] and i<=n;i+=2){
        used[i]=false;
        used[i+1]=true;
    }
}

int main(){

    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }

    for(int i=1;i<=n/2+n%2;i++){
        maxs=-inf;

        for(int f=1;f<=n;f++){
            if(used[f])continue;

            curr=calc(f);
            if(curr>maxs){
                maxs=curr; best=f;
            }
        }

        mark(best); sum+=maxs;

        maxs=0;
        for(int f=1;f<=n;f++){
            if(used[f])maxs+=a[f];
        }
        if(maxs!=sum)cout<<1/0;

        cout<<sum<<"\n";
    }

    return 0;
}

Compilation message (stderr)

candies.cpp: In function 'int main()':
candies.cpp:66:29: warning: division by zero [-Wdiv-by-zero]
   66 |         if(maxs!=sum)cout<<1/0;
      |                            ~^~

• Subtask 1: 0 / 8
• Subtask 2: 0 / 92