codeforse1285D. Dr. Evil Underscores(01字典树)

D. Dr. Evil Underscores

链接: https://codeforces.com/problemset/problem/1285/D.

思路

应该是挺简单的一个题，但是我考试的时候用的另一个小思路，哎，，做不出来就该换思路的。。
正解应该就是按照01字典树插入，然后按照像线段树那样的遍历（如果一个节点有两个子节点，那么对于结果来说，这个节点代表的位置必然是1，就要加上，否者就可以不是1）。当时我觉得这样可能会TLE，因为如果遇到两个的就分开便利，那不就是要遍历2^30？后来想了想才知道，只有1e5 个点，分开遍历最差不过是全部的点都遍历一遍，就是1e5 * 30。

---

我的思路是，想要寻找一个标志，代表它（一个节点）下面离他最近的“只有一个子节点的节点”，想了好久怎么表示，最后用的set，也没过，第一个点就wa了。也可能是我代码能力不行。。。感觉思路没啥毛病啊
反而正解写了一次就过了

AC代码

#include<iostream>
#include<algorithm>
#include<cstring>

using namespace std;
const int N = 1e5 + 1e3;
const int M = 3e6 + 1e3;

int n , tot = 0;
struct Node{
	int next[3];
}trie[M];

void insert(int num){
	int k = 0;
	for(int i = 31 ; i >= 0 ; i -- ){
		int to = ((num>>i)&1);
		if(!trie[k].next[to]) trie[k].next[to] = ++tot;
		k = trie[k].next[to];
	}
}
int query(int k , int num , int index){
	int ans = 0;
	if(trie[k].next[0]&&trie[k].next[1]){
		ans += (1<<index);
		ans += min(query(trie[k].next[0] , num , index-1) , query(trie[k].next[1] , num , index-1));
	}else if(trie[k].next[0]){
		ans += query(trie[k].next[0] , num , index-1);
	}else if(trie[k].next[1]){
		ans += query(trie[k].next[1] , num , index-1);
	}else{
		return 0;
	}
	return ans;
}
int main(){
	ios::sync_with_stdio(false);
	int i , j , num;
	cin>>n;
	for(i = 1 ; i <= n ; i ++ ){
		cin>>num;
		insert(num);
	}
	cout<<query(0,num,31)<<endl;
	return 0;
}