QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#379728#8571. Palworlducup-team1198#AC ✓809ms46932kbC++205.4kb2024-04-06 18:34:472024-04-06 18:34:48

Judging History

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

  • [2024-04-06 18:34:48]
  • 评测
  • 测评结果:AC
  • 用时:809ms
  • 内存:46932kb
  • [2024-04-06 18:34:47]
  • 提交

answer

#include <map>
#include <set>
#include <array>
#include <cmath>
#include <deque>
#include <bitset>
#include <random>
#include <string>
#include <vector>
#include <cassert>
#include <complex>
#include <iomanip>
#include <iostream>
#include <algorithm>
#include <unordered_map>
#include <unordered_set>

using namespace std;

const int MAXN = 400200;
const int K = 19;

int lg[MAXN];

struct SuffMass {
    vector<int> table[K];
    vector<int> order;
    vector<int> lcp;
    vector<int> id;
    SuffMass(string s) {
        s += '$';
        int n = s.size();
        order.resize(n);
        vector<int> classes(n);
        vector<int> new_order(n);
        vector<int> new_classes(n);
        vector<int> cnt(n);
        iota(order.begin(), order.end(), 0);
        sort(order.begin(), order.end(), [&](int a, int b) {
            return s[a] < s[b];
        });
        classes[order[0]] = 0;
        for (int i = 1; i < n; ++i)
            classes[order[i]] = classes[order[i - 1]] + (s[order[i]] != s[order[i - 1]]);
        auto safe = [&](int x) {
            if (x >= n)
                return x - n;
            return x;
        };
        for (int l = 1; l < n; l *= 2) {
            fill(cnt.begin(), cnt.end(), 0);
            for (int i = 0; i < n; ++i)
                ++cnt[classes[i]];
            for (int i = 1; i < n; ++i)
                cnt[i] += cnt[i - 1];
            for (int i = n - 1; i >= 0; --i) {
                int j = order[i] - l;
                if (j < 0)
                    j += n;
                --cnt[classes[j]];
                new_order[cnt[classes[j]]] = j;
            }
            new_classes[new_order[0]] = 0;
            for (int i = 1; i < n; ++i) {
                new_classes[new_order[i]] = new_classes[new_order[i - 1]] + 
                    (make_pair(classes[new_order[i]], classes[safe(new_order[i] + l)]) != 
                        make_pair(classes[new_order[i - 1]], classes[safe(new_order[i - 1] + l)]));
            }
            swap(classes, new_classes);
            swap(order, new_order);
        }
        lcp.resize(n);
        id.resize(n);
        for (int i = 0; i < n; ++i)
            id[order[i]] = i;
        int tmp = 0;
        for (int i = 0; i < n; ++i) {
            if (id[i] == n - 1) {
                tmp = 0;
                continue; 
            }
            int j = order[id[i] + 1];
            while (s[i + tmp] == s[j + tmp])
                ++tmp;
            lcp[id[i]] = tmp;
            tmp = max(tmp - 1, 0);
        }
        for (int j = 0; j < K; ++j)
            table[j].resize(n);
        table[0] = lcp;
        for (int j = 1; j < K; ++j) {
            for (int i = 0; i + (1 << j) <= n; ++i)
                table[j][i] = min(table[j - 1][i], table[j - 1][i + (1 << (j - 1))]);
        }
    }
    int get_lcp(int i, int j) {
        i = id[i];
        j = id[j];
        if (i == j)
            return -1;
        if (i > j)
            swap(i, j);
        int k = lg[j - i];
        return min(table[k][i], table[k][j - (1 << k)]);
    }
};

void solve() {
    int n, k;
    cin >> n >> k;
    string s;
    cin >> s;
    /// reverse(s.begin(), s.end());
    string rs = s;
    reverse(rs.begin(), rs.end());
    string t = s + " " + rs;

    SuffMass arr(t);

    int ans = 0;
    for (int i = 0; i < n; ++i) {
        for (int j = 0; j <= k && i + j <= n; ++j) {
            int l3 = j;
            int l1 = 0;
            if (i > 0 && i + j < n) {
                l1 = arr.get_lcp(i + j, 2 * n + 1 - i);
            }
            ans = max(ans, 2 * l1 + l3 + k);
        }
    }

    auto is_pal = [&](int l, int r) {
        if (l < 0 || r > n) return false;
        if (l > r) return true;
        int len = arr.get_lcp(l, 2 * n + 1 - r);
        return len >= r - l;
    };

    for (int sum = 0; sum < 2 * n; ++sum) {
        int lft = -1, rgh = n;
        while (rgh - lft > 1) {
            int ml = (lft + rgh) / 2;
            int mr = sum - ml;
            if (is_pal(ml, mr)) {
                rgh = ml;
            } else {
                lft = ml;
            }
        }
        int ml = rgh, mr = sum - rgh;
        if (ml >= mr) continue;
        /// cerr << ml << " " << mr << endl;
        for (int j = 0; j <= k; ++j) {
            /// cerr << "j: " << j << endl;
            int pos1 = 2 * n + 1 - ml;
            int pos2 = mr + j;
            int l1 = 0;
            if (pos1 >= n + 1 && pos1 <= 2 * n && pos2 >= 0 && pos2 < n) {
                l1 = arr.get_lcp(pos1, pos2);
            }
            if (j + mr <= n) {
                ans = max(ans, 2 * l1 + 2 * j + mr - ml);
            }
            /// cerr << ans << endl;

            pos1 = 2 * n + 1 - ml + j;
            pos2 = mr;
            l1 = 0;
            if (pos1 >= n + 1 && pos1 <= 2 * n && pos2 >= 0 && pos2 < n) {
                l1 = arr.get_lcp(pos1, pos2);
            }
            if (j <= ml) {
                ans = max(ans, 2 * l1 + 2 * j + mr - ml);
            }
            /// cerr << ans << endl;
        }
        /// cerr << ans << "\n";
    }
    cout << ans << "\n";
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    
    lg[0] = -1;
    for (int i = 1; i < MAXN; ++i)
        lg[i] = lg[i / 2] + 1;

    int tst;
    cin >> tst;
    while (tst--) {
        solve();
    }
    
    return 0;
}

这程序好像有点Bug,我给组数据试试?

详细

Test #1:

score: 100
Accepted
time: 1ms
memory: 5160kb

input:

4
1 3
a
4 1
icpc
4 2
icpc
8 4
icecream

output:

4
5
5
11

result:

ok 4 number(s): "4 5 5 11"

Test #2:

score: 0
Accepted
time: 590ms
memory: 46904kb

input:

1
200000 66
jsmwjmibgkjvdscqllsxpaxiycmpauhnzschbivtqbjfrxrqvhvfbqecozjewqqpwdfbeqppjkgxhbnniopkptkygspcdswhwadfwhnzovvpshgcdukrupeztkpxwhmctaquqbxtidzbbxsyuaeikuldaoeudletrsmqptaejibkymsjhmwykqsjdvvdaqwelrcpxwrwhuvodipjniowfofbjktkdezwqqbvwsppsmpilntmdmlxgkaxymnugmmcsljkjzjuudnllydwdwwanadynsoiolso...

output:

141

result:

ok 1 number(s): "141"

Test #3:

score: 0
Accepted
time: 730ms
memory: 46712kb

input:

1
200000 100
qhiaajzxinenucrnfffumuhnovpuwcnojbhsktztapgyivmfqrlntwazwnfetwqieckxcnkskpidltiydfoveaucckximydxxfbiwbdufmbhywqkflyqxbijakadqkzftlciccbpnldsqhtjxuxnvkusvizavuyfhdroiuominebadhfqzrpjnzpgyvkejfwmueiltyeqpvwrkanqknyacqganbszktocfuvvsfrboaennwpaabfdnaurvvurysrijnfaonesihbhrrvbvyhpbremsuhhbc...

output:

209

result:

ok 1 number(s): "209"

Test #4:

score: 0
Accepted
time: 70ms
memory: 5132kb

input:

10000
21 7
fhcjhdjcfbfbdeeheibhg
18 8
hccgjfaadefhjhcijc
15 7
ahefiiheaigjiid
15 3
fgjjebidbfgbdij
15 5
jccebbgjbhifjhb
20 2
hcbddhibgfccjjcfebcc
19 1
ehbdgiicchijebidbcd
20 8
gbcfghefhbjggecdhcbj
17 2
jjaeaccjbcfiehfdd
23 4
iafjedfbebbhcfgfjbehdaf
22 1
jgiiiaacehcbaiehfggcfi
17 2
jefbbfigfhbhfcici
...

output:

17
21
17
11
13
9
6
18
9
11
6
9
22
8
21
23
5
23
7
16
9
23
25
11
17
12
9
5
20
5
23
12
7
13
16
17
21
21
21
12
16
21
21
8
9
11
21
15
5
24
13
7
21
13
11
8
7
19
25
7
8
12
13
22
15
14
7
15
7
20
15
13
11
15
5
11
23
17
17
19
15
7
19
17
25
13
23
10
17
21
9
21
7
23
21
13
16
7
14
10
13
10
9
11
7
21
15
19
11
23
...

result:

ok 10000 numbers

Test #5:

score: 0
Accepted
time: 200ms
memory: 46796kb

input:

1
200000 100
aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa...

output:

200100

result:

ok 1 number(s): "200100"

Test #6:

score: 0
Accepted
time: 551ms
memory: 46888kb

input:

1
200000 73
jgqahsycxmrrwvszqigkxmxxckratkgiyyqbxgjzaqkufvkgmepixhjdugpsntlsgtiedueukiytrytrcvlkymwibsfdhytbffmtaytnfczbavrrbdnpegwxubyeoopzopqhrwukohojtaizsureefffglwjfawvpqqnjteveuscyelgjiukskvymqejzwlnrjgvnpwnttzahoupruedryqeyahxgltnibxhbpxmhxxnmrebsgbjwsfpipezkekaqptnwtuanbfyfwyotwkrvlamnuvhunty...

output:

175682

result:

ok 1 number(s): "175682"

Test #7:

score: 0
Accepted
time: 49ms
memory: 5420kb

input:

10000
12 1
jelwdrimpvtq
19 8
qcogumialfqhaahfhtb
11 12
tbnfsiqbibv
5 4
hdehb
4 1
ivxi
7 8
nonrwqa
14 16
obvpfqjekvmnyq
4 5
fzhi
18 17
gploorpjnrrhcrgxlp
2 3
pf
7 9
uiliuyr
1 3
v
3 3
bgj
9 4
tektrwtmk
10 4
vvkvbzpbye
20 5
jqzqpqiwkssksbilayvj
2 4
kl
5 2
vptsk
20 9
kuiwsxhfmirnrhzuysqd
8 8
aairrgjl
15...

output:

3
20
23
9
5
15
30
9
35
5
16
4
6
11
11
14
6
5
21
16
13
11
2
23
13
3
30
5
23
15
21
16
18
21
3
3
11
23
13
13
8
14
4
22
6
5
13
3
18
27
9
3
24
32
28
3
13
3
31
18
3
7
8
3
15
20
7
13
19
15
3
7
3
15
12
4
7
29
5
3
11
36
23
5
26
22
11
8
12
30
7
9
25
11
3
3
9
12
15
22
10
2
17
15
21
7
31
33
7
35
18
31
11
10
10
...

result:

ok 10000 numbers

Test #8:

score: 0
Accepted
time: 56ms
memory: 5372kb

input:

20480
1 1
d
1 2
d
1 1
z
1 2
z
2 1
pp
2 2
pp
2 3
pp
2 1
zp
2 2
zp
2 3
zp
2 1
pz
2 2
pz
2 3
pz
2 1
zz
2 2
zz
2 3
zz
3 1
www
3 2
www
3 3
www
3 4
www
3 1
mww
3 2
mww
3 3
mww
3 4
mww
3 1
wmw
3 2
wmw
3 3
wmw
3 4
wmw
3 1
mmw
3 2
mmw
3 3
mmw
3 4
mmw
3 1
wwm
3 2
wwm
3 3
wwm
3 4
wwm
3 1
mwm
3 2
mwm
3 3
mwm
3 ...

output:

2
3
2
3
3
4
5
3
4
5
3
4
5
3
4
5
4
5
6
7
4
5
6
7
4
5
6
7
4
5
6
7
4
5
6
7
4
5
6
7
4
5
6
7
4
5
6
7
5
6
7
8
9
5
6
7
8
9
5
6
7
8
9
4
6
7
8
9
5
6
7
8
9
5
5
7
8
9
5
6
7
8
9
5
6
7
8
9
5
6
7
8
9
5
6
7
8
9
5
5
7
8
9
5
6
7
8
9
4
6
7
8
9
5
6
7
8
9
5
6
7
8
9
5
6
7
8
9
6
7
8
9
10
11
6
7
8
9
10
11
6
7
8
9
10
11
5
...

result:

ok 20480 numbers

Test #9:

score: 0
Accepted
time: 164ms
memory: 5204kb

input:

2000
88 14
hbwgbmjtqmgdpbgsdkjndhwjrhluountepcjpecnaogiamrzmnomjcgzzvdrzsfjspyxpooxhjtzlqtzvpeydwap
104 88
osirzoiablktpnobqvhckmttlpbgaeakzhemoxgteqbkmfzaerhqfwkoqpsiqqymmbmufxhvfhovkhdduzybzzhqnvbmswqepkqzrkox
82 25
rumtzeygumtvcmyiowwmxkjekajxpubgazyxabibdirnbsmzsjawtaksrydmuroohkrrajlqqyxogtvuul...

output:

31
177
53
71
113
9
153
107
97
51
149
153
121
193
111
38
129
175
99
183
91
183
115
182
203
41
193
155
159
11
149
145
111
19
76
152
67
107
79
15
49
175
149
63
115
59
45
39
138
37
102
14
23
157
75
175
31
33
65
95
171
61
59
34
73
107
195
185
113
11
65
16
128
184
59
139
23
131
115
135
143
166
42
34
49
29...

result:

ok 2000 numbers

Test #10:

score: 0
Accepted
time: 783ms
memory: 46892kb

input:

1
200000 100
cbdadabcbdaaacbedaccdeeeaabececdbcdebcedaeacddcdeaacaccedeadecacaabbdbdbbbbbeaecddbcaaeeaddcabcbebecebcddeccaeceacdebecaddcaddaaadaecabcbbbababedbbccbaceadaeccbbceaeacdccaaaabcbdaebcbcacaaeebebdddeeeeddcaaadbbbccacecacccebeeaabaecbdeccccbdbeedcddececabeaadeaadadcccbeebaecbacbccdbaceebdb...

output:

222

result:

ok 1 number(s): "222"

Test #11:

score: 0
Accepted
time: 809ms
memory: 46932kb

input:

1
200000 100
bcacdaadcdbcaeeebacabdaadebbceabaacdebddeacdebaabdccacdcacadbceceecdccebdebdbccdadbebbaebcabeddeabbdebdadcdedaadedbccbbdbcccdceddeadccdaadeecbebddcddccccbebccbbadaaadeeabaedecdacbeaabadcdeaeddadecdeddbeedeccaabaeebebdbecceaecbdcebceaadeebeaaceadebedcbdbdbcdbbecabddecabcdbccbbbecdaceabdd...

output:

220

result:

ok 1 number(s): "220"

Extra Test:

score: 0
Extra Test Passed