HDU5119 - Happy Matt Friends
做法:拆成两堆数,分别暴力出两组的所有异或值A,B,枚举A, 将B全部插入Trie树,通过枚举的数每一位的值,确定异或后构成的新树,然后在新树上统计比m大的值的个数即可。
#include <bits/stdc++.h>
#define pb push_back
typedef long long ll;
const int N = 1e6 + 7;
using namespace std;
int n, m, a[42], b[42], numa, numb;
ll ans = 0;
vector<int> va;
struct node{
int ch[2], num;
void init() {
ch[0] = ch[1] = -1;
num = 0;
}
}T[N*20];
int cc, rt;
void ins(int x) {
int now = rt;
for(int i = 22; i >= 0; --i) {
int t = !!(x&(1<<i));
if(T[now].ch[t] == -1) {
T[now].ch[t] = ++cc;
T[cc].init();
}
++T[now].num;
now = T[now].ch[t];
}
++T[now].num;
}
int cal(int x) {
int now = rt, ff = 0, ans = 0;
for(int i = 22; i >= 0; --i) {
int t = !!(m&(1<<i));
int f = !!(x&(1<<i));
if(!t) {
if(T[now].ch[1^f]!=-1) ans += T[T[now].ch[1^f]].num;
now = T[now].ch[0^f];
}
else {
now = T[now].ch[1^f];
}
if(now == -1) break;
}
if(now != -1) ans+=T[now].num;
return ans;
}
int TT, CC = 0;
int main() {
memset(T,0,sizeof(T));
scanf("%d",&TT);
while(TT--) {
scanf("%d%d",&n,&m);
for(int i = 0; i < n; ++i) scanf("%d",&a[i]);
numa = n/2; numb = 0;
for(int i = numa; i < n; ++i) b[numb++] = a[i];
ans = 0;
va.clear();
for(int s = 0; s < (1<<numa); ++s) {
int tmp = 0;
for(int i = 0; i < numa; ++i) if(s&(1<<i)) tmp^=a[i];
va.pb(tmp);
}
rt = cc = 0;
rt = ++cc;
T[rt].init();
for(int s = 0; s < (1<<numb); ++s) {
int tmp = 0;
for(int i = 0; i < numb; ++i) if(s&(1<<i)) tmp^=b[i];
ins(tmp);
}
for(int i = 0; i < va.size(); ++i) ans += 1LL*cal(va[i]);
printf("Case #%d: %lld\n",++CC,ans);
for(int i = 1; i <= cc; ++i) T[i].init();
}
}