QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#181389#6353. Kth Lex Min Min Min Subpalindromesucup-team859AC ✓1403ms98132kbC++176.2kb2023-09-16 18:14:292023-09-16 18:14:29

Judging History

你现在查看的是最新测评结果

  • [2023-09-16 18:14:29]
  • 评测
  • 测评结果:AC
  • 用时:1403ms
  • 内存:98132kb
  • [2023-09-16 18:14:29]
  • 提交

answer

#include <bits/stdc++.h>

#define lsb(x) (x & (-x))

using ull = unsigned long long;
using ll = long long;

using namespace std;

constexpr ll INF = 2e18;

inline ll mul(ll a, ll b) {
    if (INF / b <= a) {
        return INF;
    }
    return a * b;
}

inline ll logpow(ll a, ll b) {
    ll result = 1;
    while (b > 0) {
        if (b & 1) result = mul(a, result);
        b >>= 1;
        a = mul(a, a);
    }
    return result;
}

int main() {
#ifdef HOME
    ifstream cin("input.in");
    ofstream cout("output.out");
#endif
    ios::sync_with_stdio(false);
    cin.tie(0), cout.tie(0);

    int n, m;
    ll k;
    cin >> n >> m >> k;

    if (n == 1) {
        if (m < k) {
            cout << -1;
        } else {
            cout << k;
        }
        return 0;
    }

    if (m == 1) {
        if (k > 1) {
            cout << -1;
            return 0;
        }
        for (int i = 1; i <= n; i++) {
            cout << 1 << " "; 
        }
        return 0;
    }

    if (m == 2) {
        if (n <= 10) {
            vector<string> strings;
            int result = 1e9;

            auto Get = [](string str) {
                    int n = (int)str.size();
                    int cnt = 0;

                    for (int i = 0; i < n; i++) {
                        for (int j = i; j < n; j++) {
                            int l = i, r = j;
                            while (l < r && str[l] == str[r]) {
                                l++, r--;
                            }
                            if (l >= r) {
                                cnt++;
                            }
                        }
                    }
                    return cnt;
                };

            for (int mask = 0; mask < (1 << n); mask++) {
                string str;
                for (int i = 0; i < n; i++) {
                    if (mask & (1 << i)) {
                        str += '2';
                    } else {
                        str += '1';
                    }
                }
                
                int cnt = Get(str);
                if (result > cnt) {
                    strings.clear();
                    result = cnt;
                }
                if (result == cnt) {
                    strings.push_back(str);
                }
            }

            sort(strings.begin(), strings.end());
            strings.resize(unique(strings.begin(), strings.end()) - strings.begin());

            // cerr << strings.size() << "\n";
            // for (auto str : strings) {
            //     cerr << str << " " << Get(str) << "\n";
            // }

            if ((int)strings.size() < k) {
                cout << -1;
            } else {
                for (auto itr : strings[k - 1]) {
                    cout << itr << " ";
                }
            }

            return 0;
        }

        vector<string> strings;
        int result = (int) 1e9;

        // 12212|112212112212|112

        auto Get = [](string str) {
            int n = (int)str.size();
            int cnt = 0;
            for (int i = 0; i < n; i++) {
                for (int j = i; j < n && j < i + 15; j++) {
                    int l = i, r = j;
                    while (l < r && str[l] == str[r]) {
                        l++, r--;
                    }
                    cnt += (l >= r);
                }
            }
            return cnt;
        };

        vector<string> seq = {"121122",
                              "122112",
                              "112122",
                              "112212",
                              "212211",
                              "211221",
                              "221211",
                              "221121"};

        for (auto pref : seq) {
            for (int len_pref = 0; len_pref < pref.size(); len_pref++) {
                string str = pref.substr(pref.size() - len_pref, len_pref);
                while ((int)str.size() < n) {
                    if ((int)str.size() + pref.size() <= n) {
                        str += pref;
                    } else {
                        str += pref.substr(0, n - (int)str.size());
                    }
                }
                assert ((int)str.size() == n);

                int cnt = Get(str);
                if (result > cnt) {
                    result = cnt;
                    strings.clear();
                }
                if (result == cnt) {
                    strings.push_back(str);
                }
            }
        }
        
        int size = (int)strings.size();
        for (int i = 0; i < size; i++) {
            auto s = strings[i];
            reverse(s.begin(), s.end());
            strings.push_back(s);
        }

        sort(strings.begin(), strings.end());
        strings.resize(unique(strings.begin(), strings.end()) - strings.begin());

        // cerr << strings.size() << "\n";
        // for (auto itr : strings) {
        //     cerr << itr << " " << Get(itr) << "\n";
        // }

        if ((int)strings.size() < k) {
            cout << -1;
        } else {
            for (auto itr : strings[k - 1]) {
                cout << itr << " ";
            }
        }
        return 0;
    }

    ll total = mul(m, mul(m - 1, logpow(m - 2, n - 2)));
    if (total < k) {
        cout << -1;
        return 0;
    }

    vector<ll> v(n);

    ll coef0 = mul(m - 1, logpow(m - 2, n - 2));
    ll v0 = (k - 1) / coef0 + 1;
    k -= (v0 - 1) * coef0;
    v[0] = v0;

    if (n > 1) {
        ll coef1 = logpow(m - 2, n - 2);
        ll v1 = (k - 1) / coef1 + 1;
        k -= (v1 - 1) * coef1;
        v[1] = (v1 < v0 ? v1 : v1 + 1);
        
        for (int i = 2; i < n; i++) {
            ll coef = logpow(m - 2, n - i - 1);
            ll vi = (k - 1) / coef + 1;
            k -= (vi - 1) * coef;
            
            if (vi >= min(v[i - 1], v[i - 2])) {
                vi++;
            }
            if (vi >= max(v[i - 1], v[i - 2])) {
                vi++;
            }
            v[i] = vi;
        }
    }

    for (auto itr : v) {
        cout << itr << " ";
    }

    return 0;
}

详细

Test #1:

score: 100
Accepted
time: 0ms
memory: 3568kb

input:

1 1 1

output:

1

result:

ok 1 number(s): "1"

Test #2:

score: 0
Accepted
time: 0ms
memory: 3768kb

input:

2 2 2

output:

2 1 

result:

ok 2 number(s): "2 1"

Test #3:

score: 0
Accepted
time: 0ms
memory: 3628kb

input:

3 3 3

output:

2 1 3 

result:

ok 3 number(s): "2 1 3"

Test #4:

score: 0
Accepted
time: 0ms
memory: 3632kb

input:

9 9 8244353

output:

2 4 1 2 6 8 1 2 7 

result:

ok 9 numbers

Test #5:

score: 0
Accepted
time: 0ms
memory: 3632kb

input:

10 7 998244353

output:

-1

result:

ok 1 number(s): "-1"

Test #6:

score: 0
Accepted
time: 0ms
memory: 3544kb

input:

3 1000 994253860

output:

998 244 353 

result:

ok 3 number(s): "998 244 353"

Test #7:

score: 0
Accepted
time: 0ms
memory: 3628kb

input:

58 4 864691128455135232

output:

4 3 2 4 3 2 4 3 2 4 3 2 4 3 2 4 3 2 4 3 2 4 3 2 4 3 2 4 3 2 4 3 2 4 3 2 4 3 2 4 3 2 4 3 2 4 3 2 4 3 2 4 3 2 4 3 2 4 

result:

ok 58 numbers

Test #8:

score: 0
Accepted
time: 0ms
memory: 3640kb

input:

58 4 864691128455135233

output:

-1

result:

ok 1 number(s): "-1"

Test #9:

score: 0
Accepted
time: 130ms
memory: 11108kb

input:

1000000 1000000 1000000000000000000

output:

1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 ...

result:

ok 1000000 numbers

Test #10:

score: 0
Accepted
time: 136ms
memory: 11208kb

input:

1000000 4 1000000000000000000

output:

1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 ...

result:

ok 1000000 numbers

Test #11:

score: 0
Accepted
time: 0ms
memory: 3544kb

input:

1 1 2

output:

-1

result:

ok 1 number(s): "-1"

Test #12:

score: 0
Accepted
time: 0ms
memory: 3544kb

input:

1 2 2

output:

2

result:

ok 1 number(s): "2"

Test #13:

score: 0
Accepted
time: 0ms
memory: 3764kb

input:

2 2 1

output:

1 2 

result:

ok 2 number(s): "1 2"

Test #14:

score: 0
Accepted
time: 0ms
memory: 3768kb

input:

3 2 4

output:

2 1 1 

result:

ok 3 number(s): "2 1 1"

Test #15:

score: 0
Accepted
time: 0ms
memory: 3604kb

input:

3 2 7

output:

-1

result:

ok 1 number(s): "-1"

Test #16:

score: 0
Accepted
time: 0ms
memory: 3548kb

input:

4 2 10

output:

2 2 1 2 

result:

ok 4 number(s): "2 2 1 2"

Test #17:

score: 0
Accepted
time: 0ms
memory: 3628kb

input:

4 2 3

output:

1 2 1 1 

result:

ok 4 number(s): "1 2 1 1"

Test #18:

score: 0
Accepted
time: 0ms
memory: 3544kb

input:

5 2 7

output:

2 1 1 2 1 

result:

ok 5 number(s): "2 1 1 2 1"

Test #19:

score: 0
Accepted
time: 0ms
memory: 3544kb

input:

5 2 13

output:

-1

result:

ok 1 number(s): "-1"

Test #20:

score: 0
Accepted
time: 0ms
memory: 3612kb

input:

6 2 5

output:

1 2 2 1 1 2 

result:

ok 6 numbers

Test #21:

score: 0
Accepted
time: 1392ms
memory: 97984kb

input:

1000000 2 3

output:

1 2 1 1 2 2 1 2 1 1 2 2 1 2 1 1 2 2 1 2 1 1 2 2 1 2 1 1 2 2 1 2 1 1 2 2 1 2 1 1 2 2 1 2 1 1 2 2 1 2 1 1 2 2 1 2 1 1 2 2 1 2 1 1 2 2 1 2 1 1 2 2 1 2 1 1 2 2 1 2 1 1 2 2 1 2 1 1 2 2 1 2 1 1 2 2 1 2 1 1 2 2 1 2 1 1 2 2 1 2 1 1 2 2 1 2 1 1 2 2 1 2 1 1 2 2 1 2 1 1 2 2 1 2 1 1 2 2 1 2 1 1 2 2 1 2 1 1 2 2 ...

result:

ok 1000000 numbers

Test #22:

score: 0
Accepted
time: 1381ms
memory: 98128kb

input:

1000000 2 5

output:

1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 ...

result:

ok 1000000 numbers

Test #23:

score: 0
Accepted
time: 1403ms
memory: 98132kb

input:

1000000 2 7

output:

2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 2 1 1 2 1 2 ...

result:

ok 1000000 numbers

Test #24:

score: 0
Accepted
time: 1352ms
memory: 98128kb

input:

1000000 2 1000000000000000000

output:

-1

result:

ok 1 number(s): "-1"

Test #25:

score: 0
Accepted
time: 0ms
memory: 3572kb

input:

1 3 2

output:

2

result:

ok 1 number(s): "2"

Test #26:

score: 0
Accepted
time: 0ms
memory: 3612kb

input:

2 3 5

output:

3 1 

result:

ok 2 number(s): "3 1"

Test #27:

score: 0
Accepted
time: 130ms
memory: 11036kb

input:

1000000 3 5

output:

3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 ...

result:

ok 1000000 numbers

Test #28:

score: 0
Accepted
time: 0ms
memory: 3632kb

input:

1000000 3 7

output:

-1

result:

ok 1 number(s): "-1"

Test #29:

score: 0
Accepted
time: 140ms
memory: 11224kb

input:

1000000 4 211106232532991

output:

1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 ...

result:

ok 1000000 numbers

Test #30:

score: 0
Accepted
time: 136ms
memory: 11032kb

input:

1000000 5 1000000000000000000

output:

1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 ...

result:

ok 1000000 numbers

Test #31:

score: 0
Accepted
time: 131ms
memory: 11044kb

input:

1000000 123123 1000000000000000000

output:

1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 ...

result:

ok 1000000 numbers

Test #32:

score: 0
Accepted
time: 0ms
memory: 3576kb

input:

6 1000000 1000000000000000000

output:

1 2 4 9 15 8 

result:

ok 6 numbers

Test #33:

score: 0
Accepted
time: 0ms
memory: 3644kb

input:

4 1000000 1000000000000000000

output:

2 7 15 9 

result:

ok 4 number(s): "2 7 15 9"

Test #34:

score: 0
Accepted
time: 0ms
memory: 3552kb

input:

3 1000000 999997000002000000

output:

1000000 999999 999998 

result:

ok 3 number(s): "1000000 999999 999998"

Test #35:

score: 0
Accepted
time: 0ms
memory: 3768kb

input:

3 1000000 999997000002000001

output:

-1

result:

ok 1 number(s): "-1"