Submission #1071109

#	Time	Username	Problem	Language	Result	Execution time	Memory
1071109		phong	Candies (JOI18_candies)	C++17	0 / 100	3 ms	5208 KiB

candies

#include<bits/stdc++.h>

#define ll long long
#define int long long
const int nmax = 3e5 + 5, N = 1e5;
const ll oo = 1e18 + 1, base = 311;
const int lg = 19, M = 10;
const ll mod = 998244353, mod2 = 1e9 + 5277;
#define pii pair<int, int>
#define fi first
#define se second
#define endl "\n"
#define debug(a, n) for(int i = 1; i <= n; ++i) cout << a[i] << ' '; cout << "\n";
using namespace std;

int n, a[nmax];
int ans[nmax];

main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);
//    freopen("code.inp", "r", stdin);
//    freopen("code.out", "w", stdout);
    cin >> n;
    for(int i = 1; i <= n; ++i) cin >> a[i];
    multiset<pii> one, two;
    a[0] = -oo;
    a[n + 1] = -oo;
    for(int i = 0; i <= n + 1; ++i){
        one.insert({a[i], i});
        two.insert({i, a[i]});
    }
    for(int i = 1; i <= (n + 1) / 2; ++i){
        auto x = *one.rbegin();
        ans[i] = ans[i - 1] + x.fi;
        cout << ans[i] << endl;
        auto it = two.lower_bound({x.se, -oo});
        --it;
        auto s1 = it;
        one.erase({it->se, it->fi});
        two.erase(it++);
        auto s2 = it;
        one.erase({it->se, it->fi});
        two.erase(it++);
        auto s3 = it;
        one.erase({it->se, it->fi});
        two.erase(it++);
        one.insert({s1->se + s3->se - s2->se, s2->fi});
        two.insert({s2->fi, s1->se + s3->se - s2->se});


    }

}
/*




*/

Compilation message (stderr)

candies.cpp:19:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   19 | main(){
      | ^~~~

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