QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#639034#7876. Cyclic Substringskinoko777WA 20ms122308kbC++233.0kb2024-10-13 17:43:192024-10-13 17:43:21

Judging History

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

  • [2024-10-13 17:43:21]
  • 评测
  • 测评结果:WA
  • 用时:20ms
  • 内存:122308kb
  • [2024-10-13 17:43:19]
  • 提交

answer

#include <algorithm>
#include <cstring>
#include <iostream>
#include <string>
using namespace std;
constexpr int MAXN = 3000000 + 5, mod = 998244353;
int n;
namespace pam {
int sz, tot, last, sz1;
int cnt[MAXN], ch[MAXN][10], len[MAXN], fail[MAXN];
char s[MAXN];
int cnt1[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;
  }
  last = ch[now][c - '0'];
  cnt[last]++;
}

void work1()
{
    for(int i = 2; i <= sz; i ++) cnt1[i] = cnt[i];
    for(int i = sz; i >= 2; i --) cnt1[fail[i]] += cnt1[i];
    sz1 = sz;
}

int work(int op)
{
    // int ans = 0;
    // for(int i = sz; i >= 0; i --)
    // {
    //     cnt[fail[i]] += cnt[i];
    // }
    int ans = 0;
    for(int i = sz; i >= 2; i --)
    {
        cnt[fail[i]] += cnt[i];
        int c = cnt[i];
        if(i <= sz1) c -= cnt1[i];
        if(len[i] <= n) 
            ans = (ans + c * c % mod * len[i] % mod) % mod;
    }
    ans = (ans % mod + mod) % mod;
    // for(int i = 1; i <= sz; i ++)
    // {
    //     if(len[i] == -1) continue;
    //     if(op == 2 && len[i] >= tot / 2) continue;
    //     cout << cnt[i] << ' ' << len[i] << endl;
    //     ans += 1ll * cnt[i] * cnt[i] % mod * len[i] % mod;
    //     ans %= mod;
    // }
    return ans;
}
}

string s;

int main() {
    cin.tie(nullptr)->sync_with_stdio(false);
    pam::clear();
    cin >> n;
    cin >> s;
    pam::clear();
    for(int i = 0; i < s.size(); i ++)
        pam::insert(s[i]);
    pam::work1();
    for(int i = 0; i < s.size(); i ++)
        pam::insert(s[i]);
    cout << pam::work(1) << endl;
    
    // vector<pair<int, int>> ccnt;
    // for(int i = 2; i <= pam::sz; i ++)
    //     ccnt.push_back({pam :: cnt[i], pam :: len[i]});
    // for(auto [c, l] : ccnt)
    //     cout << c << ' ' << l << endl;
    // for(int i = 0; i < s.size(); i ++)
    //     pam::insert(s[i]);
    // int now2 = pam::work(2);
    // vector<pair<int, int>> ccnt2;
    // for(int i = 2; i <= pam::sz; i ++)
    // {
    //     if(pam::len[i] <= pam::tot / 2)
    //     ccnt2.push_back({pam :: cnt[i], pam :: len[i]});
    // }
    // for(auto [c, l] : ccnt2)
    // {
        
    // }
    // cout << now2 << endl;
    // cout << now2 - now1 << endl;
    // pam1::clear();
    // for(int i = 0; i < t.size(); i ++)
    //     pam1::insert(t[i]);
    // int now2 = pam1::work(2);
    // int now = ((now2 - now1) % mod + mod) % mod; 
    // cout << now << endl;
    return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

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

input:

5
01010

output:

39

result:

ok 1 number(s): "39"

Test #2:

score: 0
Accepted
time: 2ms
memory: 13952kb

input:

8
66776677

output:

192

result:

ok 1 number(s): "192"

Test #3:

score: 0
Accepted
time: 2ms
memory: 13880kb

input:

1
1

output:

1

result:

ok 1 number(s): "1"

Test #4:

score: 0
Accepted
time: 1ms
memory: 13888kb

input:

2
22

output:

12

result:

ok 1 number(s): "12"

Test #5:

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

input:

2
21

output:

2

result:

ok 1 number(s): "2"

Test #6:

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

input:

3
233

output:

10

result:

ok 1 number(s): "10"

Test #7:

score: 0
Accepted
time: 1ms
memory: 13816kb

input:

3
666

output:

54

result:

ok 1 number(s): "54"

Test #8:

score: -100
Wrong Answer
time: 20ms
memory: 122308kb

input:

1000000
3333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333...

output:

758251518

result:

wrong answer 1st numbers differ - expected: '496166704', found: '758251518'