QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#274611 | #7876. Cyclic Substrings | WilliamHu | WA | 69ms | 178704kb | C++14 | 1.3kb | 2023-12-03 18:22:54 | 2023-12-03 18:22:55 |
Judging History
answer
#include<bits/stdc++.h>
using namespace std;
int n, ch[3000010][11], fail[3000010], cnt[3000010], len[3000010], pos = 0, las;
char s[3000010];
long long mod = 998244353;
void init()
{
pos = 1;
len[0] = 0;
fail[0] = 1;
len[1] = -1;
fail[1] = 1;
las = 0;
}
int find(int x, int now)
{
while(s[now - len[x] - 1] != s[now])x = fail[x];
return x;
}
void insert(int x)
{
int p = find(las, x);
if(! ch[p][s[x] - '0'])
{
len[++ pos] = len[p] + 2;
fail[pos] = ch[find(fail[p], x)][s[x] - '0'];
ch[p][s[x] - '0'] = pos;
}
++ cnt[las = ch[p][s[x] - '0']];
}
int tmp[3000010];
signed main() {
cin>>n;
s[0] = '*';
for(int i = 1;i <= n;i ++)
{
cin>>s[i];
s[n + i] = s[i];
}
init();
for(int i = 1;i <= n;i ++)
{
insert(i);
//cout<<i<<endl;
}
for(int i = 1;i <= pos;i ++)tmp[i] = cnt[i];
memset(ch, 0, sizeof(ch));
memset(fail, 0, sizeof(fail));
memset(cnt, 0, sizeof(cnt));
memset(len, 0, sizeof(len));
pos = 0;
long long ans = 0;
init();
for(int i = 1;i <= 2 * n;i ++)
{
insert(i);
//cout<<i<<endl;
}
for(int i = pos;i > 0;i --)
{
cnt[fail[i]] = (cnt[fail[i]]+ cnt[i]) % mod;
tmp[fail[i]] = (tmp[fail[i]]+ tmp[i]) % mod;
if(len[i] <= n)ans = (ans + (cnt[i] - tmp[i]) * (cnt[i] - tmp[i]) % mod * len[i] % mod) % mod;
}
cout<<ans;
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 4ms
memory: 170352kb
input:
5 01010
output:
39
result:
ok 1 number(s): "39"
Test #2:
score: 0
Accepted
time: 4ms
memory: 169832kb
input:
8 66776677
output:
192
result:
ok 1 number(s): "192"
Test #3:
score: 0
Accepted
time: 4ms
memory: 171252kb
input:
1 1
output:
1
result:
ok 1 number(s): "1"
Test #4:
score: 0
Accepted
time: 4ms
memory: 170676kb
input:
2 22
output:
12
result:
ok 1 number(s): "12"
Test #5:
score: 0
Accepted
time: 7ms
memory: 171248kb
input:
2 21
output:
2
result:
ok 1 number(s): "2"
Test #6:
score: 0
Accepted
time: 4ms
memory: 170820kb
input:
3 233
output:
10
result:
ok 1 number(s): "10"
Test #7:
score: 0
Accepted
time: 8ms
memory: 170100kb
input:
3 666
output:
54
result:
ok 1 number(s): "54"
Test #8:
score: -100
Wrong Answer
time: 69ms
memory: 178704kb
input:
1000000 3333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333...
output:
-822846910
result:
wrong answer 1st numbers differ - expected: '496166704', found: '-822846910'