QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#537576#7876. Cyclic Substringsucup-team956TL 768ms276612kbC++203.2kb2024-08-30 16:06:092024-08-30 16:06:09

Judging History

你现在查看的是最新测评结果

  • [2024-08-30 16:06:09]
  • 评测
  • 测评结果:TL
  • 用时:768ms
  • 内存:276612kb
  • [2024-08-30 16:06:09]
  • 提交

answer

#include <bits/stdc++.h>

using namespace std;

#define SZ(x) ((int)((x).size()))
#define lb(x) ((x) & (-(x)))
#define bp(x) __builtin_popcount(x)
#define bpll(x) __builtin_popcountll(x)
#define mkp make_pair
#define pb push_back
#define fi first
#define se second
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> pii;
typedef pair<ll, int> pli;
typedef pair<ll, ll> pll;
typedef pair<double, int> pdi;

const int Mod = 998244353;

struct PAM {
    string s;
    int lst, tot, n;
    vector<int> fa, len, sz, cnt, pos;
    vector<array<int, 10>> ch;
    /*
    - len[i] :  the length of the longest palindrome string end with i
    - sz[i]  :  the number of the palindrome string end with i
    - cnt[i] :  the occurrence number of the longest palindrome string end with i
    - fa[i]  :  i 的回文后缀链接
    - pos[i] :  下标为 i 的字符在回文树上对应的结点
    - st     :  fa 的倍增数组
    */
    PAM(){}
    PAM(string s): s(s), n(s.length() + 2) {    // String s (Index from 0)
        fa.resize(n); len = sz = cnt = pos = fa;
        ch = vector(n, array<int, 10>());
        len[0] = 0; len[1] = -1;
        fa[0] = 1; fa[1] = 0;
        lst = 0; tot = 1;
    }
    int get_fail(int x, int now) {
        while (s[now - len[x] - 1] != s[now]) {
            x = fa[x];
        }        
        return x;
    }
    void insert(int c, int now) {
        int p = get_fail(lst, now);
        if (!ch[p][c]) {
            len[++tot] = len[p] + 2;
            int tmp = get_fail(fa[p], now);
            fa[tot] = ch[tmp][c];
            sz[tot] = sz[fa[tot]] + 1;
            ch[p][c] = tot;
        }
        lst = ch[p][c];
        cnt[lst] += 1;
        pos[now] = ch[p][c];
    }
    void init() {
        for (int i = 0; i < SZ(s); i++) {
            insert(s[i] - '0', i);
        }
        for (int i = tot; i >= 0; i--) {
            cnt[fa[i]] += cnt[i];
        }
    }
};

void solve() {
    int n;
    string s;
    cin >> n >> s;

    PAM pam1(s);
    PAM pam2(s + s);
    pam1.init();
    pam2.init();

    ll ans = 0;
    map<int, bool> mp;
    for (int i = 0; i < n; i++) {
        int u = pam2.pos[i];
        int len = pam2.len[u];
        int cnt = pam2.cnt[u];
        if (mp[u]) {
            continue;
        }
        mp[u] = true;

        // cout << "i = " << i << '\n';
        // cout << "cnt = " << cnt << '\n';

        int c = cnt - pam1.cnt[pam1.pos[i]];

        // cout << "c = " << c << '\n';
        // cout << "len = " << len << '\n';
        // cout << '\n';

        (ans += 1ll * c * c % Mod * len) %= Mod;
    }

    for (int i = n; i < 2 * n; i++) {
        int u = pam2.pos[i];
        int len = pam2.len[u];
        int cnt = pam2.cnt[u];
        if (mp[u]) {
            continue;
        }
        mp[u] = true;

        if (len > n) {
            continue;
        }

        int c = cnt;
        (ans += 1ll * c * c % Mod * len) %= Mod;
    }

    cout << ans << '\n';
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0), cout.tie(0);
    int T = 1;
    while (T--) solve();
    return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 0ms
memory: 3512kb

input:

5
01010

output:

39

result:

ok 1 number(s): "39"

Test #2:

score: 0
Accepted
time: 0ms
memory: 3632kb

input:

8
66776677

output:

192

result:

ok 1 number(s): "192"

Test #3:

score: 0
Accepted
time: 0ms
memory: 3632kb

input:

1
1

output:

1

result:

ok 1 number(s): "1"

Test #4:

score: 0
Accepted
time: 0ms
memory: 3860kb

input:

2
22

output:

12

result:

ok 1 number(s): "12"

Test #5:

score: 0
Accepted
time: 0ms
memory: 3620kb

input:

2
21

output:

2

result:

ok 1 number(s): "2"

Test #6:

score: 0
Accepted
time: 0ms
memory: 3628kb

input:

3
233

output:

10

result:

ok 1 number(s): "10"

Test #7:

score: 0
Accepted
time: 0ms
memory: 3556kb

input:

3
666

output:

54

result:

ok 1 number(s): "54"

Test #8:

score: 0
Accepted
time: 768ms
memory: 276612kb

input:

1000000
3333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333...

output:

496166704

result:

ok 1 number(s): "496166704"

Test #9:

score: -100
Time Limit Exceeded

input:

3000000
2222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222...

output:


result: