最小生成树(Prim实现)

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

#define INFINITY 0x3f3f3f3f
#define MaxVertexNodeNumSize 1000
#define MaxVertexNodeNameSize 100

struct VertexBodyNode
{
    char VertexName[MaxVertexNodeNameSize];
    int ArcWeight;
    int VertexIndex;
    struct VertexBodyNode *Next;
};

struct VertexHeadNode
{
    char VertexName[MaxVertexNodeNameSize];
    int VertexWeight;
    struct VertexBodyNode *Next;
};

struct _Graph
{
    struct VertexHeadNode VertexHeadNodeList[MaxVertexNodeNumSize];
    int ArcNum,VertexNum;
};

int VertexName2Index(struct _Graph *UnsignedGraph,char *VName)
{
    int i;
    for(i = 0; i < UnsignedGraph -> VertexNum; i ++)
    {
        if(strcmp(UnsignedGraph -> VertexHeadNodeList[i].VertexName,VName)==0)
        {
            return i;
        }
    }
    return -1;
}

void AddOneArc(struct _Graph *UnsignedGraph,int ArcIndex_1,int ArcIndex_2,int AWeight)
{
    struct VertexBodyNode *BNode_1 = malloc(sizeof(struct VertexBodyNode));
    struct VertexBodyNode *BNode_2 = malloc(sizeof(struct VertexBodyNode));

    strcpy(BNode_1 -> VertexName,UnsignedGraph -> VertexHeadNodeList[ArcIndex_1].VertexName);
    strcpy(BNode_2 -> VertexName,UnsignedGraph -> VertexHeadNodeList[ArcIndex_2].VertexName);
    BNode_1 -> ArcWeight = AWeight;
    BNode_2 -> ArcWeight = AWeight;
    BNode_1 -> VertexIndex = ArcIndex_1;
    BNode_2 -> VertexIndex = ArcIndex_2;
    BNode_1 -> Next = BNode_2 -> Next = NULL;

    struct VertexBodyNode *TmpPointer;
    TmpPointer = UnsignedGraph -> VertexHeadNodeList[ArcIndex_1].Next;
    while(TmpPointer != NULL && TmpPointer -> Next != NULL)
    {
        TmpPointer = TmpPointer -> Next;
    }
    if(TmpPointer==NULL)
    {
        UnsignedGraph -> VertexHeadNodeList[ArcIndex_1].Next = BNode_2;
    }
    else
    {
        TmpPointer -> Next = BNode_2;
    }

    TmpPointer = UnsignedGraph -> VertexHeadNodeList[ArcIndex_2].Next;
    while(TmpPointer != NULL && TmpPointer -> Next != NULL)
    {
        TmpPointer = TmpPointer -> Next;
    }
    if(TmpPointer==NULL)
    {
        UnsignedGraph -> VertexHeadNodeList[ArcIndex_2].Next = BNode_1;
    }
    else
    {
        TmpPointer -> Next = BNode_1;
    }
}

void DeleteOneArc(struct _Graph *UnsignedGraph,int ArcIndex_1,int ArcIndex_2)
{
    struct VertexBodyNode *TmpPointer_1,*TmpPointer_2;

    TmpPointer_1 = UnsignedGraph -> VertexHeadNodeList[ArcIndex_1].Next;
    TmpPointer_2 = UnsignedGraph -> VertexHeadNodeList[ArcIndex_1].Next;

    while(TmpPointer_2!=NULL && TmpPointer_2->VertexIndex!=ArcIndex_2)
    {
        TmpPointer_1 = TmpPointer_2;
        TmpPointer_2 = TmpPointer_2 -> Next;
    }

    if(TmpPointer_2==NULL)
    {
        return ;
    }
    else if(TmpPointer_1 == TmpPointer_2)
    {
        UnsignedGraph -> VertexHeadNodeList[ArcIndex_1].Next = NULL;
        free(TmpPointer_2);
    }
    else
    {
        TmpPointer_1 -> Next = TmpPointer_2 -> Next;
        free(TmpPointer_2);
    }

    TmpPointer_1 = UnsignedGraph -> VertexHeadNodeList[ArcIndex_2].Next;
    TmpPointer_2 = UnsignedGraph -> VertexHeadNodeList[ArcIndex_2].Next;

    while(TmpPointer_2!=NULL && TmpPointer_2->VertexIndex!=ArcIndex_1)
    {
        TmpPointer_1 = TmpPointer_2;
        TmpPointer_2 = TmpPointer_2 -> Next;
    }

    if(TmpPointer_2==NULL)
    {
        return ;
    }
    else if(TmpPointer_1 == TmpPointer_2)
    {
        UnsignedGraph -> VertexHeadNodeList[ArcIndex_2].Next = NULL;
        free(TmpPointer_2);
    }
    else
    {
        TmpPointer_1 -> Next = TmpPointer_2 -> Next;
        free(TmpPointer_2);
    }
}

struct _Graph *UGCreat(int ArcSum,int VertexSum)
{
    int i,j;
    struct _Graph *UnsignedGraph = malloc(sizeof(struct _Graph));
    UnsignedGraph -> ArcNum = ArcSum;
    UnsignedGraph -> VertexNum = VertexSum;

    for(i = 0; i < VertexSum; i ++)
    {
        scanf("%s %d",UnsignedGraph -> VertexHeadNodeList[i].VertexName,&UnsignedGraph -> VertexHeadNodeList[i].VertexWeight);
    }

    for(i = 0; i < VertexSum; i ++)
    {
        UnsignedGraph -> VertexHeadNodeList[i].Next = NULL;
    }

    for(i = 0; i < ArcSum; i ++)
    {
        char Arc_1[MaxVertexNodeNameSize];
        char Arc_2[MaxVertexNodeNameSize];
        int ArcIndex_1;
        int ArcIndex_2;
        int ArcWeight;

        scanf("%s %s %d",Arc_1,Arc_2,&ArcWeight);

        ArcIndex_1 = VertexName2Index(UnsignedGraph,Arc_1);
        ArcIndex_2 = VertexName2Index(UnsignedGraph,Arc_2);

        AddOneArc(UnsignedGraph,ArcIndex_1,ArcIndex_2,ArcWeight);
    }
    return UnsignedGraph;
}

void Travel(struct _Graph *UnsignedGraph)
{
    char StartingPoint[MaxVertexNodeNameSize];
    char OverPoint[MaxVertexNodeNameSize];

    printf("Input start and over\n");
    scanf("%s %s",StartingPoint,OverPoint);

    int StartIndex = VertexName2Index(UnsignedGraph,StartingPoint);
    int OverIndex = VertexName2Index(UnsignedGraph,OverPoint);

    struct VertexBodyNode *TmpPointer;
    TmpPointer = UnsignedGraph -> VertexHeadNodeList[StartIndex].Next;
    while(TmpPointer != NULL && TmpPointer -> Next != NULL)
    {
        if(OverIndex==TmpPointer -> VertexIndex)
        {
            printf("Distance:%d GetVertexPointSum:%d",TmpPointer->ArcWeight
                   ,UnsignedGraph -> VertexHeadNodeList[StartIndex].VertexWeight+UnsignedGraph -> VertexHeadNodeList[OverIndex].VertexWeight);
            break;
        }
        else
        {
            TmpPointer = TmpPointer -> Next;
        }
    }
}

void PrimMST(struct _Graph *UnsignedGraph)
{
    int i;
    int ResultTail[UnsignedGraph->VertexNum];
    int ResultWeight[UnsignedGraph->VertexNum];
    int Visit[UnsignedGraph->VertexNum];
    int OtherWeight[UnsignedGraph->VertexNum];
    int ResultTailEnd;
    int ResultWeightEnd;
    memset(OtherWeight,0x3f,sizeof(OtherWeight));
    memset(Visit,0,sizeof(Visit));

    struct VertexBodyNode *TmpPointer;
    TmpPointer = UnsignedGraph -> VertexHeadNodeList[0].Next;
    ResultTail[ResultTailEnd++] = 0;
    while(TmpPointer != NULL)
    {
        OtherWeight[TmpPointer->VertexIndex] = TmpPointer -> ArcWeight;
        TmpPointer = TmpPointer -> Next;
    }

    for(i = 1; i < UnsignedGraph->VertexNum; i ++)
    {
        int TmpIndex = 0;
        int WeightMin;
        int WeightMinIndex = 0;
        WeightMin = INFINITY;

        while(TmpIndex < UnsignedGraph->VertexNum)
        {
            if(Visit[TmpIndex]==0 && OtherWeight[TmpIndex]<WeightMin)
            {
                WeightMin = OtherWeight[TmpIndex];
                WeightMinIndex = TmpIndex;
            }
            TmpIndex ++;
        }

        ResultTail[ResultTailEnd++] = WeightMinIndex;
        ResultWeight[ResultWeightEnd++] = WeightMin;
        Visit[WeightMinIndex] = 1;

        //update
        TmpPointer = UnsignedGraph -> VertexHeadNodeList[WeightMinIndex].Next;
        while(TmpPointer != NULL)
        {
            if(Visit[TmpPointer->VertexIndex]==0 && TmpPointer->ArcWeight<OtherWeight[TmpPointer->VertexIndex])
            {
                OtherWeight[TmpPointer->VertexIndex] = TmpPointer->ArcWeight;
            }
            TmpPointer = TmpPointer -> Next;
        }
    }
    
    for(i = 0;i < ResultTailEnd-1;i ++)
    {
        TmpPointer = UnsignedGraph -> VertexHeadNodeList[ResultTail[i+1]].Next;
        while(TmpPointer != NULL)
        {
            if(TmpPointer->ArcWeight==ResultWeight[i])
            {
                printf("%s ¡ª¡ª%s Distance:%d\n",UnsignedGraph->VertexHeadNodeList[ResultTail[i+1]].VertexName
                                                ,UnsignedGraph->VertexHeadNodeList[TmpPointer->VertexIndex].VertexName
                                                ,ResultWeight[i]);
                break;
            }
            TmpPointer = TmpPointer -> Next;
        }
    }
}

int main()
{
    struct _Graph *G = UGCreat(8,5);
    PrimMST(G);
//    Travel(G);
    return 0;
}

/*
        beijing 18
        zhengzhou 10
        hefei 9
        nanjing 12
        guangzhou 14
        beijing zhengzhou 7
        beijing hefei 9
        beijing nanjing 8
        zhengzhou hefei 5
        hefei nanjing 3
        zhengzhou guangzhou 7
        hefei guangzhou 8
        nanjing guangzhou 6
*/