PTA_ 镖局运镖

镖局的运镖，就是运货（类似现在的物流）。镖局每到一个新地方开展业务，都需要对运镖途中的绿林好汉进行打点。好说话的打点费就比较低，不好说话的打点费就比较高。龙门镖局现在有一趟镖请你来规划路线，已知城市的地图，你需要选择一些道路进行疏通，以便镖局可以到达任意一个城市，要求花费的银子越少越好。

输入格式:

第一行有两个数n和m，n表示有n个城市（编号从1到n），m表示有m条道路。接下来m行，每行形如“a b c”用来表示一条道路，意思是城市a到城市b连通且打点需要花费的银子数是c。

输出格式:

若通过打点能抵达所有城市，则输出最少需要花费的银子总数。若不能抵达所有的城市则输出“impossible”。
输入样例:

3 3
1 2 1
1 3 2
2 3 4

输出样例:

3

**分析：**是否能通过打点抵达所有城市，我理解为判断图是否连通。把花费的银子当作图的权值。我先用深搜判断图是否连通即是否能抵达所有城市（当访问过的顶点数等于城市数则连通，反之则不能）。用prim算法求最小权值和。

#include <stdio.h>
#include <stdlib.h>

#define MAX 10005

typedef struct
{
    int n;
    int e;

    int edges[MAX][MAX];
}graph;

int visit[MAX]={0};
int flag=0;
void DFS(graph *g,int i)
{
    flag++;
    visit[i]=1;
    int j=0;
    for(j=1;j<=g->n;j++)
    {
        if(g->edges[i][j]<MAX&&visit[j]!=1)
        {
            DFS(g,j);
        }
    }
}

void input(graph *g)
{
    scanf("%d%d",&g->n,&g->e);
    int i,j,k,l;
    for(i=1;i<=g->n;i++)
    {
        for(j=1;j<=g->n;j++)
        {
            g->edges[i][j]=MAX;
        }

    }
    for (i=0;i<g->e;i++)
    {
        scanf("%d%d%d",&j,&k,&l);
        g->edges[j][k]=l;
        g->edges[k][j]=l;
    }
    return ;
}

int weight=0;
void prim(graph *g)
{
    int min,i,j,k;
    int adjvex[MAX];
    int lowcost[MAX];
    lowcost[1]=0;
    adjvex[1]=0;
    for(i=1;i<=g->n;i++)
    {
        lowcost[i]=MAX;
        if(g->edges[1][i]!=0)
        {
            lowcost[i]=g->edges[1][i];
            adjvex[i]=1;
        }

    }

    for(i=2;i<=g->n;i++)
    {
        min=MAX;
        j=2;
        k=0;
        while(j<=g->n)
        {
            if(lowcost[j]!=0&&lowcost[j]<min)
            {
                min=lowcost[j];
                k=j;
            }
            j++;
        }
        weight+=g->edges[adjvex[k]][k];
        //printf("(%d,%d)\n",adjvex[k],k);
        lowcost[k]=0;

        for(j=1;j<=g->n;j++)
        {
            if(lowcost[j]!=0&&g->edges[k][j]<lowcost[j])
            {
                lowcost[j]=g->edges[k][j];
                adjvex[j]=k;
            }
        }
    }
    return ;
}

int main()
{
    graph *g=(graph*)malloc(sizeof(graph));
    input(g);
    DFS(g,1);
    if(flag!=g->n)
    {
        printf("impossible");
        return 0;
    }
    prim(g);
    printf("%d",weight);
    return 0;
}