poj1679----最小生成树

题目大意：

        判断是否存在最小生成树。

        如果存在，判断其是否唯一。

       

        用prim算法求最小生成树。

        将其中的每一条边依次替换，权值是否和最小生成树相同。

        如果相同，则最小生成树不唯一。

        如果都不相同，则唯一。

       

        第一次做最小生成树，不会写，参考的别人的代码：

        这个prim的模板比较好，可以再求最小生成树的同时直接进行判断唯一性。

 

代码如下：

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<iostream> 
using namespace std;
#define MAX 250
#define INF 1000000

int dis[MAX][MAX]; 
int t, n, m;
int result;
struct node
{
    int v1;
    int v2;
    int len;
}e[MAX]; 

void init()
{
    for(int i=0; i<n; i++)     
        for(int j=0; j<n; j++)         
        {    dis[i][j] = INF;
             //printf("dis[%d][%d]=%d\n", i, j, dis[i][j]); 
        }
}
int prim() 
{
    int result=0;
    int i, j, min, minL, vx, vy, leng;
    for(i=0; i<n-1; i++)
    {         
        e[i].v1=0;         
        e[i].v2=i+1;         
        e[i].len=dis[0][i+1];
        //printf("e[%d].len=%d\n", i, dis[0][i+1]);
    }
    for(i=0; i<n-1; i++)
    {  
        minL=INF;  
        min=-1;  
        for(j=i; j<n-1; j++)  
        {    
            if(e[j].len < minL)    
            {     
                minL=e[j].len;     
                min=j;    
            }  
        }  
        if(min==-1)  
            break;  
        swap(e[min],e[i]);  
        result += minL;  
        for(j=0; j<i; j++)//在prim模板的基础上加上这个循环判断mst是否唯一  
        {    
            if(dis[e[j].v2][e[i].v2]==minL && e[j].v2!=e[i].v1)    
                return -1;  
        } 
        vx=e[i].v2;  
        for(j=i+1; j<n-1; j++)  
        {    
            vy=e[j].v2;
            leng=dis[vx][vy];    
            if(leng < e[j].len)    
            {     
                e[j].len=leng;     
                e[j].v1=vx;    
            }      
        }
    } 
    return result;
}
int main()
{
    while(scanf("%d", &t)!=EOF)
    {  
        int x, y, z;  
        for(int i=0; i<t;i++)  
        {    
            scanf("%d%d", &n, &m);    
            init();    
            for(int j=0; j<m; j++)    
            {     
                scanf("%d%d%d", &x, &y, &z);     
                x--;     
                y--;     
                dis[x][y] = dis[y][x] = z;   
            }   
            /*for(x=0; x<n; x++)
            {    for(y=0; y<n; y++)
                    printf("%7d ", dis[x][y]); 
                 printf("\n");        
            }*/
            result=prim();    
            if(result < 0)     
                printf("Not Unique!\n");    
            else printf("%d\n", result);  
        }
    }
    //system("pause");
    return 0;
}