hihocoer - 1097（最小產生樹·Prim演算法）

標籤：記憶體   現在   描述   iostream   target   提示   hoc   orm   scan   

題目連結：

http://hihocoder.com/problemset/problem/1097

題目：

#1097 : 最小產生樹一·Prim演算法

時間限制:10000ms

單點時限:1000ms

記憶體限制:256MB

描述

最近，小Hi很喜歡玩的一款遊戲類比城市開放出了新Mod，在這個Mod中，玩家可以擁有不止一個城市了！

但是，問題也接踵而來——小Hi現在手上擁有N座城市，且已知這N座城市中任意兩座城市之間建造道路所需要的費用，小Hi希望知道，最少花費多少就可以使得任意兩座城市都可以通過所建造的道路互相到達（假設有A、B、C三座城市，只需要在AB之間和BC之間建造道路，那麼AC之間也是可以通過這兩條道路連通的）。

提示：不知道為什麼Prim演算法和Dijstra演算法很像呢Σ(っ °Д °;)っ 。

輸入

每個測試點（輸入檔案）有且僅有一組測試資料。

在一組測試資料中：

第1行為1個整數N，表示小Hi擁有的城市數量。

接下來的N行，為一個N*N的矩陣A，描述任意兩座城市之間建造道路所需要的費用，其中第i行第j個數為Aij，表示第i座城市和第j座城市之間建造道路所需要的費用。

對於100%的資料，滿足N<=10^3，對於任意i，滿足Aii=0，對於任意i, j滿足Aij=Aji, 0<Aij<10^4.

輸出

對於每組測試資料，輸出1個整數Ans，表示為了使任意兩座城市都可以通過所建造的道路互相到達至少需要的建造費用。

範例輸入

50 1005 6963 392 1182 1005 0 1599 4213 1451 6963 1599 0 9780 2789 392 4213 9780 0 5236 1182 1451 2789 5236 0 

範例輸出

4178

EmacsNormalVim

最小產生樹prime演算法 模板題

AC代碼:

#include <stdio.h>#include <string.h>#include <iostream>using namespace std;int road[1005][1005],vis[1005],dis[1005];const int inf = 0x3f3f3f3f; int main(){//    freopen("in.txt","r",stdin);    int n,sum,minroad,fre,j,i,q,a,b;    while(scanf("%d",&n)!=EOF)    {        for(i=1;i<=n;i++)            for(j=1;j<=n;j++)            scanf("%d",&road[i][j]);        scanf("%d",&q);        for(int i=1;i<=q;i++)//把已經建好的路徑修改為0        {            scanf("%d %d",&a,&b);            road[a][b]=0;            road[b][a]=0;        }         for(i=1;i<=n;i++)            dis[i]=road[1][i];        memset(vis,0,sizeof(vis));         vis[1]=1;        sum=0;        fre=1;        while(fre < n)        {            minroad = inf;            for(j=0,i=1;i<=n;i++)            {                if(vis[i]==0 && dis[i]<minroad)                {                    minroad = dis[i];                    j=i;                }            }            sum += minroad;            fre++;            vis[j]=1;            for(i=1;i<=n;i++)                if(vis[i]==0 && road[j][i]<dis[i])                    dis[i] = road[j][i];        }        printf("%d\n",sum);    } }

 

 

 

hihocoer - 1097（最小產生樹·Prim演算法）