HDU5536 chip factory [01 사전 트리]

2635 단어
HDU5536 첫 번째 사전 트리2015 장춘역 ACM-ICPC의 한 문제는 생방송 O(n^3)에서 통과할 수 있다고 합니다.
인터넷 소코드를 보고 자기가 두드린 걸 깨달았어요...느끼해.코드를 알아내는 데 두 시간이 걸렸다.이렇게 복잡한 나무를 두드려서 기념하는 건 처음이에요.
#include<stdio.h>
#include<math.h>
#include<iostream>
#include<string.h>
#include<algorithm>
int a[1010];
using namespace std;
int root ;
int tot ;
struct node
{
    int size;
    int ch[2];
    node()
    {
        size = 0;
    }

};
node node[100005];
void INSERT(int x)
{
    int r = root;
    int c;
    for(int k = 30; k >=0 ; k--)
    {
        if(x&(1<<k))
            c = 1;
        else
            c = 0;
        if(!node[r].ch[c])
            node[r].ch[c] = ++tot;
        r = node[r].ch[c];
        node[r].size++;
    }
}
void DELETE(int x)
{
    int r = root;
    int c;
    for(int k = 30; k >=0; k--)
    {
        if(x&(1<<k))
            c = 1;
        else
            c = 0;
        node[node[r].ch[c]].size--;
        r = node[r].ch[c];
    }
}
int QUERY(int x)
{
    int r = root;
    int c;
    for(int k = 30; k >=  0; k--)
    {
        if(x&(1<<k))
            c = 1;
        else
            c = 0;
        if(c==1)
        {
            if(node[r].ch[0]&&node[node[r].ch[0]].size)
            {
                r = node[r].ch[0];
            }
            else
            {
                r = node[r].ch[1];
                x^=(1<<k);
            }
        }
        else
        {
            if(node[r].ch[1]&&node[node[r].ch[1]].size)
            {
                r = node[r].ch[1];
                x^=(1<<k);
            }
            else
            {
                r = node[r].ch[0];
            }
        }
    }
    return x;
}
int main()
{
    int T;
    int n;
    scanf("%d",&T);
    while(T--)
    {
        tot = 1; root = 1;
        scanf("%d",&n);
        for(int i = 0 ; i < n ; i++)
        {
            scanf("%d",&a[i]);
            INSERT(a[i]);
        }
        int ans = (a[0]+a[1])^a[2];
        for(int i = 0 ; i < n ; i++)
        {
            DELETE(a[i]);
            for(int j = i+1 ; j < n ;j++)
            {
                DELETE(a[j]);
                ans = max(ans,QUERY(a[i]+a[j]));
//                cout<<QUERY(a[i]+a[j]);
                INSERT(a[j]);
            }
            INSERT(a[i]);
        }
        cout<<ans<<endl;
        for(int i = 0 ; i <= tot;i++)
        {
            node[i].ch[0] = 0;
            node[i].ch[1] = 0;
            node[i].size = 0;
        }
    }
    return 0;
}

좋은 웹페이지 즐겨찾기