邻接表解决多段图问题

#include <iostream>
#include <algorithm>
#include <math.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
using namespace std;
const int MAXN=100;
struct Lis
{
    int id;
    int dis;
    Lis * next;
};
struct Num
{
    int src_dis;
    Lis * head;
}num[MAXN];

int stage[10][100];
void init(int n,int k)//初始化数据；
{
    int m;
    memset(num,0,sizeof(num));
    for (int i = 1; i <= n; ++i)
    {
        num[i].src_dis=10000;
        num[i].head=(Lis *)malloc(sizeof(Lis));
        num[i].head->next=NULL;
    }
    int d1,d2,distance;
    Lis *p;
    cout<<"输入边的长度：输入1 2 4 表示点1 与 2的边的长度为 4：首数字为0表示结束输入."<<endl;
    while(1)
    {
        cin>>d1;
        if(d1==0)break;
        cin>>d2>>distance;

        p=(Lis *)malloc(sizeof(Lis));
        p->id=d2;p->dis=distance;
        if(num[d1].head->next==NULL)
        {
            num[d1].head->next=p;
            p->next=NULL;
        }
        else{
            p->next=num[d1].head->next;
            num[d1].head->next=p;
        }
        
    }
    cout<<"输入阶段"<<endl;
 

    memset(stage,-1,sizeof stage);//若为－1  则无数据了
    num[1].src_dis=0;
    for(int i=1;i<=k;i++)
    {
        cin>>stage[i][0];    //第i个阶段的［0］是存当前阶段的个数

        for(int j=1;j<=stage[i][0];j++)
        {
            cin>>stage[i][j];
        }
    }




}

void deal(int n,int k)
{
    int i,j;
    int a,b,c;
    num[stage[1][1]].src_dis=0;//第一阶段的数到自己的距离




    for (i = 2; i <= k; ++i)
    {
        if(i==2)
        {
            a=stage[1][1];
            Lis *p=num[a].head->next;
            for (j = 1; j <= stage[i][0]; ++j)
            {
                //while(p.id==a)
                b=p->id;c=p->dis;
                num[b].src_dis=c;
                p=p->next;

            }
        }
        else
        {
            for(int k = 1;k<=stage[i-1][0];k++)
            {
                int tmp=stage[i-1][k];
                cout<<"tmp="<<tmp<<endl;
                a=tmp;
                    Lis *p=num[a].head->next;
                for (j = 1; j <= stage[i][0]; ++j)
                {
                    
                    b=p->id;c=p->dis+num[a].src_dis;//b为当前的点
                    //判断下第b个点的最短到达的数

                    num[b].src_dis=min(c,num[b].src_dis);
                    p=p->next;
                    if(!p)break;
                }

            }


    // for(int x=1;x<=n;x++)
    // {
    //     cout<<"stage "<<x<<" "<<num[x].src_dis<<"  ";
    // }
    // cout<<endl<<endl;


            
        }

        


    }
    int las = stage[k][1];
    //cout<<"world"<<endl;
    cout<<"dis = "<<num[las].src_dis<<endl;
}
int main()
{
    int n,k;
    cout<<"输入数据个数，和阶段数：";
    cin>>n>>k;
    init(n,k);
    
    deal(n,k);





    return 0;
}


/*

12
5
1 2 9
1 3 7
1 4 3
1 5 2
2 6 4
2 7 2
2 8 1
3 6 2
3 7 7
4 8 11
5 7 11
5 8 8
6 9 6
6 10 5
7 9 4
7 10 3
8 10 5
8 11 6
9 12 4
10 12 2
11 12 5
0
1
1
4
2 3 4 5
3
6 7 8
3
9 10 11
1
12


4 3
1 2 3
1 3 4
2 4 5
3 4 6
0


*/