結果

問題 No.430 文字列検索
コンテスト
ユーザー tkmst201
提出日時 2017-05-14 21:24:18
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 169 ms / 2,000 ms
コード長 1,990 bytes
コンパイル時間 1,577 ms
コンパイル使用メモリ 176,880 KB
実行使用メモリ 27,136 KB
最終ジャッジ日時 2024-11-10 00:15:07
合計ジャッジ時間 2,856 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 14
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

#define FOR(i,a,b) for(int i=(a);i<(b);i++)
#define REP(i,n) FOR(i,0,n)
#define ALL(v) (v).begin(),(v).end()

template<typename A, typename B> inline bool chmax(A &a, B b) { if (a<b) { a=b; return 1; } return 0; }
template<typename A, typename B> inline bool chmin(A &a, B b) { if (a>b) { a=b; return 1; } return 0; }

typedef unsigned long long ull;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef pair<int, pii> P;

const ll INF = 1ll<<60;
const ll MOD = 1000000007;
const double EPS  = 1e-10;

struct RollingHash {
	vector<ll> mod;
	vector<vector<ll> > hs, pw;
	string s;
	ll base;
	int n;
	
	RollingHash() {}
	RollingHash(string s) { init(s); }
	
	void init(string _s) {
		s = _s;
		n = s.size();
		
		mod.clear();
		hs.clear();
		pw.clear();
		
		//mod.push_back(999999937);
		mod.push_back(1000000007);
		
		base = 9973;
		
		hs.resize(mod.size(), vector<ll>(n + 1));
		pw.resize(mod.size(), vector<ll>(n + 1));
		
		REP(i, mod.size()) {
			pw[i][0] = 1;
			REP(j, n) {
				hs[i][j + 1] = (hs[i][j] + s[j]) * base % mod[i];
				pw[i][j + 1] = pw[i][j] * base % mod[i];
			}
		}
	}
	
	inline ll hash(int l, int r, int i) {
		return (hs[i][r] - hs[i][l] * pw[i][r - l] % mod[i] + mod[i]) % mod[i];
	}
	
	inline bool match(int l1, int r1, int l2, int r2) {
		bool res = true;
		REP(i, mod.size()) if (hash(l1, r1, i) != hash(l2, r2, i)) res = false;
		return res;
	}
	
	int lcp(int i, int j) {
		int l = 0, r = min(n - i, n - j) + 1;
		while (r - l > 1) {
			int m = (l + r) / 2;
			if (match(i, i + m, j, j + m)) l = m;
			else r = m;
		}
		return l;
	}
};

int main() {
	string s;
	int m;
	
	cin >> s >> m;
	
	ll ans = 0;
	
	RollingHash rh(s);
	
	map<ll, int> ss[11];
	FOR(i, 1, 11) {
		REP(j, s.size()) {
			ss[i][rh.hash(j, j + i, 0)]++;
		}
	}
	
	REP(i, m) {
		string c;
		cin >> c;
		
		RollingHash tmp(c);
		ans += ss[c.size()][tmp.hash(0, c.size(), 0)];
	}
	
	cout << ans << endl;
	
	return 0;
}
0