168022025-05-13 14:38:52AblablablaÖsszegzésekcpp17Elfogadva 100/10017ms1368 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 last value until reaching 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];
        }

        while(eddig + b[poz] < s){
            cout << b[poz] << " ";
            eddig += b[poz];
        }

        if(eddig != s){
            cout << s - eddig << "\n";
        }
    } else{
        cout << "0\n";
    }
}
RészfeladatÖsszpontTesztVerdiktIdőMemória
subtask10/0
1Elfogadva1ms316 KiB
2Elfogadva1ms316 KiB
3Elfogadva2ms316 KiB
subtask26/6
4Elfogadva1ms316 KiB
5Elfogadva1ms316 KiB
6Elfogadva1ms316 KiB
7Elfogadva1ms508 KiB
8Elfogadva1ms316 KiB
subtask36/6
9Elfogadva1ms316 KiB
10Elfogadva1ms316 KiB
11Elfogadva1ms316 KiB
12Elfogadva4ms564 KiB
13Elfogadva4ms564 KiB
subtask410/10
14Elfogadva1ms316 KiB
15Elfogadva1ms316 KiB
16Elfogadva1ms316 KiB
17Elfogadva4ms564 KiB
18Elfogadva4ms564 KiB
19Elfogadva1ms316 KiB
20Elfogadva2ms316 KiB
21Elfogadva16ms1368 KiB
22Elfogadva17ms1320 KiB
23Elfogadva16ms1300 KiB
subtask510/10
24Elfogadva1ms508 KiB
25Elfogadva1ms508 KiB
26Elfogadva1ms316 KiB
27Elfogadva1ms316 KiB
28Elfogadva1ms508 KiB
subtask610/10
29Elfogadva1ms316 KiB
30Elfogadva1ms316 KiB
31Elfogadva1ms316 KiB
32Elfogadva1ms316 KiB
33Elfogadva1ms316 KiB
subtask758/58
34Elfogadva1ms316 KiB
35Elfogadva1ms316 KiB
36Elfogadva1ms316 KiB
37Elfogadva1ms316 KiB
38Elfogadva1ms316 KiB
39Elfogadva1ms316 KiB
40Elfogadva1ms508 KiB
41Elfogadva1ms316 KiB
42Elfogadva1ms316 KiB
43Elfogadva1ms316 KiB
44Elfogadva1ms316 KiB
45Elfogadva4ms564 KiB
46Elfogadva4ms564 KiB
47Elfogadva1ms316 KiB
48Elfogadva2ms316 KiB
49Elfogadva16ms1368 KiB
50Elfogadva17ms1320 KiB
51Elfogadva1ms508 KiB
52Elfogadva1ms316 KiB
53Elfogadva1ms316 KiB
54Elfogadva1ms316 KiB
55Elfogadva1ms316 KiB
56Elfogadva1ms316 KiB
57Elfogadva1ms316 KiB
58Elfogadva1ms508 KiB
59Elfogadva1ms316 KiB
60Elfogadva1ms316 KiB
61Elfogadva1ms316 KiB
62Elfogadva1ms316 KiB
63Elfogadva2ms436 KiB
64Elfogadva2ms316 KiB
65Elfogadva3ms316 KiB
66Elfogadva4ms564 KiB