ACwing 142.字首統計 143.最大異或對

LeBronGod發表於2020-10-04

在這裡插入圖片描述

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e6+10;
int trie[maxn][26],tot = 1,End[maxn];
void insert(char *str) {
	int len = strlen(str), p = 1;
	for(int i=0;i<len;i++) {
		int ch = str[i]-'a';
		if(trie[p][ch] == 0) trie[p][ch] = ++tot;
		p = trie[p][ch];
	}
	End[p]++; 
}
int search(char *str) {
	int len = strlen(str), p = 1, ans = 0;
	for(int i=0;i<len;i++) {
		p = trie[p][str[i]-'a'];
		if(!p) return ans;
		ans+=End[p];
	}
	return ans;
}
int main()
{
	int n,m;
	cin>>n>>m;
	char str[maxn];
	for(int i=1;i<=n;i++) {
		scanf("%s",str);
		insert(str);
	}
	for(int i=1;i<=m;i++) {
		scanf("%s",str);
		cout<<search(str)<<endl;	
	}
}

Trie樹

思路:

有插入和查詢兩個操作函式,字串快速檢索的多叉樹。
從根節點往下遍歷,選擇滿足條件的樹枝方向繼續往下,直至葉子節點或字串結束節點。最大異或對則將整型轉化為32位的二進位制數插入Trie樹,然後根據異或的性質遍歷相反的枝,沒有就往相同的方向,遍歷所有數取max就能得到最大值。

在這裡插入圖片描述

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e6+10;
int trie[maxn*31][2],tot,a[maxn];
void insert(int x) {
	int p = 0;
	for(int i=30;i>=0;i--) {
		int u = x>>i&1;
		if(trie[p][u] == 0) trie[p][u] = ++tot;
		p = trie[p][u];
	}
}
int search(int x) {
	int  p = 0,res = 0;
	for(int i=30;i>=0;i--) {
		int u = x>>i&1;
		if(trie[p][!u]) {//如果有相反的 
			p = trie[p][!u];
			res = res*2+1;
		} else {//沒有繼續走相同的 
			p = trie[p][u];
			res = res*2;
		}
	}
	return res;
}
int main()
{
	int n;
	cin>>n;
	for(int i=0;i<n;i++) {
		cin>>a[i];
		insert(a[i]);
	}
	int ans = 0;
	for(int i=0;i<n;i++) {
		ans = max(ans,search(a[i]));
	}
	cout<<ans<<endl;
}

相關文章