HDU 6059 Kanade's trioTrie + 计数

xiaoxiao2021-02-28  103

传送门:HDU6059

题意:给定一个序列a,求有多少个三元组(i,j,k) 满足1<=i<j<k<=n  且  a[i] xor a[j] < a[j] xor a[k]?

思路:

利用字典树维护前 k-1 个数。当前处理第 k 个数。

显然对于 k 与 i 的最高不相同位  kp  与  ip  :

当  ip=0  ,  kp=1  时,该最高不相同位之前的  ihigher=khigher  。则  jhigher  可以为任意数,均不对 i, k 更高位(指最高不相同位之前的高位,后同)的比较产生影响。而此时  jp  位必须为 0 才可保证不等式  (AiAj)<(AjAk)  成立。

当  ip=1,kp=0  时, jp  位必须为 1 ,更高位任意。

故利用数组 cnt[31][2] 统计每一位为 0 ,为 1 的有多少个。在字典树插入第 k 个数时,同时统计最高不相同位,即对于每次插入的 p 位为 num[p] (取值 0 或 1),在同父节点对应的 1-num[p] 为根子树的所有节点均可作为 i 来寻找 j 以获取对答案的贡献。其中又仅要求  jp  与  ip  ( ip  值即 1-num[p]) 相同,故  jp  有 cnt[p][ 1-num[p] ] 种取值方案。

但是,同时需要注意 i 与 j 有在 A 数组的先后关系 ( i<j ) 需要保证。故在字典树中额外维护一个 ng 点,记录将每次新加入的点与多少原有点可构成 i, j 关系(注意不是j,i关系)。在后续计算贡献时去掉。

以上转自: 点击打开链接

总的思想就是将所有数转化为二进制插到字典树当中去,先枚举A[k],在将A[k]插入的过程中动态的枚举A[k]和A[i]的最高不同位,然后利用字典树动态的计数,关键是ng的计数不好理解,需要画图体会体会。

代码:

#include<bits/stdc++.h> #define ll long long #define pb push_back #define fi first #define se second #define pi acos(-1) #define inf 0x3f3f3f3f #define lson l,mid,rt<<1 #define rson mid+1,r,rt<<1|1 #define rep(i,x,n) for(int i=x;i<n;i++) #define per(i,n,x) for(int i=n;i>=x;i--) using namespace std; typedef pair<int,int>P; const int MAXN = 500010; int gcd(int a,int b){return b?gcd(b,a%b):a;} ll ans; struct node{ int son[2]; int cnt, ng; }Trie[MAXN * 32]; int cnt[32][2]; int dig[32], sz; void calc(int id, int sum) { ans += 1ll * Trie[id].cnt * (Trie[id].cnt - 1) / 2; // 计算a[i]高位和a[j]高位相同的所有可能组合(i<j) ans += 1ll * Trie[id].cnt * (sum - Trie[id].cnt);// 计算a[i]高位和[j]高位不同的所有可能组合(i有可能大于j) ans -= Trie[id].ng;// 去掉i>j的情况 } void insert() { int tmp = 0; for(int i = 0; i < 30; i++) { if(!Trie[tmp].son[dig[i]]) Trie[tmp].son[dig[i]] = ++sz; if(Trie[tmp].son[dig[i] ^ 1]) calc(Trie[tmp].son[dig[i] ^ 1], cnt[i][dig[i] ^ 1]); tmp = Trie[tmp].son[dig[i]]; Trie[tmp].cnt++; Trie[tmp].ng += cnt[i][dig[i]] - Trie[tmp].cnt; } } int main() { int T, n, t; cin >> T; while(T--) { ans = 0; sz = 0; memset(cnt, 0, sizeof(cnt)); scanf("%d", &n); memset(Trie, 0, n * 32 * sizeof(node));//只将可能用到的部分初始化就好了,否则时间增加的飞起 for(int i = 0; i < n; i++) { scanf("%d", &t); for(int i = 29; i >= 0; i--) { cnt[i][t & 1]++; dig[i] = t & 1; t >>= 1; } insert(); } cout << ans << endl; } return 0; }

转载请注明原文地址: https://www.6miu.com/read-51200.html

最新回复(0)