Submission #946235

#	Time	Username	Problem	Language	Result	Execution time	Memory
946235		Captain_Georgia	Labels (NOI20_labels)	C++17	25 / 100	137 ms	18772 KiB

Labels

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

#define file \
            freopen("in.txt" , "r" , stdin); \
            freopen("out.txt" , "w" , stdout);

typedef long long ll;

const int mod = 998244353;

void test_case () {
    int n;
    cin >> n;
    int dif[n - 1] , a[n];
    a[0] = 1;
    set<int> s;
    s.insert(1);
    for (int i = 0;i < n - 1;i ++) {
        cin >> dif[i];
        a[i + 1] = a[i] + dif[i];
        s.insert(a[i + 1]);
    }
    if (s.size() < n || (*s.begin() < 1 && *s.rbegin() > n)) {
        cout << "-1\n";
        return;
    }
    if (*s.begin() < 1) {
        int tmp = 1 - *s.begin();
        for (int i = 0;i < n;i ++) {
            a[i] += tmp;
            cout << a[i] << " ";
        }
        cout << "\n";
    }
    else if (*s.rbegin() > n) {
        int tmp = *s.rbegin() - n;
        for (int i = 0;i < n;i ++) {
            a[i] -= tmp;
            cout << a[i] << " ";
        }
        cout << "\n";
    }
    else {
        for (int i = 0;i < n;i ++) {
            cout << a[i] << " ";
        }
        cout << "\n";
    }
}

int32_t main () {

    ///  file
    int TIME = clock();

    int t = 1;
    ///    cin >> t;

    while (t --) {
        test_case();
    }

    cerr << "\nTime elapsed: " << (clock() - TIME) * 1000.0 / CLOCKS_PER_SEC << " ms\n";

}

Compilation message (stderr)

Labels.cpp: In function 'void test_case()':
Labels.cpp:24:18: warning: comparison of integer expressions of different signedness: 'std::set<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   24 |     if (s.size() < n || (*s.begin() < 1 && *s.rbegin() > n)) {
      |         ~~~~~~~~~^~~

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