QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#661360#7876. Cyclic SubstringsCalculatelove#ML 112ms500464kbC++141.7kb2024-10-20 15:55:162024-10-20 15:55:16

Judging History

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

  • [2024-10-20 15:55:16]
  • 评测
  • 测评结果:ML
  • 用时:112ms
  • 内存:500464kb
  • [2024-10-20 15:55:16]
  • 提交

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

result: