Submission #74963168

Source Code Expand

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

typedef long long ll;

void solve()
{
    int n;
    cin >> n;
    vector<ll> a(n + 1);
    ll sum = 0;
    for (int i = 1; i <= n; i++)
    {
        cin >> a[i];
        sum += a[i];
    }
    if (sum % n != 0) { cout << -1 << '\n'; return; }
    ll t = sum / n;
    for (int i = 1; i <= n; i++)
    {
        a[i] = a[i] - t;
    }
    ll ans = 0;
    int extra = 0;
    for (int i = 1; i <= n; i++)
    {
        if (a[i] < 0)
        {
            ans += -a[i];
            a[i] += extra;
            if (i + 1 <= n)
                a[i + 1] += a[i];
            if (a[i] > 0) { extra += a[i]; a[i] = 0; }
            else extra = 0;
        }
        else if (a[i] > 0)
        {
            extra += a[i];
            ans += extra;
            a[i] = 0;
        }

    }
    cout << ans << '\n';
}

int main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    
    // int t;
    // cin >> t;
    // while (t--) solve();

    solve();
    
    return 0;
}

Submission Info

Submission Time
Task D - Equal Distribution of Cake
User rotas
Language C++ IOI-Style(GNU++20) (GCC 14.2.0)
Score 0
Code Size 1072 Byte
Status WA
Exec Time 9 ms
Memory 3116 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 0 / 400
Status
AC × 3
AC × 36
WA × 37
Set Name Test Cases
Sample sample01.txt, sample02.txt, sample03.txt
All sample01.txt, sample02.txt, sample03.txt, in01.txt, in02.txt, in03.txt, in04.txt, in05.txt, in06.txt, in07.txt, in08.txt, in09.txt, in10.txt, in11.txt, in12.txt, in13.txt, in14.txt, in15.txt, in16.txt, in17.txt, in18.txt, in19.txt, in20.txt, in21.txt, in22.txt, in23.txt, in24.txt, in25.txt, in26.txt, in27.txt, in28.txt, in29.txt, in30.txt, in31.txt, in32.txt, in33.txt, in34.txt, in35.txt, in36.txt, in37.txt, in38.txt, in39.txt, in40.txt, in41.txt, in42.txt, in43.txt, in44.txt, in45.txt, in46.txt, in47.txt, in48.txt, in49.txt, in50.txt, in51.txt, in52.txt, in53.txt, in54.txt, in55.txt, in56.txt, in57.txt, in58.txt, in59.txt, in60.txt, in61.txt, in62.txt, in63.txt, in64.txt, in65.txt, in66.txt, in67.txt, in68.txt, in69.txt, in70.txt
Case Name Status Exec Time Memory
in01.txt AC 0 ms 1580 KiB
in02.txt AC 0 ms 1584 KiB
in03.txt AC 0 ms 1580 KiB
in04.txt AC 0 ms 1580 KiB
in05.txt WA 0 ms 1580 KiB
in06.txt AC 0 ms 1580 KiB
in07.txt AC 9 ms 3116 KiB
in08.txt AC 8 ms 3116 KiB
in09.txt WA 6 ms 3116 KiB
in10.txt AC 9 ms 3116 KiB
in11.txt WA 9 ms 3116 KiB
in12.txt WA 7 ms 3116 KiB
in13.txt WA 6 ms 3116 KiB
in14.txt WA 6 ms 3116 KiB
in15.txt AC 0 ms 1580 KiB
in16.txt AC 8 ms 3116 KiB
in17.txt AC 7 ms 3116 KiB
in18.txt WA 7 ms 3116 KiB
in19.txt WA 9 ms 3116 KiB
in20.txt AC 7 ms 3116 KiB
in21.txt AC 6 ms 3116 KiB
in22.txt WA 7 ms 3116 KiB
in23.txt WA 0 ms 1592 KiB
in24.txt WA 0 ms 1580 KiB
in25.txt WA 0 ms 1580 KiB
in26.txt WA 0 ms 1604 KiB
in27.txt WA 0 ms 1580 KiB
in28.txt WA 0 ms 1580 KiB
in29.txt WA 0 ms 1580 KiB
in30.txt AC 0 ms 1580 KiB
in31.txt WA 0 ms 1580 KiB
in32.txt WA 0 ms 1580 KiB
in33.txt AC 0 ms 1580 KiB
in34.txt AC 0 ms 1580 KiB
in35.txt AC 6 ms 3116 KiB
in36.txt AC 5 ms 3116 KiB
in37.txt AC 0 ms 1580 KiB
in38.txt AC 0 ms 1580 KiB
in39.txt WA 6 ms 3116 KiB
in40.txt AC 0 ms 1580 KiB
in41.txt WA 0 ms 1580 KiB
in42.txt WA 8 ms 3116 KiB
in43.txt WA 7 ms 3116 KiB
in44.txt AC 8 ms 3116 KiB
in45.txt WA 9 ms 3116 KiB
in46.txt WA 7 ms 3116 KiB
in47.txt WA 8 ms 3116 KiB
in48.txt WA 9 ms 3116 KiB
in49.txt WA 7 ms 3116 KiB
in50.txt WA 9 ms 3116 KiB
in51.txt WA 9 ms 3116 KiB
in52.txt WA 9 ms 3116 KiB
in53.txt AC 0 ms 1580 KiB
in54.txt WA 0 ms 1580 KiB
in55.txt AC 0 ms 1580 KiB
in56.txt WA 0 ms 1580 KiB
in57.txt WA 0 ms 1580 KiB
in58.txt WA 0 ms 1580 KiB
in59.txt AC 0 ms 1580 KiB
in60.txt AC 0 ms 1580 KiB
in61.txt AC 0 ms 1580 KiB
in62.txt AC 0 ms 1580 KiB
in63.txt AC 0 ms 1580 KiB
in64.txt AC 0 ms 1580 KiB
in65.txt AC 0 ms 1580 KiB
in66.txt AC 0 ms 1608 KiB
in67.txt WA 0 ms 1580 KiB
in68.txt WA 0 ms 1580 KiB
in69.txt AC 9 ms 3116 KiB
in70.txt WA 9 ms 3116 KiB
sample01.txt AC 0 ms 1580 KiB
sample02.txt AC 0 ms 1604 KiB
sample03.txt AC 0 ms 1580 KiB