結果
問題 | No.430 文字列検索 |
ユーザー | rniya |
提出日時 | 2020-10-05 01:02:36 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 279 ms / 2,000 ms |
コード長 | 3,929 bytes |
コンパイル時間 | 1,681 ms |
コンパイル使用メモリ | 177,324 KB |
実行使用メモリ | 27,136 KB |
最終ジャッジ日時 | 2024-11-10 00:47:12 |
合計ジャッジ時間 | 3,093 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 279 ms
27,136 KB |
testcase_02 | AC | 6 ms
5,248 KB |
testcase_03 | AC | 6 ms
5,248 KB |
testcase_04 | AC | 2 ms
5,248 KB |
testcase_05 | AC | 2 ms
5,248 KB |
testcase_06 | AC | 1 ms
5,248 KB |
testcase_07 | AC | 2 ms
5,248 KB |
testcase_08 | AC | 269 ms
26,880 KB |
testcase_09 | AC | 2 ms
5,248 KB |
testcase_10 | AC | 15 ms
5,888 KB |
testcase_11 | AC | 51 ms
7,936 KB |
testcase_12 | AC | 52 ms
8,192 KB |
testcase_13 | AC | 54 ms
8,064 KB |
testcase_14 | AC | 42 ms
6,528 KB |
testcase_15 | AC | 31 ms
5,632 KB |
testcase_16 | AC | 10 ms
5,248 KB |
testcase_17 | AC | 8 ms
5,248 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; #define LOCAL #pragma region Macros typedef long long ll; #define ALL(x) (x).begin(),(x).end() const long long MOD=1000000007; // const long long MOD=998244353; const int INF=1e9; const long long IINF=1e18; const int dx[4]={1,0,-1,0},dy[4]={0,1,0,-1}; const char dir[4]={'D','R','U','L'}; template<typename T> istream &operator>>(istream &is,vector<T> &v){ for (T &x:v) is >> x; return is; } template<typename T> ostream &operator<<(ostream &os,const vector<T> &v){ for (int i=0;i<v.size();++i){ os << v[i] << (i+1==v.size()?"": " "); } return os; } template<typename T,typename U> ostream &operator<<(ostream &os,const pair<T,U> &p){ os << '(' << p.first << ',' << p.second << ')'; return os; } template<typename T,typename U> ostream &operator<<(ostream &os,const map<T,U> &m){ os << '{'; for (auto itr=m.begin();itr!=m.end();++itr){ os << '(' << itr->first << ',' << itr->second << ')'; if (++itr!=m.end()) os << ','; --itr; } os << '}'; return os; } template<typename T> ostream &operator<<(ostream &os,const set<T> &s){ os << '{'; for (auto itr=s.begin();itr!=s.end();++itr){ os << *itr; if (++itr!=s.end()) os << ','; --itr; } os << '}'; return os; } void debug_out(){cerr << '\n';} template<class Head,class... Tail> void debug_out(Head&& head,Tail&&... tail){ cerr << head; if (sizeof...(Tail)>0) cerr << ", "; debug_out(move(tail)...); } #ifdef LOCAL #define debug(...) cerr << " ";\ cerr << #__VA_ARGS__ << " :[" << __LINE__ << ":" << __FUNCTION__ << "]" << '\n';\ cerr << " ";\ debug_out(__VA_ARGS__) #else #define debug(...) 42 #endif template<typename T> T gcd(T x,T y){return y!=0?gcd(y,x%y):x;} template<typename T> T lcm(T x,T y){return x/gcd(x,y)*y;} template<class T1,class T2> inline bool chmin(T1 &a,T2 b){ if (a>b){a=b; return true;} return false; } template<class T1,class T2> inline bool chmax(T1 &a,T2 b){ if (a<b){a=b; return true;} return false; } #pragma endregion class RollingHash{ using u64=uint64_t; using u128=__uint128_t; static const uint64_t mod=(1ULL<<61)-1; const u64 base; vector<u64> hash,power; static inline u64 add(u64 a,u64 b){ if ((a+=b)>=mod) a-=mod; return a; } static inline u64 mul(u64 a,u64 b){ u128 c=(u128)a*b; return add(c>>61,c&mod); } public: static inline u64 generate_base(){ mt19937_64 mt(chrono::steady_clock::now().time_since_epoch().count()); uniform_int_distribution<u64> rand(2,RollingHash::mod-1); return rand(mt); } template<typename T> RollingHash(const T &s,u64 base):base(base){ int n=s.size(); hash.assign(n+1,0); power.assign(n+1,0); power[0]=1; for (int i=0;i<n;++i){ hash[i+1]=add(mul(hash[i],base),s[i]); power[i+1]=mul(power[i],base); } } u64 query(int l,int r) const { return add(hash[r],mod-mul(hash[l],power[r-l])); } int lcp(int a,int b) const { int len=min(hash.size()-a,hash.size()-b); int lb=0,ub=len; while (ub-lb>1){ int mid=(lb+ub)>>1; (query(a,a+mid)==query(b,b+mid)?lb:ub)=mid; } return lb; } u64 get(const string &t){ u64 res=0; for (int i=0;i<t.size();++i) res=add(mul(res,base),t[i]); return res; } }; int main(){ cin.tie(0); ios::sync_with_stdio(false); string S; int M; cin >> S >> M; int n=S.size(); RollingHash RH(S,RollingHash::generate_base()); map<uint64_t,int> mp; for (int j=0;j<=min(n,10);++j){ for (int i=0;i+j<=n;++i){ ++mp[RH.query(i,i+j)]; } } int ans=0; for (;M--;){ string C; cin >> C; ans+=mp[RH.get(C)]; } cout << ans << '\n'; }