hdu6625 three arrays 01字典树

题意:有两个数组a和b,可以随意调换数组中的元素的顺序,令 c[ i ] = a[ i ] ^ b[ i ],使c的字典序最小。

思路:每个数按二进制从高位到低位看作一个字符串,对a和b数组分别建一棵01字典树,并记录每个结点出现的次数。然后同时遍历这两棵字典树,优先走当前位相同的(0和0,1和1),这样一定是最优的,因为每个数是由二进制从高位到低位排的,异或可以抵消高位。

求出来后还要再排个序。

代码:

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 typedef long long ll;
 4 const int N=1e5+10;
 5 
 6 int n;
 7 int ans[N];
 8 
 9 struct Tire{
10     int t[N*30][2],num[N*30];
11     int tot,root;
12     int New(){
13         t[tot][0]=t[tot][1]=0;
14         num[tot]=0;
15         return tot++;
16     }
17     void init(){
18         tot=0;
19         root=New();
20     }
21     void insert(int x){
22         int p=root;
23         for(int i=29;i>=0;i--){
24             int k=(x>>i)&1;
25             if(!t[p][k])
26                 t[p][k]=New();
27             p=t[p][k];
28             num[p]++;
29         }
30     }
31 }; 
32 Tire A,B;
33 
34 void solve(){
35     for(int i=1;i<=n;i++){
36         int p1=0,p2=0;
37         ans[i]=0;
38         for(int j=29;j>=0;j--){
39             if(A.num[A.t[p1][0]]&&B.num[B.t[p2][0]]){
40                 p1=A.t[p1][0];
41                 p2=B.t[p2][0];
42                 A.num[p1]--,B.num[p2]--;
43             }
44             else if(A.num[A.t[p1][1]]&&B.num[B.t[p2][1]]){
45                 p1=A.t[p1][1];
46                 p2=B.t[p2][1];
47                 A.num[p1]--,B.num[p2]--;
48             }
49             else if(A.num[A.t[p1][0]]&&B.num[B.t[p2][1]]){
50                 p1=A.t[p1][0];
51                 p2=B.t[p2][1];
52                 A.num[p1]--,B.num[p2]--;
53                 ans[i]+=(1<<j);
54             }
55             else if(A.num[A.t[p1][1]]&&B.num[B.t[p2][0]]){
56                 p1=A.t[p1][1];
57                 p2=B.t[p2][0];
58                 A.num[p1]--,B.num[p2]--;
59                 ans[i]+=(1<<j);
60             }
61         }
62     }
63 }
64 
65 int main()
66 {
67     int T;
68     scanf("%d",&T);
69     while(T--){
70         A.init(),B.init();
71         scanf("%d",&n);
72         for(int i=1;i<=n;i++){
73             int x;
74             scanf("%d",&x);
75             A.insert(x);
76         }
77         for(int i=1;i<=n;i++){
78             int x;
79             scanf("%d",&x);
80             B.insert(x);
81         }
82         solve();
83         sort(ans+1,ans+n+1);
84         for(int i=1;i<=n;i++){
85             printf("%d%c",ans[i],i==n? '\n':' ');
86         }
87     }
88     return 0;
89 }

 

posted @ 2020-08-18 10:41  --HY--  阅读(169)  评论(0编辑  收藏  举报