oj.uz

Submission #224748

# Submission time Handle Problem Language Result Execution time Memory
224748 2020-04-18T17:53:05 Z muhammad_hokimiyon Candies (JOI18_candies) C++14
0 / 100
 7 ms 384 KB 
#include <bits/stdc++.h>

//mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
//#pragma GCC optimize("Ofast")
//1.0 * clock() / CLOCKS_PER_SEC

#define fi first
#define se second
#define ll long long
#define dl double long

using namespace std;

const ll NN = 1e10 + 7;
const int N = 1e5 + 7;
const int M = 6;
const int mod = 1e9 + 7;
const ll inf = 1e18 + 7;
const dl rf = 1e-14;
const int B = sqrt(N);

int n;
int p[N];
ll a[N];
set < pair < ll , ll > > s,ss;

int get( int x )
{
    return (p[x] == x ? x : p[x] = get(p[x]));
}

void solve1()
{
    cin >> n;
    for( int i = 1; i <= n; i++ ){
        cin >> a[i];
        p[i] = i;
        ss.insert({ a[i] , i });
    }
    p[n + 1] = n + 1;
    ll ans = 0;
    for( int i = 1; i <= (n + 1) / 2; i++ ){
        if( ss.empty() )break;
        auto x = *(--ss.end());
        ss.erase(--ss.end());
        ans += x.fi;
        cout << ans << "\n";
        int l = get( x.se - 1 );
        int r = get( x.se + 1 );
        if( l != 0 ){
            ss.erase({ a[l] , l });
        }
        if( r != n + 1 ){
            ss.erase({ a[r] , r });
        }
        int f = get( x.se );
        if( l != 0 && r != n + 1 ){
            ll mz = a[l] + a[r] - x.fi;
            p[f] = l;
            f = get( x.se );
            p[r] = f;
            a[f] = mz;
            ss.insert({ mz , f });
        }
    }
}

int main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);
    freopen( "input.txt" , "r" , stdin );
    freopen( "output.txt" , "w" , stdout );

    int cghf = 1;//cin >> cghf;
    while( cghf-- ){
        solve1();
    }
}

Compilation message

candies.cpp: In function 'int main()':
candies.cpp:72:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)', declared with attribute warn_unused_result [-Wunused-result]
     freopen( "input.txt" , "r" , stdin );
     ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~
candies.cpp:73:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)', declared with attribute warn_unused_result [-Wunused-result]
     freopen( "output.txt" , "w" , stdout );
     ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

Subtask #1
0 / 8.0

# Verdict Execution time Memory Grader output
1 Incorrect 7 ms 384 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
0 / 92.0

# Verdict Execution time Memory Grader output
1 Incorrect 7 ms 384 KB Output isn't correct
2 Halted 0 ms 0 KB -