Submission #3107152
Source Code Expand
#include <bits/stdc++.h> using namespace std; using ll = long long; // never fail to call build() !! // constructor() // add(string) // match(string, f) : [left, right], f(int left, int right, int key) in right-asc left-desc 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; } } } void match(string s, const function<void(int, int, int)> &f) { 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; while(tmp != top) { int word = tmp->word; if(word >= 0) { f(i - dict[word].size() + 1, i, word); } tmp = tmp->failure; // } while(tmp != nullptr && tmp != top); } } } }; /// }}}--- /// 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(); const int mod = 1e9 + 7; vector< ll > dp(m + 1); dp[0] = 1; ecasd.match(s, [&](int le, int ri, int key) { 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 | 2532 Byte |
Status | TLE |
Exec Time | 2103 ms |
Memory | 29696 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 | 21248 KB |
10_rand_01 | AC | 64 ms | 23168 KB |
10_rand_02 | AC | 73 ms | 22016 KB |
10_rand_03 | AC | 73 ms | 26240 KB |
10_rand_04 | AC | 37 ms | 15488 KB |
11_rand_00 | AC | 84 ms | 29004 KB |
11_rand_01 | AC | 91 ms | 28416 KB |
11_rand_02 | AC | 84 ms | 29004 KB |
11_rand_03 | AC | 94 ms | 28032 KB |
11_rand_04 | AC | 96 ms | 27648 KB |
20_fixed_length_00 | AC | 81 ms | 23420 KB |
20_fixed_length_01 | AC | 83 ms | 23420 KB |
20_fixed_length_02 | AC | 81 ms | 23420 KB |
21_fixed_length_00 | AC | 53 ms | 29696 KB |
21_fixed_length_01 | AC | 52 ms | 29696 KB |
21_fixed_length_02 | AC | 51 ms | 29696 KB |
22_fixed_length_00 | AC | 86 ms | 28496 KB |
22_fixed_length_01 | AC | 90 ms | 28496 KB |
22_fixed_length_02 | AC | 86 ms | 28496 KB |
23_fixed_length_00 | AC | 76 ms | 29440 KB |
23_fixed_length_01 | AC | 75 ms | 29440 KB |
23_fixed_length_02 | AC | 75 ms | 29440 KB |
24_fixed_length_00 | AC | 64 ms | 29184 KB |
24_fixed_length_01 | AC | 65 ms | 29184 KB |
24_fixed_length_02 | AC | 65 ms | 29184 KB |
30_mixed_length_00 | AC | 67 ms | 29272 KB |
30_mixed_length_01 | AC | 67 ms | 29272 KB |
30_mixed_length_02 | AC | 66 ms | 29272 KB |
30_mixed_length_03 | AC | 66 ms | 29272 KB |
30_mixed_length_04 | AC | 67 ms | 29272 KB |
90_challenge_00 | TLE | 2103 ms | 2944 KB |
90_challenge_01 | AC | 494 ms | 1536 KB |
90_challenge_02 | TLE | 2103 ms | 2944 KB |
90_challenge_03 | AC | 286 ms | 1536 KB |
90_challenge_04 | AC | 25 ms | 4532 KB |
90_challenge_05 | AC | 1 ms | 256 KB |