QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#363536#2343. First of Her NameTWTP_TCTF#WA 0ms3868kbC++172.2kb2024-03-23 23:59:022024-03-23 23:59:03

Judging History

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

  • [2024-03-23 23:59:03]
  • 评测
  • 测评结果:WA
  • 用时:0ms
  • 内存:3868kb
  • [2024-03-23 23:59:02]
  • 提交

answer

#include <bits/stdc++.h>

#define ll long long
using namespace std;
const int N = 3e5 + 5, mod = 1e9 + 7, A = 26;

/*
int trie[N][A];
int go[N][A]; ///holds the node that you will go to after failure and stuff
int fail[N];    ///the failure function for each
ll ans[N];


*/
struct SA {
    struct node {
        int to[26];
        int link, len;
        ll co = 0;

        node() {
            memset(to, 0, sizeof to);
            co = 0, link = 0, len = 0;
        }
    };

    int last, sz;
    vector<node> v;

    SA(int n) {
        v = vector<node>(n + 1);
        last = 0, sz = n + 1;
    }

    void add_letter(int c, int last, int p) {

        v[last].len = v[p].len + 1;
        v[last].co = 1;
        for (; v[p].to[c] == 0; p = v[p].link)
            v[p].to[c] = last;
        if (v[p].to[c] == last) {
            v[last].link = 0;
            return;
        }
        int q = v[p].to[c];
        if (v[q].len == v[p].len + 1) {
            v[last].link = q;
            return;
        }
        int cl = sz++;
        v.push_back(v[q]);
        v.back().co = 0;
        v.back().len = v[p].len + 1;
        v[last].link = v[q].link = cl;

        for (; v[p].to[c] == q; p = v[p].link)
            v[p].to[c] = cl;
    }

    void build_co() {
        priority_queue<pair<int, int>> q;
        for (int i = sz - 1; i > 0; i--)
            q.push({v[i].len, i});
        while (q.size()) {
            int i = q.top().second;
            q.pop();
            v[v[i].link].co += v[i].co;
        }
    }
};

SA *sa;

ll solve(string s) {
    std::reverse(s.begin(), s.end());
    int cur = 0;
    for (auto ch: s) {
        if (sa->v[cur].to[ch - 'A'] == 0)return 0;
        cur = sa->v[cur].to[ch - 'A'];
    }
    return sa->v[cur].co;
}

int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    int n, q;
    cin >> n >> q;
    sa = new SA(n);
    for (int i = 0; i < n; ++i) {
        char x;
        int p;
        cin >> x >> p;
        x -= 'A';
        sa->add_letter(x,i+1, p);
    }
    sa->build_co();
    while (q--) {
        string s;
        cin >> s;
        cout << solve(s) << '\n';
    }


    return 0;
}

Details

Test #1:

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

Test #2:

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

Test #3:

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

Test #4:

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

Test #5:

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

Test #6:

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

Test #7:

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

Test #8:

score: -100
Wrong Answer
time: 0ms
memory: 3608kb