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<