pta 公路村村通

现有村落间道路的统计数据表中，列出了有可能建设成标准公路的若干条道路的成本，求使每个村落都有公路连通所需要的最低成本。

输入格式:
输入数据包括城镇数目正整数N（≤1000）和候选道路数目M（≤3N）；随后的M行对应M条道路，每行给出3个正整数，分别是该条道路直接连通的两个城镇的编号以及该道路改建的预算成本。为简单起见，城镇从1到N编号。

输出格式:
输出村村通需要的最低成本。如果输入数据不足以保证畅通，则输出−1，表示需要建设更多公路。

输入样例:
6 15
1 2 5
1 3 3
1 4 7
1 5 4
1 6 2
2 3 4
2 4 6
2 5 2
2 6 6
3 4 6
3 5 1
3 6 1
4 5 10
4 6 8
5 6 3
输出样例:
12

毫无疑问，是最小生成树。这里用prime算法来写。选1为根节点。更新从与1邻边的节点的距离。每次选一个距离最小的节点，然后用这个节点去更新与他邻接的其他节点的路径长度

#include<stdio.h>
#define  inf 0x3f3f3f3f
int size=0;
typedef struct NODE node;
struct NODE{
	int cell;//节点是谁
	int key; 
};
void heapify(node *am,int i)
{
	int l,r,min;
	while(1)
	{
		l=i<<1;
		r=i<<1|1;
		if(l<=size&&am[l].key<am[i].key) min=l;
		else min=i;
		if(r<=size&&am[r].key<am[min].key) min=r;
		if(min==i) return;
		node t=am[min];
		am[min]=am[i];
		am[i]=t;
		i=min;
	}
}
node extract_min(node *am)
{
	node tep=am[1];
	am[1]=am[size--];
	heapify(am,1);
	return tep;	
}
void change(node *am,int i)
{
	while(i>1&&am[i>>1].key>am[i].key)
	{
		node tep=am[i>>1];
		am[i>>1]=am[i];
		am[i]=tep;
		i=i>>1;
	}
}
int findit(node *dis,int v)
{
	for(int i=1;i<=size;i++)
	{
		if(dis[i].cell==v)
		{
			return i;		
		}
	}
}
int cost[1001][1001]={0};
int main(void)
{
	node dis[1002];
	node tep[1002];
	int n,m;
	scanf("%d%d",&n,&m);
	size=n;
	for(int i=1;i<=n;i++)//初始化，dis是队列，tep保存结点的更新情况
	{
		dis[i].key=tep[i].key=inf;
		dis[i].cell=tep[i].cell=i;
		
	}
	for(int i=1;i<=m;i++)
	{
		int r1,r2,co;
		scanf("%d%d%d",&r1,&r2,&co);
		cost[r1][r2]=co;
		cost[r2][r1]=co;
	}
	int ans=0;
	dis[1].key=0;//初始时，设根结点1的路径长度为0
	int mark[1002]={0};//标记有木有在生成树里面
	int flag=0;
	while(size!=0)
	{
		node u=extract_min(dis);在队列中选一个结点，这个点是到生成树里面任意结点的距离最小
		if(u.key==inf)如果没有最小，说明不连通
		{
			flag=1;
			break;
		}
		mark[u.cell]=1;
		ans+=u.key;
		for(int v=1;v<=n;v++)//用临接链表更快，但也麻烦
		{
			if(cost[u.cell][v]!=0&&mark[v]==0&&cost[u.cell][v]<tep[v].key)
			连通与否  。。。。。。。是否在生成树里。。。能否通过中间结点来更新路径长度
			{
				int t=findit(dis,v);//找到结点v在队列里的位置
				tep[v].key=cost[u.cell][v];
				dis[t].key=cost[u.cell][v];
				change(dis,t);//更新队列
			}
		}
	}
	if(flag)printf("-1\n");
	else printf("%d\n",ans);
} 

上下两个算法几乎一样。上面是change函数，下面是heap函数。
上面算法第一次错，是因为extract之后，优先队列没有得到维护，所以即使后面有change函数，此优先队列也是不对的。

#include<stdio.h>
#define  inf 0x3f3f3f3f
int size=0;
typedef struct NODE node;
struct NODE{
	int cell;//节点是谁
	int key; 
};
void heapify(node *am,int i)
{
	int l,r,min;
	while(1)
	{
		l=i<<1;
		r=i<<1|1;
		if(l<=size&&am[l].key<am[i].key) min=l;
		else min=i;
		if(r<=size&&am[r].key<am[min].key) min=r;
		if(min==i) return;
		node t=am[min];
		am[min]=am[i];
		am[i]=t;
		i=min;
	}
}
void heap(node *am)
{
	for(int i=size>>1;i>=1;i--)
	{
		heapify(am,i);
	}
}
node extract_min(node *am)
{
	node tep=am[1];
	am[1]=am[size--];
//	heapify(am,1);
	return tep;	
}
int findit(node *dis,int v)
{
	for(int i=1;i<=size;i++)
	{
		if(dis[i].cell==v)
		{
			return i;		
		}
	}
	
}
int cost[1001][1001]={0};
int main(void)
{
	node dis[1002];
	node tep[1002];
	int n,m;
	scanf("%d%d",&n,&m);
	size=n;
	for(int i=1;i<=n;i++)
	{
		dis[i].key=tep[i].key=inf;
		dis[i].cell=tep[i].cell=i;
		
	}
	for(int i=1;i<=m;i++)
	{
		int r1,r2,co;
		scanf("%d%d%d",&r1,&r2,&co);
		cost[r1][r2]=co;
		cost[r2][r1]=co;
	}
	int ans=0;
	dis[1].key=0;
	int mark[1002]={0};
	int flag=0;
	while(size!=0)
	{
		node u=extract_min(dis);
		if(u.key==inf)
		{
			flag=1;
			break;
		}
		mark[u.cell]=1;
		ans+=u.key;
		for(int v=1;v<=n;v++)//节点 
		{
			if(cost[u.cell][v]!=0&&mark[v]==0&&cost[u.cell][v]<tep[v].key)
			{
				tep[v].key=cost[u.cell][v];
				dis[findit(dis,v)].key=cost[u.cell][v];
			}
		}
		heap(dis);
	}
	if(flag)printf("-1\n");
	else printf("%d\n",ans);
}