QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#624603 | #7876. Cyclic Substrings | Yurily | RE | 2ms | 13840kb | C++20 | 2.1kb | 2024-10-09 16:13:44 | 2024-10-09 16:13:46 |
Judging History
answer
#include <algorithm>
#include <cstring>
#include <iostream>
#include <string>
using namespace std;
const int maxn = 1e6 + 5;
const long long MOD=998244353;
int tmp,pre,d[maxn];
namespace pam {
int sz, tot, last;
int cnt[maxn], ch[maxn][26], len[maxn], fail[maxn];
char s[maxn];
int node(int l) { // 建立一个新节点,长度为 l
sz++;
memset(ch[sz], 0, sizeof(ch[sz]));
len[sz] = l;
fail[sz] = cnt[sz] = 0;
return sz;
}
void clear() { // 初始化
sz = -1;
last = 0;
s[tot = 0] = '$';
node(0);
node(-1);
fail[0] = 1;
}
int getfail(int x) { // 找后缀回文
while (s[tot - len[x] - 1] != s[tot]) x = fail[x];
return x;
}
void insert(char c) { // 建树
s[++tot] = c;
int now = getfail(last);
if (!ch[now][c - '0']) {
int x = node(len[now] + 2);
fail[x] = ch[getfail(fail[now])][c - '0'];
ch[now][c - '0'] = x;
// cnt[x]=cnt[fail[x]]+1;
}
last = ch[now][c - '0'];
cnt[last]++;
}
void solve1() {
long long ans = 0;
for (int i = sz; i >= 0; i--) {
cnt[fail[i]] += cnt[i];
}
for (int i = 2; i <= sz; i++) {
// cout<<i<<" "<<cnt[i]<<" "<<len[i]<<endl;// 更新答案
d[i]=cnt[i];
}
}
long long solve2() {
long long ans = 0;
for (int i = sz; i >= 0; i--) {
cnt[fail[i]] += cnt[i];
}
for (int i = 2; i <= sz; i++) {
if(len[i]<=tmp)
if(i>pre) // 更新答案
ans = (ans + (long long)cnt[i]*cnt[i]%MOD*len[i]%MOD)%MOD;
else
ans=(ans + (long long)(cnt[i]-d[i])*(cnt[i]-d[i])%MOD*len[i]%MOD)%MOD;
}
return ans;
}
} // namespace pam
string s;
int main() {
cin>>tmp;
cin.tie(nullptr)->sync_with_stdio(false);
pam::clear();
long long ans1=0,ans2=0;
cin >> s;
for (int i = 0; i < s.size(); i++) {
pam::insert(s[i]);
// cout<<pam::last<<" "<<i<<" "<<pam::cnt[pam::last]<<endl;
}
pre=pam::sz;
pam::solve1();
s=s+s;
pam::clear();
for (int i = 0; i < s.size(); i++) {
// cout<<s[i]<<endl;
pam::insert(s[i]);
}
ans2=pam::solve2();
cout<<ans2<<endl;
return 0;
}
详细
Test #1:
score: 100
Accepted
time: 2ms
memory: 11764kb
input:
5 01010
output:
39
result:
ok 1 number(s): "39"
Test #2:
score: 0
Accepted
time: 0ms
memory: 11764kb
input:
8 66776677
output:
192
result:
ok 1 number(s): "192"
Test #3:
score: 0
Accepted
time: 0ms
memory: 11832kb
input:
1 1
output:
1
result:
ok 1 number(s): "1"
Test #4:
score: 0
Accepted
time: 2ms
memory: 11764kb
input:
2 22
output:
12
result:
ok 1 number(s): "12"
Test #5:
score: 0
Accepted
time: 0ms
memory: 11828kb
input:
2 21
output:
2
result:
ok 1 number(s): "2"
Test #6:
score: 0
Accepted
time: 2ms
memory: 11824kb
input:
3 233
output:
10
result:
ok 1 number(s): "10"
Test #7:
score: 0
Accepted
time: 2ms
memory: 13840kb
input:
3 666
output:
54
result:
ok 1 number(s): "54"
Test #8:
score: -100
Runtime Error
input:
1000000 3333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333...