QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#182814 | #2343. First of Her Name | PetroTarnavskyi | WA | 3ms | 57148kb | C++17 | 2.2kb | 2023-09-18 16:15:20 | 2023-09-18 16:15:21 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
#define FOR(i, a, b) for(int i = (a); i < (b); i++)
#define RFOR(i, b, a) for(int i = (b) - 1; i >= (a); i--)
#define SZ(a) (int)a.size()
#define ALL(a) a.begin(), a.end()
#define PB push_back
#define MP make_pair
#define F first
#define S second
typedef long long LL;
typedef vector<int> VI;
typedef pair<int, int> PII;
typedef double db;
const int MAX = 1'000'447;
const int AL = 26;
struct Node
{
int p;
int c;
int g[AL];
int nxt[AL];
int link;
void init()
{
c = -1;
p = -1;
fill(g, g + AL, -1);
fill(nxt, nxt + AL, -1);
link = -1;
}
};
struct AC
{
Node A[MAX];
int sz;
void init()
{
A[0].init();
sz = 1;
}
int addStr(const string& s)
{
int x = 0;
FOR (i, 0, SZ(s))
{
int c = s[i] - 'a';
if (A[x].nxt[c] == -1)
{
A[x].nxt[c] = sz;
A[sz].init();
A[sz].c = c;
A[sz].p = x;
sz++;
}
x = A[x].nxt[c];
}
return x;
}
int go(int x, int c)
{
if (A[x].g[c] == -1)
{
if (A[x].nxt[c] != -1)
A[x].g[c] = A[x].nxt[c];
else if (x != 0)
A[x].g[c] = go(getLink(x), c);
else
A[x].g[c] = 0;
}
return A[x].g[c];
}
int getLink(int x)
{
if (A[x].link != -1)
return A[x].link;
if (x == 0 || A[x].p == 0)
return 0;
return A[x].link = go(getLink(A[x].p), A[x].c);
}
} A;
int c[MAX];
VI g[MAX];
VI g2[MAX];
int a[MAX];
int val[MAX];
void dfs(int v, int x)
{
x = A.go(x, c[v] - 'A');
val[x]++;
for (auto to : g[v])
dfs(to, x);
}
void dfs2(int v)
{
for (auto to : g2[v])
{
if (to != 0)
{
dfs2(to);
val[v] += val[to];
}
}
}
int main()
{
ios::sync_with_stdio(0);
cin.tie(0);
cout << fixed << setprecision(15);
int n, k;
cin >> n >> k;
vector<string> v(k);
FOR (i, 0, n)
{
char ch;
int par;
cin >> ch >> par;
c[i + 1] = ch;
g[par].PB(i + 1);
}
A.init();
FOR (i, 0, k)
{
cin >> v[i];
reverse(ALL(v[i]));
a[i] = A.addStr(v[i]);
}
FOR (i, 0, A.sz)
{
g2[A.getLink(i)].PB(i);
}
dfs(1, 0);
dfs2(0);
FOR (i, 0, k)
{
cout << val[a[i]] << '\n';
}
cerr << double(clock()) / CLOCKS_PER_SEC << '\n';
return 0;
}
Details
Test #1:
score: 0
Wrong Answer
time: 3ms
memory: 57148kb