#include<bits/stdc++.h>
using namespace std;
typedef pair<int,int> pii;
typedef pair<pii,pii> piipii;
const int N = 1e5;
const int MULT = 23;
inline int getNum(char x){
return x-'A';
}
inline pii getHash(string &s, int len){
int ret = 0;
for(int i=0;i<len;i++){
ret *= MULT;
ret += getNum(s[i]);
}
return make_pair(ret, len);
}
unordered_map<int,int> memoDep[N+5];
map<int,int> memoBel[N+5];
string li[N+5];
int n,m;
string pre[N+5], suf[N+5];
vector<int> isi[N+5];
vector<int> dep, bel;
map<piipii, int> cnt;
int main(){
ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
cin >> n >> m;
for(int i=1;i<=n;i++){
cin >> li[i];
}
for(int i=1;i<=m;i++){
cin >> pre[i] >> suf[i];
reverse(suf[i].begin(), suf[i].end());
isi[pre[i].size()].push_back(suf[i].size());
dep.push_back(pre[i].size());
bel.push_back(suf[i].size());
cnt[make_pair(getHash(pre[i], pre[i].size()), getHash(suf[i], suf[i].size()))] = 0;
}
sort(dep.begin(), dep.end());
sort(bel.begin(), bel.end());
for(int i=1;i<=n;i++){
int tmp = 0;
int prev = 0;
for(auto x: dep){
if(x > li[i].size()) break;
for(;prev<x;prev++){
tmp *= MULT;
tmp += getNum(li[i][prev]);
}
memoDep[i][x] = tmp;
}
}
for(int i=1;i<=n;i++){
int tmp = 0;
int prev = 0;
string tmps = li[i];
reverse(tmps.begin(), tmps.end());
for(auto x: bel){
if(x > li[i].size()) break;
for(;prev<x;prev++){
tmp *= MULT;
tmp += getNum(tmps[prev]);
}
memoBel[i][x] = tmp;
}
}
for(int i=1;i<=n;i++){
for(auto cand: dep){
if(cand > li[i].size()) break;
for(auto cand2: isi[cand]){
if(cand2 > li[i].size()) continue;
piipii tmp = make_pair(make_pair(memoDep[i][cand], cand), make_pair(memoBel[i][cand2], cand2));
if(cnt.find(tmp) != cnt.end()) cnt[tmp]++;
}
}
}
for(int i=1;i<=m;i++){
cout << cnt[make_pair(getHash(pre[i], pre[i].size()), getHash(suf[i], suf[i].size()))] << endl;
}
return 0;
}
Compilation message
selling_rna.cpp: In function 'int main()':
selling_rna.cpp:55:9: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
55 | if(x > li[i].size()) break;
| ~~^~~~~~~~~~~~~~
selling_rna.cpp:70:9: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
70 | if(x > li[i].size()) break;
| ~~^~~~~~~~~~~~~~
selling_rna.cpp:81:12: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
81 | if(cand > li[i].size()) break;
| ~~~~~^~~~~~~~~~~~~~
selling_rna.cpp:83:14: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
83 | if(cand2 > li[i].size()) continue;
| ~~~~~~^~~~~~~~~~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
19 ms |
22228 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
547 ms |
74560 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
1584 ms |
23260 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
19 ms |
22228 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |