杭电OJ1879 继续畅通工程求解答

Problem Description

省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可）。现得到城镇道路统计表，表中列出了任意两城镇间修建道路的费用，以及该道路是否已经修通的状态。现请你编写程序，计算出全省畅通需要的最低成本。

Input

测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( 1< N < 100 )；随后的 N(N-1)/2 行对应村庄间道路的成本及修建状态，每行给4个正整数，分别是两个村庄的编号（从1编号到N），此两村庄间道路的成本，以及修建状态：1表示已建，0表示未建。

当N为0时输入结束。

Output

每个测试用例的输出占一行，输出全省畅通需要的最低成本。

Sample Input

3
1 2 1 0
1 3 2 0
2 3 4 0
3
1 2 1 0
1 3 2 0
2 3 4 1
3
1 2 1 0
1 3 2 1
2 3 4 1
0

Sample Output

3
1
0

接下来是我的代码，是由prim算法改的 不知道为什么总是wrong，样例试了几个也都是对的，求大神解答；
#include<stdio.h>
#include<string.h>
#define max 0x3f3f3f
#define N 101
int vis[N],map[N][N],low[N],n,p;
int biaoji;
int prim()
{
int i,j,min,wei;

for(i=1;i<=n;i++)
	low[i]=map[p][i];
/*vis[1]=1;
low[1]=0;*/
int sum=0;
for(i=1;i<=n;i++)
{
	min=max;
	for(j=1;j<=n;j++)
	{
		if(vis[j]==0&&low[j]<min)
		{
			min=low[j];
			wei=j;
		}
	}
	if(min==max)
		break;
	vis[wei]=1;
	sum=sum+min;
	for(j=1;j<=n;j++)
		if(vis[j]==0&&map[wei][j]<low[j])
			low[j]=map[wei][j];	
}
return sum;	

}

int main()
{
int q,w,e,r;
while(~scanf("%d",&n))
{
if(n==0)
break;

	memset(vis,0,sizeof(vis));
	int flag=1;
	for(int as=1;as<=n*(n-1)/2;as++)
	{
		scanf("%d%d%d%d",&q,&w,&e,&r);
		map[q][w]=e;
		map[w][q]=e;
		if(r==0)
		{
			vis[q]=0;
			vis[w]=0;
			flag++;	
		}
		if(r==1)
		{
			vis[q]=1;
			vis[w]=1;
			flag=0;
			biaoji=q;
		}
	}
	if(flag==n*(n-1)/2+1)
	{
	p=1;
	vis[p]=1;
	low[p]=0;
	int www=prim();
	printf("%d\n",www);	
	}
	
	if(flag!=n*(n-1)/2+1)
	{
	p=biaoji;
	vis[p]=1;
	low[p]=0;
	int www=prim();
	printf("%d\n",www);		
	}		
}
return 0;

}