Submission #3106934
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; vector< int > words; 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) { 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->words.emplace_back(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); /// only from failure!! vector< int >& words = ch.second->words; words.insert(end(words), begin(failure->words), end(failure->words)); } } } 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]); for(int word : now->words) { res.emplace_back(i - dict[word].size() + 1, i, word); } } 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 | 2427 Byte |
Status | MLE |
Exec Time | 1379 ms |
Memory | 788320 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 | 2 ms | 512 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 | 73 ms | 23936 KB |
10_rand_01 | AC | 68 ms | 25728 KB |
10_rand_02 | AC | 83 ms | 25160 KB |
10_rand_03 | AC | 83 ms | 29056 KB |
10_rand_04 | AC | 40 ms | 17152 KB |
11_rand_00 | AC | 94 ms | 32204 KB |
11_rand_01 | AC | 97 ms | 31616 KB |
11_rand_02 | AC | 94 ms | 32204 KB |
11_rand_03 | AC | 98 ms | 31232 KB |
11_rand_04 | AC | 97 ms | 30848 KB |
20_fixed_length_00 | AC | 90 ms | 26596 KB |
20_fixed_length_01 | AC | 95 ms | 26592 KB |
20_fixed_length_02 | AC | 91 ms | 26592 KB |
21_fixed_length_00 | AC | 56 ms | 32896 KB |
21_fixed_length_01 | AC | 56 ms | 32896 KB |
21_fixed_length_02 | AC | 57 ms | 32768 KB |
22_fixed_length_00 | AC | 97 ms | 31568 KB |
22_fixed_length_01 | AC | 97 ms | 31568 KB |
22_fixed_length_02 | AC | 96 ms | 31568 KB |
23_fixed_length_00 | AC | 86 ms | 32640 KB |
23_fixed_length_01 | AC | 86 ms | 32640 KB |
23_fixed_length_02 | AC | 88 ms | 32640 KB |
24_fixed_length_00 | AC | 75 ms | 32256 KB |
24_fixed_length_01 | AC | 75 ms | 32256 KB |
24_fixed_length_02 | AC | 77 ms | 32256 KB |
30_mixed_length_00 | AC | 74 ms | 32460 KB |
30_mixed_length_01 | AC | 75 ms | 32456 KB |
30_mixed_length_02 | AC | 75 ms | 32456 KB |
30_mixed_length_03 | AC | 74 ms | 32456 KB |
30_mixed_length_04 | AC | 74 ms | 32460 KB |
90_challenge_00 | AC | 39 ms | 28200 KB |
90_challenge_01 | MLE | 1379 ms | 788320 KB |
90_challenge_02 | AC | 26 ms | 16684 KB |
90_challenge_03 | AC | 519 ms | 394724 KB |
90_challenge_04 | AC | 31 ms | 12344 KB |
90_challenge_05 | AC | 1 ms | 256 KB |