Submission #3106992
Source Code Expand
#include <bits/stdc++.h> using namespace std; using ll = long long; // never fail to call build() !! // constructor() // add(string) // match(string) : [left, right], vec<tuple<int left, int right, int key>> in // right-order /// --- AhoCorasick Library {{{ /// struct AhoCorasick { struct Trie { unordered_map< char, Trie* > child; Trie* failure = nullptr; Trie* next = nullptr; // next failure that has the word // vector< int > words; int word = -1; Trie* add(char c) { if(child.count(c)) { return child[c]; } else { return child[c] = new Trie; } } Trie* go(char c) { if(child.count(c)) { return child[c]; } else { if(failure == nullptr) { // only top return this; } else { return failure->go(c); } } } }; Trie* top = new Trie; vector< string > dict; void add(string word) { Trie* now = top; for(size_t i = 0; i < word.size(); i++) { now = now->add(word[i]); } now->word = dict.size(); dict.emplace_back(word); } void build() { queue< Trie* > q; q.emplace(top); while(q.size()) { Trie* now = q.front(); q.pop(); for(pair< char, Trie* > ch : now->child) { q.emplace(ch.second); Trie* failure = ch.second->failure = now == top ? top : now->failure->go(ch.first); ch.second->next = failure->word >= 0 ? failure : failure->next; } } } vector< tuple< int, int, int > > match(string s) { Trie* now = top; vector< tuple< int, int, int > > res; for(size_t i = 0; i < s.size(); i++) { now = now->go(s[i]); Trie* tmp = now; do { int word = tmp->word; if(word >= 0) { res.emplace_back(i - dict[word].size() + 1, i, word); } tmp = tmp->next; } while(tmp != nullptr && tmp != top); } return res; } }; /// }}}--- /// int main() { std::ios::sync_with_stdio(false), std::cin.tie(0); int n; cin >> n; AhoCorasick ecasd; for(int i = 0; i < n; i++) { string t; cin >> t; ecasd.add(t); } ecasd.build(); string s; cin >> s; int m = s.size(); auto matches = ecasd.match(s); const int mod = 1e9 + 7; vector< ll > dp(m + 1); dp[0] = 1; for(auto p : matches) { int le, ri, key; tie(le, ri, key) = p; dp[ri + 1] = (dp[ri + 1] + dp[le]) % mod; } cout << dp[m] << endl; return 0; }
Submission Info
Submission Time | |
---|---|
Task | H - Separate String |
User | luma |
Language | C++14 (GCC 5.4.1) |
Score | 0 |
Code Size | 2565 Byte |
Status | MLE |
Exec Time | 1154 ms |
Memory | 788960 KB |
Judge Result
Set Name | All | ||||
---|---|---|---|---|---|
Score / Max Score | 0 / 100 | ||||
Status |
|
Set Name | Test Cases |
---|---|
All | 00_sample_00, 00_sample_01, 00_sample_02, 00_sample_03, 10_rand_00, 10_rand_01, 10_rand_02, 10_rand_03, 10_rand_04, 11_rand_00, 11_rand_01, 11_rand_02, 11_rand_03, 11_rand_04, 20_fixed_length_00, 20_fixed_length_01, 20_fixed_length_02, 21_fixed_length_00, 21_fixed_length_01, 21_fixed_length_02, 22_fixed_length_00, 22_fixed_length_01, 22_fixed_length_02, 23_fixed_length_00, 23_fixed_length_01, 23_fixed_length_02, 24_fixed_length_00, 24_fixed_length_01, 24_fixed_length_02, 30_mixed_length_00, 30_mixed_length_01, 30_mixed_length_02, 30_mixed_length_03, 30_mixed_length_04, 90_challenge_00, 90_challenge_01, 90_challenge_02, 90_challenge_03, 90_challenge_04, 90_challenge_05 |
Case Name | Status | Exec Time | Memory |
---|---|---|---|
00_sample_00 | AC | 1 ms | 256 KB |
00_sample_01 | AC | 1 ms | 256 KB |
00_sample_02 | AC | 1 ms | 256 KB |
00_sample_03 | AC | 1 ms | 256 KB |
10_rand_00 | AC | 66 ms | 23424 KB |
10_rand_01 | AC | 61 ms | 25600 KB |
10_rand_02 | AC | 84 ms | 24516 KB |
10_rand_03 | AC | 78 ms | 28928 KB |
10_rand_04 | AC | 37 ms | 17024 KB |
11_rand_00 | AC | 84 ms | 32076 KB |
11_rand_01 | AC | 88 ms | 31488 KB |
11_rand_02 | AC | 86 ms | 32204 KB |
11_rand_03 | AC | 90 ms | 31104 KB |
11_rand_04 | AC | 89 ms | 30592 KB |
20_fixed_length_00 | AC | 82 ms | 25916 KB |
20_fixed_length_01 | AC | 82 ms | 25916 KB |
20_fixed_length_02 | AC | 83 ms | 25916 KB |
21_fixed_length_00 | AC | 53 ms | 32896 KB |
21_fixed_length_01 | AC | 52 ms | 32896 KB |
21_fixed_length_02 | AC | 52 ms | 32896 KB |
22_fixed_length_00 | AC | 87 ms | 31440 KB |
22_fixed_length_01 | AC | 87 ms | 31440 KB |
22_fixed_length_02 | AC | 87 ms | 31440 KB |
23_fixed_length_00 | AC | 84 ms | 32640 KB |
23_fixed_length_01 | AC | 76 ms | 32640 KB |
23_fixed_length_02 | AC | 77 ms | 32640 KB |
24_fixed_length_00 | AC | 66 ms | 32256 KB |
24_fixed_length_01 | AC | 67 ms | 32256 KB |
24_fixed_length_02 | AC | 67 ms | 32256 KB |
30_mixed_length_00 | AC | 67 ms | 32344 KB |
30_mixed_length_01 | AC | 69 ms | 32344 KB |
30_mixed_length_02 | AC | 67 ms | 32344 KB |
30_mixed_length_03 | AC | 67 ms | 32344 KB |
30_mixed_length_04 | AC | 68 ms | 32344 KB |
90_challenge_00 | AC | 38 ms | 27176 KB |
90_challenge_01 | MLE | 1154 ms | 788960 KB |
90_challenge_02 | AC | 25 ms | 15788 KB |
90_challenge_03 | AC | 589 ms | 394724 KB |
90_challenge_04 | AC | 29 ms | 11372 KB |
90_challenge_05 | AC | 1 ms | 256 KB |