Submission #1016290

#	Time	Username	Problem	Language	Result	Execution time	Memory
1016290		toast12	Labels (NOI20_labels)	C++14	100 / 100	86 ms	7968 KiB

Labels

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

int main() {
    int n;
    cin >> n;
    vector<int> nums(n);
    for (int i = 1; i < n; i++) {
        cin >> nums[i];
    }
    vector<int> ps(n);
    for (int i = 1; i < n; i++) {
        ps[i] = ps[i-1]+nums[i];
    }
    auto it = max_element(ps.begin(), ps.end());
    int x  = *min_element(ps.begin(), ps.end());
    if (*it-x != n-1) {
        cout << "-1\n";
        return 0;
    }
    vector<int> ans(n);
    int idx = it-ps.begin();
    ans[idx] = n;
    for (int i = idx+1; i < n; i++) {
        ans[i] = ans[i-1]+nums[i];
    }
    for (int i = idx-1; i >= 0; i--) {
        ans[i] = ans[i+1]-nums[i+1];
    }
    for (int i = 0; i < n; i++) {
        cout << ans[i] << ' ';
    }
    cout << '\n';
    return 0;
}

• Subtask 1: 7 / 7
• Subtask 2: 15 / 15
• Subtask 3: 25 / 25
• Subtask 4: 18 / 18
• Subtask 5: 35 / 35