QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#103429#4375. StringzbceyondTL 0ms0kbC++111.3kb2023-05-05 20:51:132023-05-05 20:51:15

Judging History

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

  • [2023-08-10 23:21:45]
  • System Update: QOJ starts to keep a history of the judgings of all the submissions.
  • [2023-05-05 20:51:15]
  • 评测
  • 测评结果:TL
  • 用时:0ms
  • 内存:0kb
  • [2023-05-05 20:51:13]
  • 提交

answer

#include<bits/stdc++.h>
#define rep(i,x,y) for(int i=x;i<=y;i++)
using namespace std;
#define int long long
const int N = 2e5+10;
const int mod = 998244353;
int qmi(int a,int b) {
    int res = 1;
    for (; b; b >>= 1, a = a * a % mod) {
        if (b & 1)res = res * a % mod;
    }
    return res;
}
void solve(int tc) {
    string s;
    int k;
    cin >> s >> k;
    s = " " + s;
    int n = s.size();
    vector<int> nxt(n);
    vector<vector<int>> e(n);
    e[0].push_back(1);
    for (int i = 2; i < n; i++) {
        nxt[i] = nxt[i - 1];
        while (nxt[i] and s[i] != s[nxt[i] + 1])nxt[i] = nxt[nxt[i]];
        if (s[i] == s[nxt[i] + 1])nxt[i]++;
        while (2 * nxt[i] >= i)nxt[i] = nxt[nxt[i]];
        e[nxt[i]].push_back(i);
    }
    vector<int> cnt(k);
    int ans = 1;
    function<void(int, int)> dfs = [&](int u, int fa) {
        int x = cnt[u % k] + 1;
        if (u % k == 0 and u != 0)x++;
        ans = ans * x % mod;

        if (u != 0)cnt[u * 2 % k]++;
        for (auto v: e[u]) {
            if (v == fa)continue;
            dfs(v, u);
        }
        if (u != 0)cnt[u * 2 % k]--;
    };
    dfs(0, -1);
    cout << ans;
}
signed main() {
    cin.tie(0)->sync_with_stdio(0);
    int tc = 1;
    cin >> tc;
    for (int i = 1; i <= tc; i++) {
        solve(i);
    }
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 0
Time Limit Exceeded

input:

10
kkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkk...

output:


result: