QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#624603#7876. Cyclic SubstringsYurilyRE 2ms13840kbC++202.1kb2024-10-09 16:13:442024-10-09 16:13:46

Judging History

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

  • [2024-10-09 16:13:46]
  • 评测
  • 测评结果:RE
  • 用时:2ms
  • 内存:13840kb
  • [2024-10-09 16:13:44]
  • 提交

answer

#include <algorithm>
#include <cstring>
#include <iostream>
#include <string>
using namespace std;
const int maxn = 1e6 + 5;
const long long MOD=998244353;
int tmp,pre,d[maxn];
namespace pam {
int sz, tot, last;
int cnt[maxn], ch[maxn][26], 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;
 //   cnt[x]=cnt[fail[x]]+1;
  }
  last = ch[now][c - '0'];
  cnt[last]++;
}
void solve1() {
  long long ans = 0;
  for (int i = sz; i >= 0; i--) {
    cnt[fail[i]] += cnt[i];
  }
  for (int i = 2; i <= sz; i++) { 
 // 	cout<<i<<" "<<cnt[i]<<"  "<<len[i]<<endl;// 更新答案
 	d[i]=cnt[i];
  }
}
long long solve2() {
  long long ans = 0;
  for (int i = sz; i >= 0; i--) {
    cnt[fail[i]] += cnt[i];
  }
  for (int i = 2; i <= sz; i++) { 
  	if(len[i]<=tmp)
	  	if(i>pre) // 更新答案
    		ans = (ans + (long long)cnt[i]*cnt[i]%MOD*len[i]%MOD)%MOD;
    	else
    		ans=(ans + (long long)(cnt[i]-d[i])*(cnt[i]-d[i])%MOD*len[i]%MOD)%MOD;
  }
  return ans;
}
}  // namespace pam

string s;

int main() {
  cin>>tmp;
  cin.tie(nullptr)->sync_with_stdio(false);
  pam::clear();
  long long ans1=0,ans2=0;
  cin >> s;
  for (int i = 0; i < s.size(); i++) {
    pam::insert(s[i]);
  //  cout<<pam::last<<" "<<i<<" "<<pam::cnt[pam::last]<<endl;
  }
  pre=pam::sz;
  pam::solve1();
  s=s+s;
  pam::clear();
  for (int i = 0; i < s.size(); i++) {
  //	cout<<s[i]<<endl;
    pam::insert(s[i]);
  }
  ans2=pam::solve2();
  cout<<ans2<<endl;
  return 0;
}

详细

Test #1:

score: 100
Accepted
time: 2ms
memory: 11764kb

input:

5
01010

output:

39

result:

ok 1 number(s): "39"

Test #2:

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

input:

8
66776677

output:

192

result:

ok 1 number(s): "192"

Test #3:

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

input:

1
1

output:

1

result:

ok 1 number(s): "1"

Test #4:

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

input:

2
22

output:

12

result:

ok 1 number(s): "12"

Test #5:

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

input:

2
21

output:

2

result:

ok 1 number(s): "2"

Test #6:

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

input:

3
233

output:

10

result:

ok 1 number(s): "10"

Test #7:

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

input:

3
666

output:

54

result:

ok 1 number(s): "54"

Test #8:

score: -100
Runtime Error

input:

1000000
3333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333...

output:


result: