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

比赛的时候常常做出来的题目都是模拟和水题或者是DP什么的，复习一下算法以免忘掉

基本是裸地最小生成树，给出矩阵转化为边，然后把已建好的边的权值改为0，最后用kruskal就好了。

#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;
}