HDU-1102-Constructing Roads

题目传送门：http://acm.hdu.edu.cn/showproblem.php?pid=1102

 

题意：

　　有N多个小镇，现在需要建路使所有小镇相通，间接跟直接，给你这些小镇直接的距离，但是有些小镇是已经有路相连的了，所以这两个小镇之间就不用再建路。

 

方法：

　　用了prim跟kruskal都可以做出来，其实就是求最小生成树。只要把已经有路相通的两个小镇的距离改为0即可，不过在开数组时应该是开n*n*2，其他的就一如既往了。

没什么特别要注意的！都是练习Kruskal跟prim算法的习题。

 

#include<iostream>
#include<algorithm>
#include<string>
using namespace std;

const int maxn = 100+10;
int far[maxn];
int n, m;
int k;

struct Edge
{
    bool operator<(const Edge &n)const 
    {
        return value<n.value;
    }
    int strat;
    int end;
    int value;
}edge[maxn*maxn*2];

bool cmp(Edge &a, Edge &b)
{
    return a.value < b.value;
}

void init()
{
    for(int i=0; i<=n; i++)
        far[i] = i;
}

int find(int x)
{
    if(far[x] != x)
        return far[x] = find(far[x]);
    return far[x];
}

void uni(int a, int b)
{
    a = find(a);
    b = find(b);
    if(a != b)
        far[b] = a;
}


int kruskal()
{
    init();
    sort(edge, edge+k, cmp);
    int sum = 0;
    int time=1;
    for(int i=1; i< k && time<n; i++)
    {
        int a = edge[i].strat;
        int b = edge[i].end;
        if(find(a) != find(b))
        {
            uni(a, b);
            time++;
            sum += edge[i].value;
        }
    }
    return sum;
}
int main()
{

    int i, j;
    int a, b, c;
    while(cin>>n)
    {
        k=1;
        for(i=1; i<=n; i++)
        {
            for(j=1; j<=n; j++)
            {
                scanf("%d", &c);
                edge[k].strat = i;
                edge[k].end = j;
                edge[k++].value = c;
            }
        }
        cin>>m;
        for(i=0; i<m; i++)
        {
            scanf("%d%d", &a, &b);
            edge[k].strat = a;
            edge[k].end = b;
            edge[k++].value = 0;
            edge[k].strat = b;
            edge[k].end = a;
            edge[k++].value = 0;
        }
        printf("%d\n", kruskal());
    }
    return 0;
}






/*#include<iostream>
#include<string>
using namespace std;

const int INF = 0x4fffffff;
const int maxn = 100+10;
int map[maxn][maxn];
int dis[maxn];
bool used[maxn];
int n, m;

void init()
{
    int i, j;
    for(i=0; i<=n; i++)
    {
        for(j=0; j<=n; j++)
        {
            map[i][j] = i==j?0:INF;
        }
        dis[i] = INF;
    }
    memset(used, false, sizeof(used));
}

int prim()
{
    dis[1]=0;
    used[1] = true;
    int minNode;
    int min;
    int sum = 0;
    for(int i=1; i<=n; i++)
    {
        min = INF;
        minNode = 1;
        for(int j=1; j<=n; j++)
        {
            if(dis[j]<min && !used[j])
            {
                min = dis[j];
                minNode = j;
            }
        }
        sum += dis[minNode];
        used[minNode] = true;
        for(int k=1; k<=n; k++)
        {
            if(dis[k]>map[minNode][k] && !used[k])
            {
                dis[k] = map[minNode][k];
            }
        }
    }
    return sum;
}


int main()
{
    int i, j;
    int a, b;
    while(cin>>n)
    {
        init();
        for(i=1; i<=n; i++)
            for(j=1; j<=n; j++)
            {
                scanf("%d", &map[i][j]);
            }
        scanf("%d", &m);
        for(i=1; i<=m; i++)
        {
            scanf("%d%d", &a, &b);
            map[a][b] = map[b][a] = 0;
        }
        printf("%d\n", prim());
    }
    return 0;
}*/