Pagini recente » Cod sursa (job #3164250) | Cod sursa (job #2393187) | Cod sursa (job #1957086) | Cod sursa (job #3235300) | Cod sursa (job #3228509)
#include <bits/stdc++.h>
using namespace std;
ifstream fin("ahocorasick.in");
ofstream fout("ahocorasick.out");
int go(int,int);
int getsuf(int);
string v;
int where[105];
vector<int> ord;
struct node
{
int go[26];
int lit;
int suflink,par,niv;
int dp;
} emp;
vector<node> t;
bool comp(int a, int b)
{
return t[a].niv>t[b].niv;
}
void add(string s,int ind)
{
int me=0;
for(char c:s)
{
int lit=c-'a';
if(t[me].go[lit]==-1)
{
int x=t.size();
t.push_back(emp);
t[x].lit=lit;
t[x].par=me;
t[x].niv=t[me].niv+1;
t[me].go[lit]=x;
}
me=t[me].go[lit];
}
where[ind]=me;
}
int go(int me,int lit)
{
if(t[me].go[lit]!=-1)
return t[me].go[lit];
if(me==0)
return 0;
int x=getsuf(me);
t[me].go[lit]=go(x,lit);
return t[me].go[lit];
}
int getsuf(int me)
{
if(t[me].suflink!=-1)
return t[me].suflink;
if(me==0||t[me].par==0)
return 0;
t[me].suflink=go(getsuf(t[me].par),t[me].lit);
return t[me].suflink;
}
int main()
{
ios_base::sync_with_stdio(false);
fin.tie(0);
for(int i=0;i<26;i++)
emp.go[i]=-1;
emp.lit=-1;
emp.suflink=-1;
emp.par=0;
emp.dp=0;
emp.niv=0;
t.push_back(emp);
fin>>v;
int n;
fin>>n;
for(int i=1;i<=n;i++)
{
string s;
fin>>s;
add(s,i);
}
int me=0;
for(char c:v)
{
int lit=c-'a';
me=go(me,lit);
t[me].dp++;
}
for(int i=1;i<t.size();i++)
ord.push_back(i);
sort(ord.begin(),ord.end(),comp);
for(int i:ord)
{
int x=getsuf(i);
t[x].dp+=t[i].dp;
}
for(int i=1;i<=n;i++)
fout<<t[where[i]].dp<<'\n';
return 0;
}