QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#379728 | #8571. Palworld | ucup-team1198# | AC ✓ | 809ms | 46932kb | C++20 | 5.4kb | 2024-04-06 18:34:47 | 2024-04-06 18:34:48 |
Judging History
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