168012025-05-13 14:34:19AblablablaÖsszegzésekcpp17Partially correct 61/10018ms1332 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);
        }
        for (int i = 0; i < a.size(); i++)
        {
            cout << a[i] << " ";
        }
        cout << "\n";
    } else{
        cout << "0\n";
    }

    //find one lower (earlier)
    //method: drop first non-one value by one, add value needed to achieve sum
    //sum is same, but lexicographically lower now

    int poz = -1;
    for (int i = b.size() - 1; i >= 0; i--)
    {
        if (b[i] > 1)
        {
            b[i]--;
            poz = i;
            break;
        }
    }

    if(poz != -1){
        int eddig = 0;
        for(int i = 0; i <= poz; i++){
            cout << b[i] << " ";
            eddig += b[i];
        }
        cout << s - eddig << "\n";
    } else{
        cout << "0\n";
    }
}
SubtaskSumTestVerdictTimeMemory
subtask10/0
1Accepted1ms316 KiB
2Accepted1ms316 KiB
3Partially correct1ms508 KiB
subtask26/6
4Accepted1ms316 KiB
5Accepted1ms368 KiB
6Accepted1ms316 KiB
7Accepted1ms316 KiB
8Accepted1ms500 KiB
subtask36/6
9Accepted1ms316 KiB
10Accepted1ms316 KiB
11Accepted1ms316 KiB
12Accepted4ms564 KiB
13Accepted4ms748 KiB
subtask410/10
14Accepted1ms316 KiB
15Accepted1ms316 KiB
16Accepted1ms316 KiB
17Accepted4ms564 KiB
18Accepted4ms748 KiB
19Accepted1ms316 KiB
20Accepted2ms316 KiB
21Accepted16ms1328 KiB
22Accepted18ms1332 KiB
23Accepted16ms1312 KiB
subtask55/10
24Partially correct1ms316 KiB
25Partially correct1ms496 KiB
26Partially correct1ms316 KiB
27Partially correct1ms316 KiB
28Partially correct1ms316 KiB
subtask65/10
29Accepted1ms316 KiB
30Partially correct1ms316 KiB
31Accepted1ms328 KiB
32Accepted1ms316 KiB
33Accepted1ms316 KiB
subtask729/58
34Accepted1ms316 KiB
35Accepted1ms316 KiB
36Partially correct1ms316 KiB
37Accepted1ms316 KiB
38Accepted1ms368 KiB
39Accepted1ms316 KiB
40Accepted1ms316 KiB
41Accepted1ms500 KiB
42Accepted1ms316 KiB
43Accepted1ms316 KiB
44Accepted1ms316 KiB
45Accepted4ms564 KiB
46Accepted4ms748 KiB
47Accepted1ms316 KiB
48Accepted2ms316 KiB
49Accepted16ms1328 KiB
50Accepted18ms1332 KiB
51Partially correct1ms316 KiB
52Accepted1ms316 KiB
53Partially correct1ms316 KiB
54Accepted1ms328 KiB
55Accepted1ms316 KiB
56Accepted1ms316 KiB
57Partially correct1ms316 KiB
58Accepted1ms316 KiB
59Accepted1ms556 KiB
60Accepted1ms316 KiB
61Partially correct1ms412 KiB
62Partially correct1ms316 KiB
63Partially correct2ms316 KiB
64Partially correct2ms316 KiB
65Partially correct3ms316 KiB
66Partially correct4ms564 KiB