| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 1044771 | aufan | Selling RNA Strands (JOI16_selling_rna) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
// #define int long long
#define fi first
#define se second
using namespace std;
int32_t main()
{
ios_base::sync_with_stdio(false);
cin.tie(NULL);
int n, m;
cin >> n >> m;
vector<tuple<string, string, int>> v;
for (int i = 0; i < n; i++) {
string s;
cin >> s;
v.push_back({s + "]", "?", -1});
}
for (int i = 0; i < m; i++) {
string p, q;
cin >> p >> q;
reverse(q.begin(), q.end());
v.push_back({p + "$", q, i});
v.push_back({p + "a", q, i});
}
sort(v.begin(), v.end());
int id = 0;
vector<vector<int>> ids(2222222);
vector<vector<int>> ch(2222222, vector<int>(26));
auto upd = [&](string s, int i) {
int nw = 0;
reverse(s.begin(), s.end());
for (auto c : s) {
if (ch[nw][c - 'A'] == 0) ch[nw][c - 'A'] = ++id;
nw = ch[nw][c - 'A'];
ids[nw].push_back(i);
}
};
auto query = [&](string s, int x) {
int nw = 0;
for (auto c : s) {
if (ch[nw][c - 'A'] == 0) return 0ll;
nw = ch[nw][c - 'A'];
}
int res = (int)ids[nw].size() - (upper_bound(ids[nw].begin(), ids[nw].end(), x) - ids[nw].begin());
return res;
};
vector<int> l(m);
for (int i = 0; i < (int)v.size(); i++) {
auto &[p, q, j] = v[i];
if (p.back() == '$') {
l[j] = i;
}
}
int ptr = 0;
vector<int> ans(m);
for (int i = 0; i < (int)v.size(); i++) {
auto &[p, q, j] = v[i];
if (q == "?") {
p.pop_back();
upd(p, i);
} else if (p.back() == 'a') {
ans[j] = query(q, l[j]);
}
}
for (int i = 0; i < m; i++) cout << ans[i] << '\n';
return 0;
}