QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#661360 | #7876. Cyclic Substrings | Calculatelove# | ML | 112ms | 500464kb | C++14 | 1.7kb | 2024-10-20 15:55:16 | 2024-10-20 15:55:16 |
Judging History
answer
#include <bits/stdc++.h>
const int N = 3001000;
const int mod = 998244353;
int n;
char str[N];
namespace PAM {
int cur_len, str[N * 2];
int nClock, Last;
struct node {
int trans[26];
int link, len;
} t[N * 2];
int cnt[N * 2];
std::vector<int> son[N * 2];
void init() {
t[0].len = 0, t[0].link = 1;
t[1].len = -1;
str[cur_len = 0] = -1;
nClock = 1, Last = 1;
}
int find(int p) {
while (str[cur_len - t[p].len - 1] != str[cur_len]) p = t[p].link;
return p;
}
void extend(int c, int type) {
str[++ cur_len] = c;
int p = find(Last);
if (!t[p].trans[c]) {
int np = ++ nClock;
t[np].len = t[p].len + 2;
t[np].link = t[find(t[p].link)].trans[c];
t[p].trans[c] = np;
}
Last = t[p].trans[c];
if (type) cnt[Last] ++;
}
void build_tree() {
son[1].push_back(0);
for (int i = 2; i <= nClock; i ++) son[t[i].link].push_back(i);
}
void dfs(int u) {
for (int v : son[u]) {
dfs(v);
cnt[u] += cnt[v];
}
}
int calc() {
int ans = 0;
for (int i = 2; i <= nClock; i ++)
if (t[i].len <= n)
ans = (ans + 1ll * cnt[i] * cnt[i] % mod * t[i].len) % mod;
return ans;
}
}
int main() {
scanf("%d", &n);
scanf("%s", str + 1);
PAM::init();
for (int i = 1; i <= n; i ++) PAM::extend(str[i] - '0', 0);
for (int i = 1; i <= n; i ++) PAM::extend(str[i] - '0', 1);
PAM::build_tree();
PAM::dfs(1);
printf("%d\n", PAM::calc());
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 16ms
memory: 152500kb
input:
5 01010
output:
39
result:
ok 1 number(s): "39"
Test #2:
score: 0
Accepted
time: 8ms
memory: 151424kb
input:
8 66776677
output:
192
result:
ok 1 number(s): "192"
Test #3:
score: 0
Accepted
time: 30ms
memory: 152692kb
input:
1 1
output:
1
result:
ok 1 number(s): "1"
Test #4:
score: 0
Accepted
time: 12ms
memory: 152552kb
input:
2 22
output:
12
result:
ok 1 number(s): "12"
Test #5:
score: 0
Accepted
time: 12ms
memory: 151712kb
input:
2 21
output:
2
result:
ok 1 number(s): "2"
Test #6:
score: 0
Accepted
time: 16ms
memory: 152776kb
input:
3 233
output:
10
result:
ok 1 number(s): "10"
Test #7:
score: 0
Accepted
time: 14ms
memory: 152240kb
input:
3 666
output:
54
result:
ok 1 number(s): "54"
Test #8:
score: 0
Accepted
time: 112ms
memory: 500464kb
input:
1000000 3333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333...
output:
496166704
result:
ok 1 number(s): "496166704"
Test #9:
score: -100
Memory Limit Exceeded
input:
3000000 2222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222...
output:
890701718