QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#543848 | #7876. Cyclic Substrings | afy | AC ✓ | 435ms | 1029908kb | C++20 | 4.1kb | 2024-09-01 22:05:52 | 2024-09-01 22:05:52 |
Judging History
answer
#include <bits/stdc++.h>
#ifdef LOCAL
#include "debug.h"
#else
#define deb(...)
#endif
using namespace std;
#define ll long long
//#define int long long
#define ull unsigned long long
#define pii pair<int, int>
#define db double
#define baoliu(x, y) cout << fixed << setprecision(y) << x
#define endl "\n"
#define alls(x) (x).begin(), (x).end()
#define fs first
#define sec second
#define bug(x) cerr << #x << " = " << x << endl
const int N = 2e5 + 10;
const int M = 1e6 + 10;
const int inf = 0x3f3f3f3f;
const int mod = 998244353;
const double eps = 1e-8;
const double PI = acos(-1.0);
int n, m;
int a[N];
struct PAM {
static constexpr int asz = 12;
struct Node {
int len;
int fail;
int dep; // 以这个节点结尾的回文子串的数量(回文fail树的深度)
int cnt = 0; // 同样的回文结构出现次数
array<int, asz> next;
Node() : len{}, fail{}, dep{}, next{} {}
};
vector<Node> t;
vector<int> idpos;
int last;
string s;
PAM() {
init();
}
void init() {
t.assign(2, Node());
t[0].len = -1; // 0:奇根
last = 1; // 1:偶根
s.clear();
}
int newNode() {
t.emplace_back(); // Node()
return t.size() - 1;
}
bool add(char c, char offset = '0') {
int pos = s.size();
s += c;
int ch = c - offset;
int cur = last, curlen = 0;
while (true) {
curlen = t[cur].len;
if (pos - 1 - curlen >= 0 && s[pos - 1 - curlen] == s[pos])
break;
cur = t[cur].fail;
} // 找到在哪个节点后面建新点
if (t[cur].next[ch]) {
last = t[cur].next[ch];
idpos.push_back(last);
t[last].cnt += 1;
return false;
}
int num = newNode();
last = num;
idpos.push_back(last);
t[num].len = t[cur].len + 2;
t[cur].next[ch] = num;
if (t[num].len == 1) { // 如果为单字符,指向偶根
t[num].fail = 1;
t[num].dep = 1;
t[num].cnt = 1;
return true;
}
while (true) { // 为新节点找fail,从父亲的fail开始找
cur = t[cur].fail;
curlen = t[cur].len;
if (pos - 1 - curlen >= 0 && s[pos - 1 - curlen] == s[pos]) {
t[num].fail = t[cur].next[ch];
break;
}
}
t[num].cnt = 1;
t[num].dep = 1 + t[t[num].fail].dep;
return true;
}
int tot = 0;
void work(string tt) {
for (auto x : tt) add(x);
tot = t.size() - 1;
for (int i = tot; i >= 0; i--) {
int fa = t[i].fail;
t[fa].cnt += t[i].cnt;
}
}
};
PAM pam1;
PAM pam2;
void solve() {
ll ans = 0;
string s;
cin >> n >> s;
pam1.work(s);
pam2.work(s + s);
auto &t1 = pam1.t, t2 = pam2.t;
vector<bool> mp(pam2.tot + 5, 0);
for (int i = 0; i < n; i++) {
int u1 = pam1.idpos[i], u2 = pam2.idpos[i];
if (mp[u2])
continue;
mp[u2] = true;
int cnt = t2[u2].cnt - t1[u1].cnt;
ans += (((ll)cnt * cnt % mod) * (ll)t2[u2].len) % mod;
ans %= mod;
}
for (int i = n; i < 2 * n; i++) {
int u = pam2.idpos[i];
if (mp[u])
continue;
mp[u] = true;
if (t2[u].len > n)
continue;
int cnt = t2[u].cnt;
ans += (((ll)cnt * cnt % mod) *(ll) t2[u].len) % mod;
ans %= mod;
}
cout << ans << endl;
}
signed main() {
cin.tie(0);
ios::sync_with_stdio(false);
#ifdef LOCAL
double starttime = clock();
// freopen("in.txt", "r", stdin);
// freopen("out.txt", "w", stdout);
#endif
int t = 1;
// cin >> t;
while (t--) solve();
#ifdef LOCAL
double endtime = clock();
cerr << "Time Used: " << (double)(endtime - starttime) / CLOCKS_PER_SEC * 1000 << " ms" << endl;
#endif
return 0;
}
这程序好像有点Bug,我给组数据试试?
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 1ms
memory: 3576kb
input:
5 01010
output:
39
result:
ok 1 number(s): "39"
Test #2:
score: 0
Accepted
time: 0ms
memory: 3668kb
input:
8 66776677
output:
192
result:
ok 1 number(s): "192"
Test #3:
score: 0
Accepted
time: 0ms
memory: 3624kb
input:
1 1
output:
1
result:
ok 1 number(s): "1"
Test #4:
score: 0
Accepted
time: 0ms
memory: 3600kb
input:
2 22
output:
12
result:
ok 1 number(s): "12"
Test #5:
score: 0
Accepted
time: 0ms
memory: 3564kb
input:
2 21
output:
2
result:
ok 1 number(s): "2"
Test #6:
score: 0
Accepted
time: 0ms
memory: 3560kb
input:
3 233
output:
10
result:
ok 1 number(s): "10"
Test #7:
score: 0
Accepted
time: 0ms
memory: 3572kb
input:
3 666
output:
54
result:
ok 1 number(s): "54"
Test #8:
score: 0
Accepted
time: 108ms
memory: 358796kb
input:
1000000 3333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333...
output:
496166704
result:
ok 1 number(s): "496166704"
Test #9:
score: 0
Accepted
time: 307ms
memory: 1029908kb
input:
3000000 2222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222...
output:
890701718
result:
ok 1 number(s): "890701718"
Test #10:
score: 0
Accepted
time: 334ms
memory: 575576kb
input:
3000000 9999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999...
output:
224009870
result:
ok 1 number(s): "224009870"
Test #11:
score: 0
Accepted
time: 332ms
memory: 1028636kb
input:
3000000 8989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989...
output:
51985943
result:
ok 1 number(s): "51985943"
Test #12:
score: 0
Accepted
time: 331ms
memory: 1029556kb
input:
3000000 1911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911...
output:
355676465
result:
ok 1 number(s): "355676465"
Test #13:
score: 0
Accepted
time: 376ms
memory: 900140kb
input:
3000000 7777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777...
output:
788510374
result:
ok 1 number(s): "788510374"
Test #14:
score: 0
Accepted
time: 435ms
memory: 929568kb
input:
3000000 5555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555...
output:
691884476
result:
ok 1 number(s): "691884476"
Test #15:
score: 0
Accepted
time: 284ms
memory: 651956kb
input:
3000000 0990990909909909099090990990909909909099090990990909909099099090990990909909099099090990990909909099099090990909909909099099090990909909909099090990990909909909099090990990909909909099090990990909909099099090990990909909099099090990990909909099099090990909909909099099090990909909909099090990...
output:
701050848
result:
ok 1 number(s): "701050848"
Test #16:
score: 0
Accepted
time: 139ms
memory: 259940kb
input:
3000000 2772772727727727277272772772727727727277272772772727727277277272772772727727277277272772772727727277277272772727727727277277272772727727727277272772772727727727277272772772727727727277272772772727727277277272772772727727277277272772772727727277277272772727727727277277272772727727727277272772...
output:
486861605
result:
ok 1 number(s): "486861605"
Test #17:
score: 0
Accepted
time: 336ms
memory: 853456kb
input:
3000000 4554554545545545455454554554545545545455454554554545545455455454554554545545455455454554554545545455455454554545545545455455454554545545545455454554554545545545455454554554545545545455454554554545545455455454554554545545455455454554554545545455455454554545545545455455454554545545545455454554...
output:
450625621
result:
ok 1 number(s): "450625621"
Test #18:
score: 0
Accepted
time: 327ms
memory: 826676kb
input:
3000000 1181811811818118181181181811818118118181181181811818118118181181181811818118118181181811811818118118181181811811818118181181181811811818118181181181811811818118181181181811818118118181181181811818118118181181181811818118118181181811811818118118181181811811818118181181181811811818118181181181...
output:
649551870
result:
ok 1 number(s): "649551870"
Extra Test:
score: 0
Extra Test Passed