Trie——字典树
洛谷 P8306 字典树板子题
#include
using namespace std;
int n, q;
const int N = 3e6+10;
struct Trie{
int vim[N][150], idx;//邻接表, idx为下标
int cnt[N];
void insert(string s){//插入函数
int loc = 0;
for(int i = 0;i < s.size();i++){
if(!vim[loc][s[i]])vim[loc][s[i]] = ++idx;//标记未被使用过的
loc = vim[loc][s[i]];
cnt[loc]++;
}
}
int query(string s){//查询函数
int loc = 0;
for(int i = 0;i < s.size();i++){
if(!vim[loc][s[i]])return 0;//表示不存在这样的一个字符串的前缀与之匹配，直接返回0
loc = vim[loc][s[i]];
}
return cnt[loc];//返回有多少个以loc为前缀结尾的字符串的长度
}
void init(){//初始化函数
for(int i = 0;i <= idx;i++){
cnt[i] = 0;
for(int j = 0;j < 130;j++){
vim[i][j] = 0;
}
}
idx = 0;
}
}st;
void solve(){
st.init();
cin >> n >> q;
while(n--){
string s;
cin >> s;
st.insert(s);
}
while(q--){
string s;
cin >> s;
cout << st.query(s) << "\n";
}
}
int main(){
int _;
// _ = 1;
cin>>_;
while(_--){
solve();
}
return 0;
}
洛谷 P2580 于是他错误的点名开始了
hack数据是
输入
1 ad 1 a
输出
WRONG
自己测测自己的代码，其他数据没什么特殊的
#include
#define ios ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
using namespace std;
const int N = 3e5+10;
struct Trie{
int tr[N][30], idx = 0, cnt[N];
void insert(string s){
int p = 0;
for(auto it : s){
int u = it-'a';
if(!tr[p][u])tr[p][u] = ++idx;
p = tr[p][u];
}
cnt[p] ++;
}
int query(string s){
// cout << s << "\n";
int p = 0;
for(auto it : s){
int u = it-'a';
if(!tr[p][u])return 0;
p = tr[p][u];
}
return cnt[p];
}
}st;
int t, k;
map p;
int main(){
ios;
cin >> t;
while(t --){
string s;
cin >> s;
st.insert(s);
}
cin >> k;
while(k --){
string s;
cin >> s;
if(st.query(s)){
p[s] ++;
if(p[s] == 1)cout << "OK\n";
else cout << "REPEAT\n";
}
else cout << "WRONG\n";
}
return 0;
}
洛谷 P3879 [TJOI2010] 阅读理解
注意：该题要注意输出空格 去重等细节，还有输出空行的特判
#include
#define ios ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
using namespace std;
const int N = 5000010;
struct Trie
{
int tr[N][30], idx;
map > cnt;
void insert(string s, int num){
int p = 0;
for(auto it : s){
int u = it - 'a';
if(!tr[p][u])tr[p][u] = ++idx;
p = tr[p][u];
}
cnt[p].insert(num);
}
set query(string s){
int p = 0;
for(auto it : s){
int u = it-'a';
if(!tr[p][u])return {};
p = tr[p][u];
}
return cnt[p];
}
}st;
int t, n, m;
int main(){
ios;
cin >> t;
for(int i = 1;i <= t;i++){
cin >> n;
while( n -- ){
string s;
cin >> s;
st.insert(s, i);
}
}
cin >> m;
while(m -- ){
string s;
cin >> s;
set res = st.query(s);
if(res.empty()){
cout << "\n";continue;
}
for(auto it = res.begin();it != res.end();it++){
auto u = res.end();
cout << *it ;
if(it != --u)cout << " ";
else cout << "\n";
}
// cout << "\n";
}
return 0;
}
洛谷 U149703 最大异或对
板子题 秒了
#include
using namespace std;
const int N = 3100010;
int n;
struct Trie{
int vim[N][2], idx, cnt[N];
void insert(int x){
int p = 0;
for(int i = 31;i >= 0;i--){
int u = x >> i&1;
if(!vim[p][u])vim[p][u] = ++idx;
p = vim[p][u];
cnt[p]++;
}
}
int query(int x){
int p = 0, res = 0;
for(int i = 31;i >= 0;i--){
int u = x >> i&1;
if(vim[p][u^1]){
res += 1 << i;
p = vim[p][u^1];
}
else{
p = vim[p][u];
}
}
return res;
}
}st;
int main(){
cin >> n;
std::vector v(n+1);
for(int i = 1;i <= n;i++)cin>>v[i], st.insert(v[i]);
int ans = 0;
for(int i = 1;i <= n;i++){
ans = max(ans, st.query(v[i]));
}
cout << ans;
return 0;
}
Acwing 最大异或和
前缀和+贪心+Trie树
前缀和将问题转化为求最大异或对的问题。
#include
#define ios ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
using namespace std;
const int N = 3100010;
struct Trie
{
int tr[N][2], idx = 0, cnt[N];
void insert(int x){
int p = 0;
for(int i = 31;i >= 0;i--){
int u = x >> i & 1;
if(!tr[p][u])tr[p][u] = ++idx;
p = tr[p][u];
cnt[p] ++;
}
}
void remove(int x){
int p = 0;
for(int i = 31;i >= 0;i--){
int u = x >> i & 1;
p = tr[p][u], cnt[p] --;
}
}
int query(int x){
int p = 0, res = 0;
for(int i = 31;i >= 0;i--){
int u = x>>i&1;
if(cnt[tr[p][u^1]]){
res += 1 << i;
p = tr[p][u^1];
}
else{
p = tr[p][u];
}
}
return res;
}
}st;
int n, m;
int main(){
ios;
cin >> n >> m;
std::vector a(n+1, 0);
st.insert(0);
for(int i = 1;i <= n;i++)cin >> a[i], a[i] ^= a[i-1];
int ans = 0;
for(int i = 1;i <= n;i++){
if(i-m-1>=0)st.remove(a[i-m-1]);
ans = max(ans, st.query(a[i]));
// cout<