usaco6.1-Cow XOR：trie树

  1 #include <iostream>
  2 #include <cstring>
  3 #include <vector>
  4 #include <stdio.h>
  5 #include <string>
  6 using namespace std;
  7 #define BIT_AT(x,n) ((x & (1<<n))>>n)
  8 
  9 const int maxnode = 807152;
 10 const int sigma_size = 2;
 11 
 12 // 字母表为全体小写字母的Trie
 13 struct Trie
 14 {
 15     int ch[maxnode][sigma_size]; // ch[i][j]表示i节点通过字母为j的边的连接下一节点的编号，没有即为0
 16     int val[maxnode];
 17     int sz; // 结点总数
 18     void clear()
 19     {
 20         sz = 1;    // 初始时只有一个根结点
 21         memset(ch[0], 0, sizeof(ch[0]));
 22     }
 23     int idx(char c)
 24     {
 25         return c - ‘0‘;    // 字符c的编号
 26     }
 27 
 28     // 插入字符串s，附加信息为v。注意v必须非0，因为0代表“本结点不是单词结点”
 29     // 附加信息为i，即第几个单词
 30     void insert(const char *s, int v)
 31     {
 32         int u = 0, n = strlen(s);
 33         for(int i = 0; i < n; i++)
 34         {
 35             int c = idx(s[i]);
 36             if(!ch[u][c])   // 结点不存在
 37             {
 38                 memset(ch[sz], 0, sizeof(ch[sz]));
 39                 val[sz] = 0;  // 中间结点的附加信息为0，因此clear时不用memset（）
 40                 ch[u][c] = sz++; // 新建结点
 41             }
 42             u = ch[u][c]; // 往下走
 43         }
 44         val[u] = v; // 字符串的最后一个字符的附加信息为v
 45     }
 46 
 47     int find_min(int x,int &min_i)
 48     {
 49         int Min=0;
 50         int u = 0;
 51         for(int i = 0; i <= 21; i++)
 52         {
 53             int tmp=BIT_AT(x,21-i);
 54             int c = (tmp^1);
 55             if(!ch[u][c])
 56                 c=tmp;
 57             u = ch[u][c];
 58             Min=Min*2+c;
 59         }
 60         min_i=val[u];
 61         return Min;
 62     }
 63 };
 64 //////////////////////////////////////////////////////////////////////////////////////
 65 //////////////////////////////////////////////////////////////////////////////////////
 66 Trie trie;
 67 // 将某个数的二进制转换成22位字符串
 68 string toString(int x)
 69 {
 70     string str;
 71     for(int i=21;i>=0;i--)
 72     {
 73         str.append(BIT_AT(x,i)==1?"1":"0");
 74     }
 75     return str;
 76 }
 77 
 78 int a[100010]={0};
 79 int b[100010]={0};
 80 
 81 int main()
 82 {
 83     freopen("cowxor.in","r",stdin);
 84     freopen("cowxor.out","w",stdout);
 85     int n;
 86     cin>>n;
 87     trie.clear();
 88     // 初始插入0
 89     trie.insert(toString(0).c_str(),0);
 90 
 91     int ans=-1,pos1=-1,pos2=-1;
 92     for(int i=1;i<=n;i++)
 93     {
 94         int t;
 95         scanf("%d",&t);
 96         b[i]=t;
 97         a[i]=a[i-1]^t;
 98         int min_i;
 99         int tmp=trie.find_min(a[i],min_i)^a[i];
100         if(ans<tmp/* || (ans==tmp && b[pos2]<t) || (ans==tmp && b[pos2]==t && pos2-pos1>i-min_i-1)*/)
101         {
102             ans=tmp;
103             pos1=min_i+1;
104             pos2=i;
105         }
106         trie.insert(toString(a[i]).c_str(),i);
107     }
108     printf("%d %d %d\n",ans,pos1,pos2);
109 
110     return 0;
111 }