QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#724783#4375. StringXfJbUhpyzgaW#AC ✓171ms8536kbC++141.1kb2024-11-08 15:03:002024-11-08 15:03:01

Judging History

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

  • [2024-11-08 15:03:01]
  • 评测
  • 测评结果:AC
  • 用时:171ms
  • 内存:8536kb
  • [2024-11-08 15:03:00]
  • 提交

answer

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

const int N = 1000100, M = 998244353;
char s[N];
int n, k, nxt[N];

int exgcd(int a, int b, int &x, int &y) {
    if (!b) { x = 1, y = 0; return a; }
    int g = exgcd(b, a % b, y, x);
    y -= a / b * x;
    return g;
}
int foo(int t, int k, int v) { // tx + ky = v, x,y >= 0
    if (v < 0) return 0;
    int x, y;
    int g = exgcd(t, k, x, y);
    if (v % g != 0) return 0;
    t /= g, k /= g, v /= g;
    x = 1ll * x * v % k;
    if (x < 0) x += k;
    assert(1ll * t * x % k == v % k);
    long long a = v - 1ll * x * t, b = 1ll * k * t;
    if (a < 0) return 0;
    return a / b + 1;
}

void solve() {
    cin >> s + 1 >> k;
    n = strlen(s + 1);
    nxt[0] = -1;

    long long ans = 1;
    for (int i = 1; i <= n; i++) {
        int p = nxt[i - 1];
        while (~p && s[p + 1] != s[i]) p = nxt[p];
        nxt[i] = p + 1;
        ans = ans * (1 + foo(2 * (i - nxt[i]), k, i - k)) % M;
    }
    cout << ans << "\n";
}

int main() {
    cin.tie(0)->sync_with_stdio(0), cout.tie(0);
    int T;
    for (cin >> T; T; T--) solve();
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 171ms
memory: 8536kb

input:

10
kkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkk...

output:

811844748
106557744
583082277
750875845
539889742
198008691
286657978
344446711
612851418
36100066

result:

ok 10 lines