結果
| 問題 | No.430 文字列検索 |
| コンテスト | |
| ユーザー |
 smiken_61
|
| 提出日時 | 2019-09-30 22:54:07 |
| 言語 | C++11（廃止可能性あり） (gcc 13.3.0 + boost 1.89.0) |
| 結果 |
AC
|
| 実行時間 | 10 ms / 2,000 ms |
| コード長 | 11,804 bytes |
| 記録 | |
| コンパイル時間 | 1,520 ms |
| コンパイル使用メモリ | 169,416 KB |
| 実行使用メモリ | 8,704 KB |
| 最終ジャッジ日時 | 2024-11-10 00:35:09 |
| 合計ジャッジ時間 | 1,962 ms |
|
ジャッジサーバーID （参考情報） |
judge2 / judge3 |
（要ログイン）
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 4 |
| other | AC * 14 |
ソースコード
#include <bits/stdc++.h>
// #include <boost/multiprecision/cpp_int.hpp>
// #define int long long
#define inf 1000000007
#define pa pair<int,int>
#define ll long long
#define pal pair<double,double>
#define ppap pair<pa,int>
#define PI 3.14159265358979323846
#define paa pair<int,char>
#define mp make_pair
#define pb push_back
#define EPS (1e-8)
int dx[8]={0,1,0,-1,1,1,-1,-1};
int dy[8]={1,0,-1,0,-1,1,1,-1};
using namespace std;
class pa3{
public:
int x;
int y,z;
pa3(int x=0,int y=0,int z=0):x(x),y(y),z(z) {}
bool operator < (const pa3 &p) const{
if(x!=p.x) return x<p.x;
if(y!=p.y) return y<p.y;
return z<p.z;
//return x != p.x ? x<p.x: y<p.y;
}
bool operator > (const pa3 &p) const{
if(x!=p.x) return x>p.x;
if(y!=p.y) return y>p.y;
return z>p.z;
//return x != p.x ? x<p.x: y<p.y;
}
bool operator == (const pa3 &p) const{
return x==p.x && y==p.y && z==p.z;
}
bool operator != (const pa3 &p) const{
return !( x==p.x && y==p.y && z==p.z);
}
};
class pa4{
public:
int x;
int y,z,w;
pa4(int x=0,int y=0,int z=0,int w=0):x(x),y(y),z(z),w(w) {}
bool operator < (const pa4 &p) const{
if(x!=p.x) return x<p.x;
if(y!=p.y) return y<p.y;
if(z!=p.z)return z<p.z;
return w<p.w;
//return x != p.x ? x<p.x: y<p.y;
}
bool operator > (const pa4 &p) const{
if(x!=p.x) return x>p.x;
if(y!=p.y) return y>p.y;
if(z!=p.z)return z>p.z;
return w>p.w;
//return x != p.x ? x<p.x: y<p.y;
}
bool operator == (const pa4 &p) const{
return x==p.x && y==p.y && z==p.z &&w==p.w;
}
};
class pa2{
public:
int x,y;
pa2(int x=0,int y=0):x(x),y(y) {}
pa2 operator + (pa2 p) {return pa2(x+p.x,y+p.y);}
pa2 operator - (pa2 p) {return pa2(x-p.x,y-p.y);}
bool operator < (const pa2 &p) const{
return y != p.y ? y<p.y: x<p.x;
}
bool operator > (const pa2 &p) const{
return x != p.x ? x<p.x: y<p.y;
}
bool operator == (const pa2 &p) const{
return abs(x-p.x)==0 && abs(y-p.y)==0;
}
bool operator != (const pa2 &p) const{
return !(abs(x-p.x)==0 && abs(y-p.y)==0);
}
};
string itos( int i ) {
ostringstream s ;
s << i ;
return s.str() ;
}
int gcd(int v,int b){
if(v>b) return gcd(b,v);
if(v==b) return b;
if(b%v==0) return v;
return gcd(v,b%v);
}
int mod;
int extgcd(int a, int b, int &x, int &y) {
if (b == 0) {
x = 1;
y = 0;
return a;
}
int d = extgcd(b, a%b, y, x);
y -= a/b * x;
return d;
}
pa operator+(const pa & l,const pa & r) {
return {l.first+r.first,l.second+r.second};
}
pa operator-(const pa & l,const pa & r) {
return {l.first-r.first,l.second-r.second};
}
int pr[1000010];
int inv[1000010];
int beki(int wa,int rr,int warukazu){
if(rr==0) return 1%warukazu;
if(rr==1) return wa%warukazu;
wa%=warukazu;
if(rr%2==1) return ((ll)beki(wa,rr-1,warukazu)*(ll)wa)%warukazu;
ll zx=beki(wa,rr/2,warukazu);
return (zx*zx)%warukazu;
}
int comb(int nn,int rr){
if(rr<0 || rr>nn || nn<0) return 0;
int r=pr[nn]*inv[rr];
r%=mod;
r*=inv[nn-rr];
r%=mod;
return r;
}
void gya(int ert){
pr[0]=1;
for(int i=1;i<=ert;i++){
pr[i]=(pr[i-1]*i)%mod;
}
inv[ert]=beki(pr[ert],mod-2,mod);
for(int i=ert-1;i>=0;i--){
inv[i]=inv[i+1]*(i+1)%mod;
}
}
// cin.tie(0);
// ios::sync_with_stdio(false);
//priority_queue<pa3,vector<pa3>,greater<pa3>> pq;
//sort(ve.begin(),ve.end(),greater<int>());
// mt19937(clock_per_sec);
// mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count()) ;
//----------------kokomade tenpure------------
int ans=0;
struct aho{
// add_string(文字列, id)で検索パターンを追加 文字列は全部distinct!!
// shori()で前処理
// find_string(s) で文字列s内にあるパターンを列挙(YARU内でその処理を書く)
typedef struct trie{
int moji;
int id;
int depth;
int P;
trie* parent;
trie* flink;
trie* outer;
trie* child[26];
}trie;
private:
public:
int cd=0;
trie* new_trie(){
trie *p = new trie;
p->P=-1;
p->id=cd;
cd++;
p->flink=NULL;
p->parent=NULL;
p->outer=NULL;
for(int i=0;i<26;i++) p->child[i]=NULL;
return p;
}
trie* ROOT;
aho(){
ROOT=new_trie();
ROOT->moji=-1;
ROOT->depth=0;
}
void add_string(string str, int str_id){
int it=0;
trie* trie_now=ROOT;
int l=str.length();
for(int i=0;i<l;i++){
int tugi=str[i]-'a'; /////////////////////// komoji!!
if(trie_now->child[tugi]==NULL){
trie* p_new=new_trie();
p_new->moji= tugi;
p_new->parent=trie_now;
p_new->depth=i+1;
trie_now->child[tugi]=p_new;
}
trie_now=trie_now->child[tugi];
}
trie_now->P=str_id;
}
void debug(){
debug_dfs(ROOT);
}
void debug_dfs(trie* e){
// cout<<e->deb<<" "<<e->id<<" "<<e->moji<<" ";
// if(e->flink!=NULL)cout<<e->flink->deb<<" ";
cout<<"c " ;
for(int i=0;i<26;i++)if(e->child[i]!=NULL) cout<<i<<" ";
cout<<endl;
for(int i=0;i<26;i++)if(e->child[i]!=NULL) debug_dfs(e->child[i]);
}
queue<trie*> queue_trie;
void shori(){
ROOT->parent=ROOT;
ROOT->flink=ROOT;
queue_trie.push(ROOT);
while(queue_trie.size()){
trie* trie_now=queue_trie.front();
queue_trie.pop();
for(int i=0;i<26;i++)if(trie_now->child[i]!=NULL) queue_trie.push(trie_now->child[i]);
if(trie_now==ROOT) continue;
if(trie_now->parent==ROOT){
trie_now->flink=ROOT;
continue;
}
trie* itr=(trie_now->parent)->flink;
while(1){
if(itr==ROOT){
if(ROOT->child[trie_now->moji] ==NULL){
trie_now->flink=ROOT;
break;
}
else{
trie_now->flink=ROOT->child[trie_now->moji];
break;
}
}
else{
if(itr->child[trie_now->moji] !=NULL){
trie_now->flink=itr->child[trie_now->moji];
break;
}
else itr=itr->flink;
}
}
//cout<<trie_now->deb<<" "<<trie_now->flink->deb<<endl;
}
while(queue_trie.size())queue_trie.pop();
queue_trie.push(ROOT);
while(queue_trie.size()){
trie* trie_now=queue_trie.front();
queue_trie.pop();
for(int i=0;i<26;i++)if(trie_now->child[i]!=NULL)queue_trie.push(trie_now->child[i]);
if(trie_now==ROOT) continue;
if(trie_now->flink->P<0) trie_now->outer=trie_now->flink->outer;
else{
trie_now->outer=trie_now->flink;
}
}
}
//void shori(int id,int iti);
void string_find(string str){
trie* trie_now=ROOT;
int l=str.length();
for(int i=0;i<l;i++){
int moji=str[i]-'a'; //////////////////////////// komoji
while(trie_now->child[moji]==NULL){
trie_now=trie_now->flink;
if(trie_now==ROOT) break;
}
if(trie_now->child[moji]!=NULL){
trie_now=trie_now->child[moji];
}
trie* tr=trie_now;
while(tr!=NULL){
if(tr->P>=0)YARU(tr->P,i);
tr=tr->outer;
}
int sute=i-trie_now->depth-1;
// i 文字目を見ている
// prefix を sute 文字捨てた
}
}
void YARU(int id,int iti){
ans++;
return;
// ans++;
//id は文字列番号
//iti は終了位置
}
};
string f(string s){
for(auto &c:s){
c-='A'-'a';
}
return s;
}
signed main(){
cin.tie(0);
ios::sync_with_stdio(false);
string t;
aho A;
cin>>t;
int q;
cin>>q;
t=f(t);
for(int i=0;i<q;i++){
string s;
cin>>s;
s=f(s);
A.add_string(s,i);
}
A.shori();
for(int i=0;i<1;i++){
A.string_find(t);
cout<<ans<<endl;
}
return 0;
}