BZOJ 3943 [Usaco2015 Feb]SuperBull：最大生成树

题目链接：http://www.lydsy.com/JudgeOnline/problem.php?id=3943

题意：

　　有n只队伍，每个队伍有一个编号a[i]。

　　每场比赛有两支队伍参加，然后选一支队伍淘汰。共进行n-1场比赛，然后比赛结束。

　　若某场比赛是队伍i和j参加，则该场比赛的得分为a[i] xor a[j]。

　　问最大的总得分。

题解：

　　每两支队伍(i,j)连一条边，边权为a[i] xor a[j]。

　　然后求最大生成树即可。

AC Code:

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <vector>
#define MAX_N 2005

using namespace std;

struct Edge
{
    int sour;
    int dest;
    long long len;
    Edge(int _sour,int _dest,long long _len)
    {
        sour=_sour;
        dest=_dest;
        len=_len;
    }
    Edge(){}
    friend bool operator < (const Edge &a,const Edge &b)
    {
        return a.len>b.len;
    }
};

int n;
int par[MAX_N];
long long ans;
long long a[MAX_N];
vector<Edge> edge;

void read()
{
    cin>>n;
    for(int i=0;i<n;i++)
    {
        cin>>a[i];
    }
}

void build_graph()
{
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<i;j++)
        {
            edge.push_back(Edge(i,j,a[i]^a[j]));
        }
    }
}

void init_union_find()
{
    for(int i=0;i<n;i++)
    {
        par[i]=i;
    }
}

int find(int x)
{
    return par[x]==x?x:par[x]=find(par[x]);
}

void unite(int x,int y)
{
    int px=find(x);
    int py=find(y);
    if(px==py) return;
    par[px]=py;
}

bool same(int x,int y)
{
    return find(x)==find(y);
}

long long kruskal()
{
    init_union_find();
    sort(edge.begin(),edge.end());
    int cnt=0;
    long long res=0;
    for(int i=0;i<edge.size();i++)
    {
        Edge temp=edge[i];
        if(!same(temp.sour,temp.dest))
        {
            cnt++;
            res+=temp.len;
            unite(temp.sour,temp.dest);
        }
    }
    return cnt==n-1?res:-1;
}

void solve()
{
    build_graph();
    ans=kruskal();
}

void print()
{
    cout<<ans<<endl;
}

int main()
{
    read();
    solve();
    print();
}