POJ - 1679 - The Unique MST （检验最小生成 4000 树是否唯一）

#include<vector>
#include<algorithm>
#include<iostream>
using namespace std;
int father[1005];
int n,m;
struct Edge
{
int u,v,w,id;
Edge(){}
Edge(int _u,int _v,int _w,int _id):u(_u),v(_v),w(_w),id(_id){}
bool operator < (const Edge & r) const{
return w<r.w;
}
bool operator == (const Edge & r) const {
if(id==r.id)    return true;
return false;
}
};
vector<Edge> e;
vector<int> mst;
int f(const int & x)
{
if(x==father[x])    return x;
father[x]=f(father[x]);
return father[x];
}
void Union(int x,int y)
{
x=f(x);y=f(y);
father[x]=y;
}
void init()
{
for(int i=1;i<=n;i++)   father[i]=i;
}
int get_MST(int id)
{
init();
int res=0;
for(int i=0;i<e.size();i++)
{
if(e[i].id==id) continue;
if(f(e[i].u)!=f(e[i].v))
{
res+=e[i].w;
Union(e[i].u,e[i].v);
}
}
return res;
}
bool iscon()
{
for(int i=2;i<=n;i++)
{
//cout<< f(i) << " " << f(i-1) <<  endl;
if(f(i)!=f(i-1))    return false;
}
return true;
}
int main()
{
ios::sync_with_stdio(false);cin.tie(0);
int T;
cin>>T;
while(T--)
{
mst.clear();
e.clear();
cin>>n>>m;
int u,v,w;
for(int i=0;i<m;i++)
{
cin>>u>>v>>w;
e.push_back(Edge(u,v,w,i));
}
sort(e.begin(),e.end());
int ans=0;
init();
for(int i=0;i<e.size();i++)
{
if(f(e[i].u)!=f(e[i].v))
{
ans+=e[i].w;
mst.push_back(e[i].id);
Union(e[i].u,e[i].v);
}
}
bool flag=true;
for(int i=0;i<mst.size();i++)
{
if(get_MST(mst[i])==ans&&iscon())
{
flag=false;
break;
}
}
if(flag)    cout << ans << endl;
else cout << "Not Unique!" << endl;
}
return 0;
}