1822: 佈線問題 Prim

1822: 佈線問題

時間限制: 1 Sec  內存限制: 64 MB
提交: 99  解決: 32
您該題的狀態：已完成
[提交][狀態][討論版]

題目描述

南陽理工學院要進行用電線路改造，現在校長要求設計師設計出一種佈線方式，該佈線方式需要滿足以下條件：
1、把所有的樓都供上電。
2、所用電線花費最少

輸入

第一行是一個整數n表示有n組測試數據。(n<5)
每組測試數據的第一行是兩個整數v,e.
v表示學校裏樓的總個數(v<=500)
隨後的e行裏，每行有三個整數a,b,c表示a與b之間如果建鋪設線路花費爲c(c<=100)。（哪兩棟樓間如果沒有指明花費，則表示這兩棟樓直接連通需要費用太大或者不可能連通）
隨後的1行裏，有v個整數,其中第i個數表示從第i號樓接線到外界供電設施所需要的費用。( 0<e<v*(v-1)/2 )
（樓的編號從1開始），由於安全問題，只能選擇一個樓連接到外界供電設備。
數據保證至少存在一種方案滿足要求。

輸出

每組測試數據輸出一個正整數,表示鋪設滿足校長要求的線路的最小花費。

樣例輸入

<span style="color:black">1
4 6
1 2 10
2 3 10
3 1 10
1 4 1
2 4 1
3 4 1
1 3 5 6</span>

樣例輸出

<span style="color:black">4</span>

提示

nyoj38

來源

nyoj圖論 

 

題目智能推薦

1345 1308 2389 1872 1384 1302 

水題，加油，相信自己

#include<iostream>
#include<cstdio>
#include<cstring> 
using namespace std;
#define range 502
#define inf 0x7fffffff
int map[range][range];
int dis[range];
bool vis[range];
int a,b;
int  prim()
{
	int i,j,sum=0;
	int pos=0;
	for(i=0;i<a;i++)
	dis[i]=map[pos][i];
	vis[pos]=true;
	for(i=0;i<a-1;i++)
	{
		int min=inf;
		for(j=0;j<a;j++)
		{
			if(!vis[j]&&dis[j]<min&&dis[j]!=0)
			{
				min=dis[j];
				pos=j;
			}
		}
		
		sum+=min;
		vis[pos]=true;
		for(j=0;j<a;j++)
		{
			if(!vis[j]&&dis[j]>map[pos][j])
			dis[j]=map[pos][j];
		}
	}
	
	return sum;
}
int main()
{
	int n,i,j;
	int d1,d2,dd;
	cin>>n;
	while(n--)
	{
		int min=inf;
		
		memset(vis,false,sizeof(vis));
		memset(dis,inf,sizeof(dis));
		cin>>a>>b;
		for(i=0;i<a;i++)
		{
			for(j=i;j<a;j++)
			map[i][j]=map[j][i]=inf;
		 } 
//		 for(i=0;i<a;i++)
//		 {
//		 	for(j=0;j<a;j++)
//			 {
//			 	printf("%d ",map[i][j]);
//			 }
//			 printf("\n"); 
//		 }
		 
		for(i=0;i<b;i++)
		{
			cin>>d1>>d2>>dd;
			map[d1-1][d2-1]=map[d2-1][d1-1]=dd;
		}
		
		for(i=0;i<a;i++)
		{
			cin>>dd;
			if(dd<min)
			min=dd;
		}
		
		cout<<prim()+min<<endl;
	}
	return 0;
}

另外，要注意，對數組初始化，用memset只能賦0，或-1.

而fill可以賦任何值。用法

int a[200];
fill(a, a+100, 1);