結果
問題 | No.430 文字列検索 |
ユーザー | fuppy_kyopro |
提出日時 | 2021-01-19 11:00:09 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 13 ms / 2,000 ms |
コード長 | 5,701 bytes |
コンパイル時間 | 1,974 ms |
コンパイル使用メモリ | 181,888 KB |
実行使用メモリ | 8,752 KB |
最終ジャッジ日時 | 2024-11-10 00:52:18 |
合計ジャッジ時間 | 2,630 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 12 ms
8,752 KB |
testcase_02 | AC | 10 ms
5,248 KB |
testcase_03 | AC | 8 ms
5,248 KB |
testcase_04 | AC | 2 ms
5,248 KB |
testcase_05 | AC | 1 ms
5,248 KB |
testcase_06 | AC | 1 ms
5,248 KB |
testcase_07 | AC | 1 ms
5,248 KB |
testcase_08 | AC | 3 ms
5,248 KB |
testcase_09 | AC | 2 ms
5,248 KB |
testcase_10 | AC | 2 ms
5,248 KB |
testcase_11 | AC | 13 ms
6,056 KB |
testcase_12 | AC | 12 ms
6,628 KB |
testcase_13 | AC | 13 ms
6,628 KB |
testcase_14 | AC | 10 ms
6,084 KB |
testcase_15 | AC | 10 ms
6,080 KB |
testcase_16 | AC | 10 ms
6,072 KB |
testcase_17 | AC | 10 ms
6,072 KB |
ソースコード
/* #pragma GCC target("avx2") #pragma GCC optimize("O3") #pragma GCC optimize("unroll-loops") //*/ #include <bits/stdc++.h> // #include <atcoder/all> using namespace std; // using namespace atcoder; #define DEBUG(x) cerr<<#x<<": "<<x<<endl; #define DEBUG_VEC(v) cerr<<#v<<":";for(int i=0;i<v.size();i++) cerr<<" "<<v[i]; cerr<<endl; #define DEBUG_MAT(v) cerr<<#v<<endl;for(int i=0;i<v.size();i++){for(int j=0;j<v[i].size();j++) {cerr<<v[i][j]<<" ";}cerr<<endl;} typedef long long ll; // #define int ll #define vi vector<int> #define vl vector<ll> #define vii vector< vector<int> > #define vll vector< vector<ll> > #define vs vector<string> #define pii pair<int,int> #define pis pair<int,string> #define psi pair<string,int> #define pll pair<ll,ll> template<class S, class T> pair<S, T> operator+(const pair<S, T> &s, const pair<S, T> &t) { return pair<S, T>(s.first + t.first, s.second + t.second); } template<class S, class T> pair<S, T> operator-(const pair<S, T> &s, const pair<S, T> &t) { return pair<S, T>(s.first - t.first, s.second - t.second); } template<class S, class T> ostream& operator<<(ostream& os, pair<S, T> p) { os << "(" << p.first << ", " << p.second << ")"; return os; } #define X first #define Y second #define rep(i,n) for(int i=0;i<(int)(n);i++) #define rep1(i,n) for(int i=1;i<=(int)(n);i++) #define rrep(i,n) for(int i=(int)(n)-1;i>=0;i--) #define rrep1(i,n) for(int i=(int)(n);i>0;i--) #define REP(i,a,b) for(int i=a;i<b;i++) #define in(x, a, b) (a <= x && x < b) #define all(c) c.begin(),c.end() void YES(bool t=true) {cout<<(t?"YES":"NO")<<endl;} void Yes(bool t=true) {cout<<(t?"Yes":"No")<<endl;} void yes(bool t=true) {cout<<(t?"yes":"no")<<endl;} void NO(bool t=true) {cout<<(t?"NO":"YES")<<endl;} void No(bool t=true) {cout<<(t?"No":"Yes")<<endl;} void no(bool t=true) {cout<<(t?"no":"yes")<<endl;} template<class T> bool chmax(T &a, const T &b) { if (a<b) { a = b; return 1; } return 0; } template<class T> bool chmin(T &a, const T &b) { if (a>b) { a = b; return 1; } return 0; } #define UNIQUE(v) v.erase(std::unique(v.begin(), v.end()), v.end()); const ll inf = 1000000001; const ll INF = (ll)1e18 + 1; const long double pi = 3.1415926535897932384626433832795028841971L; int popcount(ll t) { return __builtin_popcountll(t); } // int dx[4] = {1, 0, -1, 0}, dy[4] = {0, 1, 0, -1}; // int dx2[8] = { 1,1,0,-1,-1,-1,0,1 }, dy2[8] = { 0,1,1,1,0,-1,-1,-1 }; vi dx = {0, 1, 0, -1}, dy = {-1, 0, 1, 0}; // vi dx2 = { 1,1,0,-1,-1,-1,0,1 }, dy2 = { 0,1,1,1,0,-1,-1,-1 }; struct Setup_io { Setup_io() { ios_base::sync_with_stdio(0), cin.tie(0), cout.tie(0); cout << fixed << setprecision(25); } } setup_io; const ll MOD = 1000000007; // const ll MOD = 998244353; // #define mp make_pair //#define endl '\n' vi ans; template<int char_size> struct TrieNode { // 持たせたい好きな状態を持たせよう // 持たせたい好きな状態を持たせよう int next_idx[char_size]; // Trie木上での次の状態のインデックス,存在しない時は-1 vector<int> exist_char; // 続きのノードが存在する文字 vector<int> state_idx; // ちょうどこのノードと同じ文字列のインデックス(同じ文字列が複数回追加される時もあるので) TrieNode() { for (int i = 0; i < char_size; i++) next_idx[i] = -1; } }; template<int char_size, int initial_char> struct Trie { using Node = TrieNode<char_size>; vector<Node> nodes; int str_num=0; // 追加された文字列の数(!= ノードの数) vector<string> strs; // 追加された文字列とそのidの対応表,idが連番じゃない時はunordered_mapに変えて頑張って Trie() { nodes.push_back(Node()); } void add(string s, int str_idx, int node_idx, int id) { if (str_idx == s.size()) { nodes[node_idx].state_idx.push_back(id); return; } if (nodes[node_idx].next_idx[s[str_idx] - initial_char] == -1) { nodes[node_idx].next_idx[s[str_idx] - initial_char] = nodes.size(); nodes[node_idx].exist_char.push_back(s[str_idx] - initial_char); nodes.push_back(Node()); } add(s, str_idx+1, nodes[node_idx].next_idx[s[str_idx] - initial_char], id); } void add(string& s, int id=-1) { if (id == -1) id = str_num; add(s, 0, 0, id); str_num++; strs.push_back(s); } void query(string s, int str_idx, int node_idx) { for (int idx: nodes[node_idx].state_idx) { // sの今のprefixにちょうど一致する追加された文字列たちに対する処理 ans[idx]++; } if (str_idx == s.size()) { // ちょうどsと一致するノードが見つかったよ! return; } // これ以上sのprefixに共通するノードはない if (nodes[node_idx].next_idx[s[str_idx] - initial_char] == -1) return; query(s, str_idx + 1, nodes[node_idx].next_idx[s[str_idx] - initial_char]); } // sのprefixに対する何らかの処理をしたい時 void query(string s) { query(s, 0, 0); } // 根からdfsしたい時,次コンテストで使う時に書いてライブラリ化 void dfs() { } }; signed main() { string s; cin >> s; int m; cin >> m; Trie<26, 'A'> trie; rep (i, m) { string t; cin >> t; trie.add(t); } ans.resize(m); rep (i, s.size()) { string t = s.substr(i, 12); trie.query(t); } ll res = 0; rep (i, m) { res += ans[i]; } cout << res << endl; }