168002025-05-13 14:09:47TortelliniJrÖsszegzésekcpp17Partially correct 68/10017ms1332 KiB
#include <iostream>
#include <vector>
using namespace std;
int main()
{
    int n, s, x, t, u;
    vector<int> a, b, o;
    s = 0;
    cin >> n;
    for (int i = 0; i < n; i++)
    {
        cin >> x;
        a.push_back(x);
        b.push_back(x);
        o.push_back(x);
        s += x;
    }
    //find one higher
    //method: go from end, find smallest number that neighbors one that is bigger, increment it and save its location
    //but: never can be the last number: cant add ones
    //so we start later
    //also end before 0: dont want overindexing
    //if none found: increment first
    //solution: copy everything until location, then spam 1 until sum = s
    if (o.size() != 1)
    {
        int z = s - a[a.size() - 1];
        int loc = 0;
        for (int i = a.size() - 2; i > 0; i--)
        {
            if (a[i] < a[i - 1])
            {
                loc = i;
                break;
            }
            else
            {
                z -= a[i];
            }
        }
        a[loc]++;
        z++;
        a.resize(loc + 1);
        for (int i = 0; i < s - z; i++)
        {
            a.push_back(1);
        }
    }
    //find one lower (earlier)
    //method: drop first non-one value by one, proceed to append one
    //sum is same, but lexicographically lower now
    for (int i = b.size() - 1; i >= 0; i--)
    {
        if (b[i] != 1)
        {
            b[i]--;
            b.push_back(1);
            break;
        }
    }
    for (int i = 0; i < a.size(); i++)
    {
        cout << a[i] << " ";
    }
    cout << endl;
    if (b != o)
    {
        for (int i = 0; i < b.size(); i++)
        {
            cout << b[i] << " ";
        }
    }
    else
    {
        cout << "0";
    }
}
SubtaskSumTestVerdictTimeMemory
subtask10/0
1Partially correct1ms316 KiB
2Accepted1ms508 KiB
3Accepted1ms316 KiB
subtask23/6
4Partially correct1ms508 KiB
5Partially correct1ms328 KiB
6Partially correct1ms316 KiB
7Partially correct1ms332 KiB
8Partially correct1ms316 KiB
subtask36/6
9Accepted1ms316 KiB
10Accepted1ms316 KiB
11Accepted1ms408 KiB
12Accepted4ms564 KiB
13Accepted4ms564 KiB
subtask410/10
14Accepted1ms316 KiB
15Accepted1ms316 KiB
16Accepted1ms408 KiB
17Accepted4ms564 KiB
18Accepted4ms564 KiB
19Accepted1ms316 KiB
20Accepted2ms508 KiB
21Accepted16ms1328 KiB
22Accepted17ms1332 KiB
23Accepted16ms1176 KiB
subtask510/10
24Accepted1ms316 KiB
25Accepted1ms544 KiB
26Accepted1ms508 KiB
27Accepted1ms500 KiB
28Accepted1ms316 KiB
subtask610/10
29Accepted1ms384 KiB
30Accepted1ms316 KiB
31Accepted1ms316 KiB
32Accepted1ms316 KiB
33Accepted1ms316 KiB
subtask729/58
34Partially correct1ms316 KiB
35Accepted1ms316 KiB
36Accepted2ms316 KiB
37Partially correct1ms508 KiB
38Partially correct1ms328 KiB
39Partially correct1ms316 KiB
40Partially correct1ms332 KiB
41Partially correct1ms316 KiB
42Accepted1ms316 KiB
43Accepted1ms316 KiB
44Accepted1ms408 KiB
45Accepted4ms564 KiB
46Accepted4ms564 KiB
47Accepted1ms316 KiB
48Accepted2ms508 KiB
49Accepted16ms1328 KiB
50Accepted17ms1332 KiB
51Accepted1ms316 KiB
52Accepted1ms384 KiB
53Accepted1ms316 KiB
54Accepted1ms316 KiB
55Accepted1ms316 KiB
56Accepted1ms316 KiB
57Accepted1ms508 KiB
58Accepted1ms384 KiB
59Accepted1ms316 KiB
60Accepted1ms316 KiB
61Accepted1ms316 KiB
62Accepted1ms316 KiB
63Accepted2ms328 KiB
64Accepted2ms316 KiB
65Accepted3ms316 KiB
66Accepted4ms564 KiB