HDOJ-1102 Constructing Roads(最小生成树)两种做法

题目

最小生成树

Kruskal算法

/*
HDU 1102 Constructing Roads
题意：有n个乡村，修路，然后保证每两条路之间连接。
如果A和B连接，那么A和B直接连接或者中间还有一个乡村，间接连接。
已经有一些路了，你的任务是找出来一路径连接所有乡村。
输入：i到j的长度，注意去重边
最小生成树，，，
*/
#include
#include
#include
using namespace std;

const int N = 10005;
struct edge
{
    int u,v,w;
}e[N];
int n,f[N];
//将边降序排序
void quicksort(int left,int right)
{
    int i,j;
    struct edge t;
    if(left > right)
        return;

    i = left;
    j = right;
    while(i!=j)
    {
        //注意顺序
        //先从右边找
        while(e[j].w >= e[left].w && i < j)
            j--;
        //从左边找
        while(e[i].w <= e[left].w && i < j)
            i++;
        if(i= j)
                    continue;
                e[num].u = i;
                e[num].v = j;
                e[num].w = x;
                num++;
            }

        quicksort(1,num-1);
        Init();//并查集优化

        int q,t1,t2;
        scanf("%d",&q);
        for(int i = 0;i < q;i++)
        {
            scanf("%d%d",&t1,&t2);
            merge(t1,t2);//合并已有的道路
        }

        int counter = 0,sum = 0;
        //Kruskal算法核心
        for(int i=1;i

Prim算法

#include
#include
#include
using namespace std;

const int INF = 99999999;
const int N = 1005;

int e[N][N],dis[N],book[N]={0};
int main()
{
    int n;

    while(~scanf("%d",&n))
    {
        for(int i = 1;i <= n;i++)
            for(int j = 1;j <= n;j++)
               scanf("%d",&e[i][j]);

        int q;
        scanf("%d",&q);
        int t1,t2;
        for(int i = 1;i <= q;i++)
        {
            scanf("%d%d",&t1,&t2);
            e[t1][t2] = e[t2][t1] = 0;
        }

        //初始化dis数组
        for(int i = 1;i <= n;i++)
            dis[i] = e[1][i];
        memset(book,0,sizeof(book));

        int counter = 0,sum = 0;
        //Prim算法核心
        //将1号顶点加入生成树
        book[1] = 1;
        counter++;
        int u,v,minn;
        while(counter < n)
        {
            minn = INF;
            for(int i = 1;i <= n;i++)
            {
                if(!book[i] && dis[i] < minn)
                {
                    minn = dis[i];
                    u = i;
                }
            }

            book[u] = 1;
            counter++;
            sum += dis[u];
            for(int v = 1;v <= n;v++)
            {
                if(!book[v] && dis[v] > e[u][v])
                   dis[v] = e[u][v];
            }
        }
        printf("%d\n",sum);
    }
    return 0;
}