4309 2023. 03. 23 11:09:27 1478 Kígyózó szavak cpp17 Accepted 100/100 8ms 4764 KiB
#include <bits/stdc++.h>

using namespace std;

int main()
{

    int h, n;
    cin>>h>>n;
    vector<vector<long long>> dp(h+1, vector<long long>(27));
    //vector<int> tas

    for(int i=1; i<=26; i++){
        dp[1][i] = 1;
    }

    for(int i=2; i<=h; i++){
        for(int j = 1; j<=26; j++){
            if(j>1) dp[i][j] += dp[i-1][j-1];
            if(j<26) dp[i][j] += dp[i-1][j+1];
            dp[i][j]=min(dp[i][j],1000000000LL);
        }
    }

    for(int i=2; i<=h; i++){
        for(int j=1; j<=26; j++){
            dp[i][j] += dp[i-1][j];
        }
    }

    for(int i=1; i<=n; i++){
        string sol;
        int k, g = h;
        cin>>k;
        int ind = 1;
        while(k-dp[g][ind] > 0){
            k -= dp[g][ind];
            ++ind;
        }
        sol +='a' + ind - 1;
        //cout<<sol<<" ";
        //cout<<k<<": ";
        bool ok = 0;
        while(1){
            k--;
            g--;
            if(k <= 0) {
                break;
            }
            if(ind > 1 && ind < 26){
                if(k-dp[g][ind-1] > 0) {
                    sol+='a'+ind;
                    //cout<<sol<<" ";

                    k -= dp[g][ind-1];
                    //cout<<k<<endl;
                    ind++;
                }
                else{
                    sol+='a'+ind-2;
                    //cout<<sol<<" "<<k<<endl;
                    ind--;
                }
            }
            else if(ind == 1){
                sol+='b';
                ind++;
            }
            else if(ind == 26){
                sol+='y';
                ind--;
            }
        }
        cout<<sol<<endl;
    }
    return 0;
}
Subtask Sum Test Verdict Time Memory
base 100/100
1 Accepted 0/0 3ms 1816 KiB
2 Accepted 0/0 3ms 2048 KiB
3 Accepted 5/5 3ms 2460 KiB
4 Accepted 5/5 3ms 2520 KiB
5 Accepted 5/5 3ms 2752 KiB
6 Accepted 5/5 4ms 2760 KiB
7 Accepted 5/5 4ms 2992 KiB
8 Accepted 5/5 4ms 3248 KiB
9 Accepted 5/5 7ms 3632 KiB
10 Accepted 5/5 6ms 3784 KiB
11 Accepted 5/5 7ms 3912 KiB
12 Accepted 5/5 7ms 4148 KiB
13 Accepted 5/5 8ms 4436 KiB
14 Accepted 6/6 4ms 4348 KiB
15 Accepted 6/6 7ms 4520 KiB
16 Accepted 6/6 4ms 4240 KiB
17 Accepted 6/6 4ms 4224 KiB
18 Accepted 7/7 4ms 4384 KiB
19 Accepted 7/7 8ms 4764 KiB
20 Accepted 7/7 6ms 4348 KiB