poj3764 The xor-longest Path 字典树与异或

这道题的主要突破点就是从1->n上边权异或上1->m上边权的异或等于m->n上边权的异或。

剩下的就是一个经典的字典树求异或最大值了。

按道理我该睡了，但是这道题的一个bug被解决了，一定要记录一下。

#include<stdio.h>
#include<algorithm>
#include<string.h>

using namespace std;

const int N = 100000+50;

int n, cnt, sz;
int fir[N];
int ones[N];

struct Trie {
    int next[2];
    int cnt;
    void init() {
        memset(next, -1, sizeof(next));
    }
};

Trie trie[N*30];

struct Edge {
    int v, w;
    int next;
};

Edge edge[N<<1];

void init() {
    cnt = 1;
    sz = 1;
    memset(fir, -1, sizeof(fir));
    trie[0].init();
}

void addEdge(int u, int v, int w) {
    edge[cnt].v = v;
    edge[cnt].w = w;
    edge[cnt].next = fir[u];
    fir[u] = cnt++;
}

void myinsert(int x) {
    int p = 0;
    for(int i = 30; i >= 0; i--) {
        int ch;
        if(x&(1<<i))
            ch = 1;
        else ch = 0;
        if(trie[p].next[ch] == -1) {
            trie[p].next[ch] = sz;
            trie[sz++].init();
        }
        p = trie[p].next[ch];
    }
}

int myquery(int x) {
    int p = 0, ret = 0;
    for(int i = 30; i >= 0; i--) {
        int ch;
        if(x&(1<<i))
            ch = 0;
        else ch = 1;
        if(trie[p].next[ch] != -1) {
            ret |= (1 << i);
            p = trie[p].next[ch];
        } else p = trie[p].next[!ch];
    }
    return ret;
}

void dfs(int u, int val, int pre) {
    ones[u] = val;
    for(int i = fir[u]; i != -1; i = edge[i].next) {
        int v= edge[i].v;
        if(v == pre) continue;
        dfs(v, val ^ edge[i].w, u);
    }
}

int main() {

    while(~scanf("%d", &n)) {
        init();
        for(int i = 0; i < n-1; i++) {
            int u, v, w;
            scanf("%d%d%d", &u, &v, &w);
            addEdge(u, v, w);
            addEdge(v, u, w); //①
        }
        dfs(0, 0, -1);
        int ans = -1;
        for(int i = 0; i < n; i++) {
            myinsert(ones[i]);
            int q = myquery(ones[i]);
            ans = max(ans, q);
        }
        printf("%d\n", ans);
    }
    return 0;
}

在①处，我的代码最初写成了 addEdge[v, u, w]（注意，这里是中括号），交上去无限WA， 在CB中还不太明显。

纳闷它居然不报编译错误，有空要看看原因。以后做题要注意了。

这道题第二个问题是dfs()的时候我把 ^ 写成了 |，居然过了，不知道是不是数据有点水。