QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#605215#7876. Cyclic Substringslonelywolf#WA 143ms409780kbC++202.1kb2024-10-02 16:11:162024-10-02 16:11:16

Judging History

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

  • [2024-10-02 16:11:16]
  • 评测
  • 测评结果:WA
  • 用时:143ms
  • 内存:409780kb
  • [2024-10-02 16:11:16]
  • 提交

answer

#include <bits/stdc++.h>
using namespace std;

#define int long long

const int maxn = 3e6 + 10;
const int mod = 998244353;

vector<pair<int, int>> a[2] {};
namespace pam {
int sz, tot, last;
int cnt[maxn], ch[maxn][10], 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;
  }
  last = ch[now][c - '0'];
  cnt[last]++;
}

void solve(int op) {
  for (int i = sz; i >= 0; i--) {
    cnt[fail[i]] += cnt[i];
  }
  for (int i = 2; i <= sz; i++) {  // 更新答案
    a[op].push_back({len[i], cnt[i]});
  }
}
}  // namespace pam

char s[maxn];

int ans = 0;

signed main() {
	int n;
	cin >> n;
  pam::clear();
  scanf("%s", s + 1);
  for (int i = 1; s[i]; i++) {
    pam::insert(s[i]);
  }
  pam::solve(0);
  pam::clear();
  for (int i = 1; s[i]; i++) {
  	pam::insert(s[i]);
  }
  for (int i = 1; s[i]; i++) {
  	pam::insert(s[i]);
  }  
  pam::solve(1);

  for (int i = 0; i < a[1].size(); i++) {
  	if (i < a[0].size()) {
  		// cout << a[0][i].first << " " << a[0][i].second << "\n";
  		// cout << a[1][i].first << " " << a[1][i].second << "\n";
  		// cout << "\n";
  		ans = (ans + a[0][i].first * (a[1][i].second - a[0][i].second) % mod * (a[1][i].second - a[0][i].second) % mod) % mod;
  	} else {
  		// cout << a[1][i].first << " " << a[1][i].second << "\n";
  		if (a[1][i].first <= n) {
  			ans = (ans + a[1][i].first * a[1][i].second % mod * a[1][i].second % mod) % mod;
  		}
  	}
  }

  cout << ans << "\n";

  return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

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

input:

5
01010

output:

39

result:

ok 1 number(s): "39"

Test #2:

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

input:

8
66776677

output:

192

result:

ok 1 number(s): "192"

Test #3:

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

input:

1
1

output:

1

result:

ok 1 number(s): "1"

Test #4:

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

input:

2
22

output:

12

result:

ok 1 number(s): "12"

Test #5:

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

input:

2
21

output:

2

result:

ok 1 number(s): "2"

Test #6:

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

input:

3
233

output:

10

result:

ok 1 number(s): "10"

Test #7:

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

input:

3
666

output:

54

result:

ok 1 number(s): "54"

Test #8:

score: 0
Accepted
time: 47ms
memory: 267196kb

input:

1000000
3333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333...

output:

496166704

result:

ok 1 number(s): "496166704"

Test #9:

score: -100
Wrong Answer
time: 143ms
memory: 409780kb

input:

3000000
2222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222...

output:

294238320

result:

wrong answer 1st numbers differ - expected: '890701718', found: '294238320'