86062024-01-23 10:14:16BenedekMartonKombináció (50)cpp17Partially correct 43/503ms3936 KiB
#include <iostream>

using namespace std;

int main()
{
    int n, m;
    cin >> n >> m;
    int a[m], x=-1, k;
    for(int i=0; i<m; i++)
    {
        cin >> a[i];
    }
    if(a[0]==n-m+1)
    {
        a[0]=max(n-m, 1);
        for(int i=0; i<m; i++)
        {
            cout << a[i] << " ";
        }
        cout << endl;
        for(int i=1; i<=m; i++)
        {
            cout << i << " ";
        }
    }
    else if(a[m-1]==m)
    {
        for(int i=n-m+1; i<=n; i++)
        {
            cout << i << " ";
        }
        cout << endl;
        a[m-1]=min(m+1, n);
        for(int i=0; i<m; i++)
        {
            cout << a[i] << " ";
        }
    }
    else
    {
        for(int i=m-1; i>=0 and x==-1; i--)
        {
            if(a[i]-a[i-1]>1)
            {
                x=i;
            }
        }
        for(int i=0; i<m; i++)
        {
            if(i<x)
            {
                cout << a[i] << " ";
            }
            else if(i==x)
            {
                k=a[i]-1;
                cout << k << " ";
            }
            else
            {
                k=n-(m-i)+1;
                cout << k << " ";
            }
        }
        cout << endl;
        x=-1;
        for(int i=m-1; i>=0 and x==-1; i--)
        {
            if((a[i]!=n and i==m-1) or a[i+1]-a[i]>1)
            {
                x=i;
            }
        }
        for(int i=0; i<m; i++)
        {
            if(x==i)
            {
                cout << a[i]+1 << " ";
            }
            else
            {
                cout << a[i] << " ";
            }
        }
    }
    return 0;
}
SubtaskSumTestVerdictTimeMemory
base43/50
1Accepted0/03ms1880 KiB
2Accepted0/03ms2124 KiB
3Accepted2/23ms2376 KiB
4Accepted2/23ms2392 KiB
5Accepted2/22ms2484 KiB
6Accepted2/23ms2868 KiB
7Accepted2/22ms2784 KiB
8Accepted2/23ms2824 KiB
9Accepted2/22ms2912 KiB
10Accepted4/43ms3032 KiB
11Partially correct2/42ms3244 KiB
12Partially correct2/42ms3260 KiB
13Accepted4/43ms3388 KiB
14Accepted4/43ms3600 KiB
15Accepted4/43ms3720 KiB
16Partially correct3/62ms3712 KiB
17Accepted6/63ms3936 KiB