結果

問題 No.430 文字列検索
コンテスト
ユーザー okaduki
提出日時 2018-01-31 00:25:34
言語 C++14
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 16 ms / 2,000 ms
コード長 3,098 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,087 ms
コンパイル使用メモリ 179,108 KB
実行使用メモリ 10,240 KB
最終ジャッジ日時 2024-11-10 00:18:38
合計ジャッジ時間 2,707 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 14
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
using namespace std;

using VI = vector<int>;
using VVI = vector<VI>;
using PII = pair<int, int>;
using LL = long long;
using VL = vector<LL>;
using VVL = vector<VL>;
using PLL = pair<LL, LL>;
using VS = vector<string>;

#define ALL(a)  begin((a)),end((a))
#define RALL(a) (a).rbegin(), (a).rend()
#define PB push_back
#define EB emplace_back
#define MP make_pair
#define SZ(a) int((a).size())
#define SORT(c) sort(ALL((c)))
#define RSORT(c) sort(RALL((c)))
#define UNIQ(c) (c).erase(unique(ALL((c))), end((c)))

#define FOR(i,a,b) for(int i=(a);i<(b);++i)
#define REP(i,n)  FOR(i,0,n)

#define FF first
#define SS second

#define DUMP(x) cout<<#x<<":"<<(x)<<endl
template<class S, class T>
istream& operator>>(istream& is, pair<S,T>& p){
  return is >> p.FF >> p.SS;
}
template<class T>
istream& operator>>(istream& is, vector<T>& xs){
  for(auto& x: xs)
	is >> x;
  return is;
}
template<class S, class T>
ostream& operator<<(ostream& os, const pair<S,T>& p){
  return os << p.FF << " " << p.SS;
}
template<class T>
ostream& operator<<(ostream& os, const vector<T>& xs){
  for(unsigned int i=0;i<xs.size();++i)
	os << (i?" ":"") << xs[i];
  return os;
}
template<class T>
void maxi(T& x, T y){
  if(x < y) x = y;
}
template<class T>
void mini(T& x, T y){
  if(x > y) x = y;
}


const double EPS = 1e-10;
const double PI  = acos(-1.0);
const LL MOD = 1e9+7;

struct Aho{
  // failure link is slways 0.
  static const int FAIL = 0;
  static const int SIZE = 26 + 1;
  static int idx(char c){
	return c-'A' + 1;
  }
  
  struct Node{
	vector<Node*> chd;
	vector<int> accept;
	Node(){ chd.assign(SIZE, nullptr); }
	//~Node(){ FOR(i,1,SIZE){ delete chd[i]; chd[i] = nullptr; } }
  };

  Node* root;
  int kind;
  Aho(){ root = nullptr; }
  
  void build(const vector<string>& vs){
	//delete root;
	root = new Node();
	kind = vs.size();
	for(int i=0;i<kind;++i){
	  Node* crt = root;
	  for(char c: vs[i]){
		int ix = idx(c);
		if(!crt->chd[ix])
		  crt->chd[ix] = new Node();
		crt = crt->chd[ix];
	  }
	  crt->accept.push_back(i);
	}

	queue<Node*> q;
	for(int i=1;i<SIZE;++i)
	  if(root->chd[i]){
		root->chd[i]->chd[FAIL] = root;
		q.push(root->chd[i]);
	  }
	  else
		root->chd[i] = root;

	while(!q.empty()){
	  Node* t = q.front();
	  q.pop();
	  for(int i=1;i<SIZE;++i)
		if(t->chd[i]){
		  q.push(t->chd[i]);
		  Node* p = t->chd[FAIL];
		  while(!p->chd[i]) p = p->chd[FAIL];
		  t->chd[i]->chd[FAIL] = p->chd[i];
		  t->chd[i]->accept.insert(t->chd[i]->accept.end(), p->chd[i]->accept.begin(), p->chd[i]->accept.end());
		}
	}
  }

  /**
   * require: |res| == kind
   */
  void match(const string& s, vector<int>& res){
	Node* crt = root;
	for(char c: s){
	  int ix = idx(c);
	  while(!crt->chd[ix]) crt = crt->chd[FAIL];
	  crt = crt->chd[ix];
	  for(int id: crt->accept)
		++res[id];
	}
  }
};

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

  string S;
  cin >> S;
  int M;
  cin >> M;
  VS vs(M);
  REP(i,M) cin >> vs[i];

  Aho aho;
  aho.build(vs);

  VI res(M);
  aho.match(S, res);

  cout << accumulate(ALL(res), 0) << endl;

  return 0;
}
0