Hihocoder 之 #1097 : 最小生成树一·Prim算法 （用vector二维 模拟邻接表，进行prim()生成树算法, *【模板】）

#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，表示为了使任意两座城市都可以通过所建造的道路互相到达至少需要的建造费用。

样例输入
5
0 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

vector可变长数组可以模拟邻接表，但是这种存储结构用于处理多组数据时，一定要把vector数组全部清空
否则，必然WA！

代码：
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <vector>
#include <iostream>
#define INF 2100000000
using namespace std;
struct node
{
    int dd;
    int w;
}t;

vector<node>q[1001];
int dis[1001];
bool vis[1001];
void Vector_prim(int n)
{
    int i, j;
    vector<node>::iterator it;
    memset(vis, false, sizeof(vis));
    memset(dis, INF, sizeof(dis));
    for(it=q[1].begin(); it!=q[1].end(); it++ )
    {
        dis[it->dd]=it->w;
    }
    vis[1]=true;
    int pos;
    int sum=0;

    for(i=1; i<n; i++ )
    {
        int mm=INF;
        for(j=1; j<=n; j++)
        {
            if(!vis[j] && dis[j]<mm )
            {
                mm=dis[j];
                pos=j;
            }
        }
        sum+=mm;
        vis[pos]=true;
        for(it=q[pos].begin(); it!=q[pos].end(); it++ )
        {
            if( !vis[it->dd] && it->w <dis[it->dd] )
            {
                dis[it->dd]=it->w;
            }
        }
    }
    printf("%d\n", sum );
}

int main()
{
    int n;
    int i, j;
    int w;
    scanf("%d", &n);
    for(i=0; i<=n; i++)
    {
        q[i].clear();
    }
    for(i=1; i<=n; i++)
    {
        for(j=1; j<=n; j++)
        {
            scanf("%d", &w);
            if(i!=j)
            {
                t.dd=j;
                t.w=w;
                q[i].push_back(t);
            }
        }
    }
    Vector_prim(n);
    return 0;
}

 

posted @ 2015-01-18 11:35  我喜欢旅行  阅读(1010)  评论(0)  编辑  收藏  举报