QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#131315 | #5065. Beautiful String | _skb_ | TL | 2436ms | 38956kb | C++14 | 3.0kb | 2023-07-26 22:00:32 | 2023-07-26 22:00:35 |
Judging History
answer
#include<bits/stdc++.h>
using namespace std;
using i64 = long long;
using u64 = unsigned long long;
struct debug {
#define contPrint { *this << "["; \
int f = 0; for(auto it : x) { *this << (f?", ":""); *this << it; f = 1;} \
*this << "]"; return *this;}
~debug(){cerr << endl;}
template<class c> debug& operator<<(c x) {cerr << x; return *this;}
template<class c, class d>
debug& operator<<(pair<c, d> x) {*this << "(" << x.first << ", " << x.second << ")";
return *this;}
template<class c> debug& operator<<(vector<c> x) contPrint;
#undef contPrint
};
#define dbg(x) "[" << #x << ": " << x << "] "
#define Wa() cerr << "[LINE: " << __LINE__ << "] -> "; debug() <<
#define FASTIO ios_base::sync_with_stdio(false); cin.tie(NULL);
vector<int> z_function(string s) {
int n = s.size();
vector<int> z(n);
int l = 0, r = 0;
for(int i = 1; i < n; i++) {
if(i < r) {
z[i] = min(r - i, z[i - l]);
}
while(i + z[i] < n && s[z[i]] == s[i + z[i]]) {
z[i]++;
}
if(i + z[i] > r) {
l = i;
r = i + z[i];
}
}
return z;
}
char ch[5005];
void solve()
{
scanf("%s", ch);
int len = strlen(ch);
vector<int> left[len];
for(int i = 0; i < len; i++) {
auto z = z_function(string(ch + i, ch + len));
for(int j = i+1; j < len; j++) {
if(z[j - i] >= j - i) {
left[j].push_back(j - i);
}
}
}
for(int i = 0; i < len; i++) {
sort(left[i].begin(), left[i].end());
}
vector<int> bit(len+1);
int bit_tot = 0;
auto update = [&] (int pos, int val) {
if(pos == 0 || pos > len) return;
bit_tot += val;
while(pos <= len) {
bit[pos] += val;
pos += pos & -pos;
}
};
auto get = [&] (int pos) {
pos--;
int ret = 0;
if(pos <= 0) return 0;
while(pos) {
ret += bit[pos];
pos -= pos & -pos;
}
return bit_tot - ret;
};
i64 ans = 0;
for(int l = 0; l < len; l++) {
auto z = z_function(string(ch + l, ch + len));
bit.assign(len+1, 0);
bit_tot = 0;
for(int i = 3; i < len - l; i++) {
update(z[i], 1);
}
int left_idx = 0;
for(int r = l + 1; r < len; r++) {
int cur_len = r - l + 1;
while(left_idx < left[l].size() && left[l][left_idx] < cur_len) {
left_idx++;
}
int right_cnt = get(cur_len);
ans += left_idx * right_cnt;
if(r - l + 2 < (int)z.size()) {
update(z[r - l + 2], -1);
}
}
}
printf("%lld\n", ans);
}
int main()
{
int T;
scanf("%d", &T);
while(T--) {
solve();
}
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 1ms
memory: 3700kb
input:
2 114514 0000000
output:
1 3
result:
ok 2 number(s): "1 3"
Test #2:
score: 0
Accepted
time: 245ms
memory: 3916kb
input:
11 79380 2483905227 37902399703986576270 39991723655814713848046032694137883815354365954917 5541883522667841718787744915558298207485830622522715211018760095628594390563630840464643840093757297 56530485554219245862513973750218715585679416120445975390556326891488719311495909340757506478400624741858999...
output:
0 0 0 2 4 20 119 113 1086 2128 15166
result:
ok 11 numbers
Test #3:
score: 0
Accepted
time: 28ms
memory: 3696kb
input:
50 11111111111111111111111111111111111111121111111111111111111111111111111111111112111111111121111111111211111121211121111111111111111111111111111211121111111111111111111111111111111111111111111111111112 111111111111111111111111111111111111111111111121111121111111111111111111111111111111111111111111...
output:
779344 799116 716078 723215 1197647 403357 652134 625671 414294 942493 390998 793444 612061 507395 473508 836065 461623 374925 539333 592574 676408 610940 463761 490048 995917 595830 424894 332669 596834 655095 521489 1032050 697420 752056 406316 360973 1180943 948628 478572 1026603 711224 429752 49...
result:
ok 50 numbers
Test #4:
score: 0
Accepted
time: 31ms
memory: 3780kb
input:
50 11211121122222111222111111222112111221112111121112221111111121211111212211212122112212221221112112221221112211211112121222112221211122112211112111112112211121222111222212211121111111112111112121111122 112112121211212111212221221222211211121212221111112122121211112221221121121112111221211112122121...
output:
7499 6375 7041 7889 6622 6804 8695 8795 7018 8387 8910 8019 8223 8820 7324 7144 8035 9941 7073 7373 7427 7280 6946 8204 7931 6769 7050 9268 7682 8232 7797 7356 7012 8967 7469 6869 11728 6562 7604 8840 7885 8658 7006 8156 10694 6716 6121 7499 7456 7981
result:
ok 50 numbers
Test #5:
score: 0
Accepted
time: 824ms
memory: 5216kb
input:
15 111111111111111111111111111111111111111111111111111111111111121111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111211111111111111111111111111111111111111111111111...
output:
6611556286 8447635347 4351265656 8244172287 6847075843 5064323828 5818821992 5187397748 6202849391 7100699750 8826693258 9304467838 9691754783 12524687288 10378182916
result:
ok 15 numbers
Test #6:
score: 0
Accepted
time: 899ms
memory: 3980kb
input:
15 111111111111111111111111111111111111111111111111111111121112111111111111111111111111211111112111111111111111111111111112112111111111211111111111111111121211111111111111111112121111211112112111111112111111111111111112111111111111111111111111111111111111111211111211211111111111111112211111111111111...
output:
99283290 121730268 95231372 139100190 109487920 93015077 138212377 180336129 94959502 88117283 81796472 100172151 133716692 92198870 119549081
result:
ok 15 numbers
Test #7:
score: 0
Accepted
time: 2296ms
memory: 13032kb
input:
6 1111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111...
output:
385321895637 278064026340 462204013200 622961805899 319151572118 194136546751
result:
ok 6 numbers
Test #8:
score: 0
Accepted
time: 2411ms
memory: 4784kb
input:
6 1111211111111111111111111112111111111111111111111111111111111111111111111111111111111111111121111111112111111111111111111111211111111111111111111111111111111111111111111111111111112111111111111111111111111111111111111111111111111111111111111111111111111111111211111111112111111111111111111111111111...
output:
4279296283 6481388714 4807510535 5043682133 5816318027 4092854445
result:
ok 6 numbers
Test #9:
score: 0
Accepted
time: 2343ms
memory: 38956kb
input:
6 1111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111...
output:
4333336111111 4333336111111 4333336111111 4333336111111 4333336111111 4333336111111
result:
ok 6 numbers
Test #10:
score: 0
Accepted
time: 2421ms
memory: 4780kb
input:
6 1111111111111111111111111111111111121111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111112111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111...
output:
4827246439 5292779668 4777240971 4935748521 5102858676 4471955490
result:
ok 6 numbers
Test #11:
score: 0
Accepted
time: 2436ms
memory: 4772kb
input:
6 1111111111112111111112111111111111111211111111111111111111111111111111111111111111111111111111111111111111111111111111111111111211111111111111111111111111111111111111112111111111111111111111111111111111111111111111121111111111111111111111111111111111111111111111111111111121111111111111111111111111...
output:
5823633843 3699828594 5341000227 4377298465 5641079075 4447500985
result:
ok 6 numbers
Test #12:
score: 0
Accepted
time: 2414ms
memory: 4888kb
input:
6 1111111111111111111111111111111121121111111111111111112111111111111111111111111111111111111111111111111211111111111111111111121111111111111111111111111111111111111111111111111111111121111111111211111111111111111111111111121111112111111111111111111111111111111111111111111111111112111111111111111111...
output:
4267388536 4492067906 3738167207 4136464174 5508219259 4485317348
result:
ok 6 numbers
Test #13:
score: -100
Time Limit Exceeded
input:
6 1101000101000011111101010011000111011101101000100101001011110000000010100000011011010010110000100000100100101110000010110010101011011100101000111010101110101010100011011111111111101001110101011000000011100011110001100001100101101111101011101110001000100110011001101100000010111111000010010110100011...