最近刚学的并查集所以用kruskal来试试最小生成树~
kruskal其实用几句话就能说完~
1.贪心所有边的权值,从小到大取值
2.取值时~将边权非0的两个顶点~进行并查操作~如果两个点的祖先不同...边权加入最小生成树...并且将两个点纳入同一个集合中
3.判断是否所有点都在同一个集合中
完毕~
下面上代码~这个代码应该可以作为模版了...但是并查集没有优化~所以复杂度约为0(n^3)但是比prim好一点
32ms水过...
mian()前的代码修改一下可以作为kruskal的模版...我再写一篇专门放模版吧~
#include <set>
#include <list>
#include <cmath>
#include <ctime>
#include <deque>
#include <queue>
#include <stack>
#include <cctype>
#include <cstdio>
#include <string>
#include <vector>
#include <cassert>
#include <cstdlib>
#include <cstring>
#include <sstream>
#include <iostream>
#include <algorithm>
using namespace std;
const int V = 101;
int father[V],map[V][V];
struct point
{
int s,v,rank;
}p[V*V];
int cmp(point a, point b)
{
return a.rank<b.rank;
}
int find(int x)
{
if(x!=father[x])
father[x]=find(father[x]);
return father[x];
}
void Union(int a,int b)
{
int x = find(a);
int y = find(b);
if(x==y) return ;
father[y]=x;
}
bool found(int n)
{
int x=find(0);
for(int i=0;i<n;i++)
if(find(i)!=x)
return false;
return true;
}
int kruskal(int map[][V],int n)
{
int i,j,cnt,mst=0;
for(i=0,cnt=0;i<n;i++)
{
father[i]=i;
for(j=0;j<n;j++)
{
p[cnt].s=i;
p[cnt].v=j;
p[cnt].rank = map[i][j];
cnt++;
}
}
sort(p,p+n*n,cmp);
for(i=0;i<n*n;i++)
{
if(p[i].rank!=0 && p[i].rank!=-1)
{
if(find(p[i].s)!=find(p[i].v))
mst+=p[i].rank;
Union(p[i].s,p[i].v);
if(found(n)==true)
return mst;
}else if(p[i].rank==-1)
{
Union(p[i].s,p[i].v);
if(found(n)==true)
return mst;
}
}
}
int main()
{
// freopen("in.txt","r",stdin);
int n;
scanf("%d",&n);
int m,i,j,p,q,cnt=0;
memset(map,0,sizeof(map));
for(i=0;i<n;i++)
for(j=0;j<n;j++)
scanf("%d",&map[i][j]);
cin>>m;
while(m--)
{
cin>>p>>q;
map[p-1][q-1]=-1;
}
cout<<kruskal(map,n)<<endl;
return 0;
}