hdu6059 Kanade's trio(字典树)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=6059

题意：

给你一组数，让你在其中找到多少个三元组，ai，aj，ak满足ai^aj<=aj^ak。

解：

我们需要判定ai^aj<=aj^ak的话，那么ai和ak的前t-1位都相等的话，第t位不相等时，只有ai[t]==aj[t]，ak[t]!=aj[t]时成立。

那么我们利用字典树进行储存数据，然后记录每个数的相同的位数以及最高不同位，进行每位不相同出现的次数。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
using namespace std;
const int maxn=5e5+10;
const int node=maxn*31;
int cnt[31][2],num[30];
long long ext=0,ans;
int size;
struct Tree{
    int son[2];
    int cnt,ext;
}tree[node];
int a[maxn];
typedef long long ll;
void cal(int tmp,long long c){
    ans+= tree[tmp].cnt*1ll*(tree[tmp].cnt-1)/2;
    ext+=(c-tree[tmp].cnt)*1ll*tree[tmp].cnt-tree[tmp].ext;
}

void insert(int x)
{
    int tmp=0;
    for(int i=0;i<30;i++)
    {
        if(!tree[tmp].son[num[i]]){
        tree[tmp].son[num[i]]=++size;
        }
        if(tree[tmp].son[1-num[i]])
        {
            cal(tree[tmp].son[1-num[i]],cnt[i][1-num[i]]);
        }
        tmp = tree[tmp].son[num[i]];
        tree[tmp].cnt++;
        tree[tmp].ext += cnt[i][num[i]] - tree[tmp].cnt;
    }
    return ;
}

int main()
{
    int t,n;
    scanf("%d",&t);
    while(t--)
    {
       scanf("%d",&n);
       memset(tree,0,sizeof(tree));
       memset(cnt,0,sizeof(cnt));
       size=0,ans=0,ext=0;
       for(int i=1;i<=n;i++)
       {
           scanf("%d",&a[i]);
           int tmp = a[i];
           for(int j=29;j>=0;j--)
           {
               num[j] = tmp%2;
               cnt[j][tmp%2]++;
               tmp/=2;
           }
           insert(i);
       }
       printf("%lld\n",ans+ext);
    }
    return 0;
}