1102 constructing roads.cpp

ACM 威士忌部分答案

#include <cstdio>
#include <string>
using namespace std;

int w[101][101];
int path[5001][2],n,q,num;

int prim_mintree(int s)
{
	int i,j,k,min,all,p,x,y;
	
	all=0;
	for(i=1;i<=n;i++)//防止出现回路 
		w[i][s]=0;
	path[0][0]=0;
	path[0][1]=s;
	num=1;
	for(i=1;i<n;i++)
	{
		min=0;
		for(j=0;j<num;j++)
		{
			p=path[j][1];
			for(k=1;k<=n;k++)
			{
				if(w[p][k]==0)
					continue;
				else if(w[p][k]<0)//此边以造 
				{
					min=0;
					x=p;y=k;
					break;
				}
				if(min==0 || w[p][k]<min)//连通边和为找到边的情况重复 
				{
					min=w[p][k];
					x=p;y=k;
				}
			}//for k
			if(w[p][k]<0)//注意当以连通，直接推出 
				break;
		}//for j
		for(j=1;j<=n;j++)
			w[j][y]=0;
		all+=min;
		path[num][0]=x;
		path[num][1]=y;
		num++;
	}//for i
	return all;
}

int main()
{
	int i,j,a,b,t,p;
	
	while( scanf("%d",&n)!=EOF && n>=3)
	{
		for(i=1;i<=n;i++)
			for(j=1;j<=n;j++)
			{
				scanf("%d",&w[i][j]);
			}

		scanf("%d",&q);
		b=1;num=0;
		for(i=0;i<q;i++)
		{
			scanf("%d %d",&a,&b);
			w[a][b]=-w[a][b];
			w[b][a]=-w[b][a];
		}
		
		printf("%d\n", prim_mintree(b) );
	}
	return 0;
}