POJ-2421 最小生成树 算法复习计划

#include <cstring>
#include <string>
#include <fstream>
#include <iostream>
#include <iomanip>
#include <cstdio>
#include <cctype>
#include <algorithm>
#include <queue>
#include <set>
#include <vector>
#include <stack>
#include <ctime>
#include <cstdlib>
#include <functional>
#include <cmath>
using namespace std;
#define PI acos(-1.0)
#define maxn 1005
#define eps 1e-7
#define INF 0x7FFFFFFF
#define ff sqrt(5.0)
typedef long long ll;
int n,m,cnt,father[maxn];
struct node
{
int s;
int next;
int w;
}edge[maxn*maxn];//邻接表
int map[maxn][maxn];//邻接矩阵
int find(int x)//寻找根节点
{
int r=x;
while(father[r]!=r)
{
r=father[r];
}
return r;
}
bool unoin(int x,int y)//判断是否构成环
{
int fx=find(x);
int fy=find(y);
if(fx!=fy)//不构成环
{
father[fx]=fy;
return 1;
}
return 0;//构成环
}
void kruskal()//算法
{
int ans=0;
for(int i=0;i<cnt;i++)
{
if(unoin(edge[i].s,edge[i].next))
{
ans+=edge[i].w;
}
}
cout<<ans<<endl;
}
bool cmp(node a,node b)//从小到大排列
{
return a.w<b.w;
}
int main()
{
for(int i=0;i<=maxn;i++)
{
father[i]=i;
}
scanf("%d",&n);
cnt=0;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
scanf("%d",&map[i][j]);
}
}
for(int i=1;i<n;i++)
{
for(int j=i+1;j<=n;j++)
{
if(map[i][j]!=0)
{
edge[cnt].s=i;
edge[cnt].next=j;
edge[cnt++].w=map[i][j];
}
}
}
scanf("%d",&m);
int p,q;
while(m--)
{
scanf("%d %d",&p,&q);
for(int i=0;i<cnt;i++)
{
if((edge[i].s==p&&edge[i].next==q)||(edge[i].s==q&&edge[i].next==p))
{
edge[i].w=0;
break;
}
}
}
sort(edge,edge+cnt,cmp);
kruskal();
return 0;
}