結果
問題 | No.430 文字列検索 |
ユーザー | smiken_61 |
提出日時 | 2019-09-30 22:54:07 |
言語 | C++11 (gcc 11.4.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 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,820 KB |
testcase_01 | AC | 10 ms
8,704 KB |
testcase_02 | AC | 3 ms
6,820 KB |
testcase_03 | AC | 4 ms
6,816 KB |
testcase_04 | AC | 2 ms
6,816 KB |
testcase_05 | AC | 2 ms
6,820 KB |
testcase_06 | AC | 2 ms
6,816 KB |
testcase_07 | AC | 1 ms
6,816 KB |
testcase_08 | AC | 2 ms
6,816 KB |
testcase_09 | AC | 2 ms
6,820 KB |
testcase_10 | AC | 1 ms
6,820 KB |
testcase_11 | AC | 8 ms
6,912 KB |
testcase_12 | AC | 9 ms
7,552 KB |
testcase_13 | AC | 9 ms
7,552 KB |
testcase_14 | AC | 6 ms
6,816 KB |
testcase_15 | AC | 5 ms
6,816 KB |
testcase_16 | AC | 5 ms
6,816 KB |
testcase_17 | AC | 5 ms
6,820 KB |
ソースコード
#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; }