题目
难度:困难
类型:字典树
题意:数组中两两异或,找出超过m的异或和有多少个。
分析:使用trie树存储每个数的每个二进制位,然后每个数在trie树中进行深度搜索。如何比较与m的大小是关键。首先树根到叶子存的都是数值的高位,所以高位时,如果树中的第k层是0,而当前遍历的数的第k位是1,m中的第k位是0,此时根据异或的结果,此节点下的所有数异或以后都会比m大。因为在高位的时候已经异或得到一个1,而m中的此位是0。所以直接取trie树中预处理的结果,剪枝,无需进行往下的遍历。因为0和1有四种状态,所以判断时需要分类讨论。
详情可以查看这个链接
题目描述
给定整数m以及n各数字A1,A2,…An,将数列A中所有元素两两异或,共能得到n(n-1)/2个结果,请求出这些结果中大于m的有多少个。
输入描述:
第一行包含两个整数n,m.
第二行给出n个整数 A 1 , A 2 , … , A n A_1,A_2,\dots,A_n A1,A2,…,An。
数据范围
对于30%的数据, 1 ≤ n , m ≤ 1000 1 \le n, m \le 1000 1≤n,m≤1000
对于100%的数据, 1 ≤ n , m , A i ≤ 1 0 5 1 \le n, m, A_i\le 10^5 1≤n,m,Ai≤105
输出描述:
输出仅包括一行,即所求的答案
示例1
输入
复制
3 10
6 5 10
输出
复制
2
字典树版
#include <bits/stdc++.h>
using namespace std;
const int N = 1e5+5;
int arr[N];
int n, m;
//创建trie结构
struct Node{
int cnt;
Node* son[2];
Node(){
son[0] = nullptr;
son[1] = nullptr;
cnt = 1;
}
};
Node* root = new Node();
//插入数 由于数的范围小于N 所以最多16位
void insert(int x){
auto p = root;
for(int i = 16; i >= 0; i--){
int u = (x >>i)&1;
if(!p->son[u]) p->son[u] = new Node();
else p->son[u]->cnt++; //以当前点为前缀的数有多少个
p = p->son[u];
}
}
typedef long long ll;
ll query(int x, int idx, Node* p){
for(int i = idx; i >= 0; i--){
int u = (x>>i)&1;
int v = (m>>i)&1;
if(u==1 && v==1){
if(!p->son[0]) return 0;
p = p->son[0];
}else if(u==0 && v==1){
if(!p->son[1]) return 0;
p = p->son[1];
}else if(u==1 && v == 0){
ll val1 = p->son[0]? p->son[0]->cnt: 0;
ll val2 = p->son[1]?query(x, i-1, p->son[1]): 0;
return val1+val2;
}else if(u==0 && v == 0){
ll val1 = p->son[1]? p->son[1]->cnt: 0;
ll val2 = p->son[0]?query(x, i-1, p->son[0]): 0;
return val1+val2;
}
}
return 0; //此时i为-1
}
int main(){
scanf("%d%d", &n, &m);
for(int i = 0; i < n; i++) {
scanf("%d", &arr[i]);
insert(arr[i]);
}
ll res = 0;
for(int i = 0; i < n; i++){
res += query(arr[i], 16, root);
}
printf("%lld\n", res/2); //异或计算重复A^B 和B^A都算了一遍重复了
return 0;
}