*浙大PAT甲级 1111

通过两边的dij最短路算法。path[]数组记录最短路中每个结点的前驱结点,第一遍dij以路径的距离为主,如果相等则判断时间最短。第二遍dij以路径所花时间为主,如果相等则判断路径的结点数最小。如果两边dij算法的路径相同则合并在一起输出,否则分开输出。

AC代码:

#include<iostream>
#include<map>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>
#include<set>
#include<stack>
#include<cmath>
#include<vector>
#include<hash_map>
#define ll long long
#define inf 999999999
using namespace std;
int a[500][500];
int b[500][500];
int dist[500];
int sj[500];
int path[500];
int mark[500];
int num[500];
int main()
{
    int n,m;
    scanf("%d %d",&n,&m);
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {

         a[i][j]=b[i][j]=inf;
        }
    }
    for(int i=0;i<m;i++)
    {
        int l,r;
        int ok;
        int d,t;
        scanf("%d %d %d %d %d",&l,&r,&ok,&d,&t);
        a[l][r]=d;
        b[l][r]=t;
        if(ok==0)
        {
            a[r][l]=d;
            b[r][l]=t;
        }
    }
    int sta,en;
    scanf("%d %d",&sta,&en);
    for(int i=0;i<n;i++)
    {
        dist[i]=a[sta][i];
        sj[i]=b[sta][i];
        if(i!=sta&&dist[i]!=inf) path[i]=sta;
        else
            path[i]=-1;
    }
    mark[sta]=1;
    int biao=sta;
    for(int i=0;i<n-1;i++)
    {
        int minn=inf;
        for(int j=0;j<n;j++)
        {
            if(mark[j]==0&&dist[j]<minn)
            {
                minn=dist[j];
                biao=j;
            }
        }
        mark[biao]=1;
        for(int j=0;j<n;j++)
        {
            if(mark[j]==0)
            {
                if(dist[biao]+a[biao][j]<dist[j])
                {
                    dist[j]=dist[biao]+a[biao][j];
                    sj[j]=sj[biao]+b[biao][j];
                    path[j]=biao;
                }
                else if(dist[biao]+a[biao][j]==dist[j])
                {
                    if(sj[biao]+b[biao][j]<sj[j])
                    {
                        sj[j]=sj[biao]+b[biao][j];
                        path[j]=biao;
                    }
                }
            }
        }
    }
    vector<int> v;
    v.push_back(en);
    int tmp=en;
    while(path[tmp]!=-1)
    {
        v.insert(v.begin(),path[tmp]);
        tmp=path[tmp];
    }
    /*for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            c[i][j]=1;
        }
    }*/
    for(int i=0;i<n;i++)
    {
        sj[i]=b[sta][i];
        if(i!=sta&&sj[i]!=inf)
        {
            path[i]=sta;
            num[i]=2;
        }
        else
           {
               path[i]=-1;
               num[i]=inf;
           }
    }
    memset(mark,0,sizeof(mark));
    mark[sta]=1;
    biao=sta;
    for(int i=0;i<n-1;i++)
    {
        int minn=inf;
        for(int j=0;j<n;j++)
        {
            if(mark[j]==0&&sj[j]<minn)
            {
                minn=sj[j];
                biao=j;
            }
        }
        mark[biao]=1;
        for(int j=0;j<n;j++)
        {
            if(mark[j]==0)
            {
                if(sj[biao]+b[biao][j]<sj[j])
                {
                    sj[j]=sj[biao]+b[biao][j];
                    num[j]=num[biao]+1;
                    path[j]=biao;
                }
                else if(sj[biao]+b[biao][j]==sj[j])
                {
                    if(num[biao]+1<num[j])
                    {
                        num[j]=num[biao]+1;
                        path[j]=biao;
                    }
                }
            }
        }
    }
    vector<int> v1;
    v1.push_back(en);
    tmp=en;
     while(path[tmp]!=-1)
    {
        v1.insert(v1.begin(),path[tmp]);
        tmp=path[tmp];
    }
    if(v==v1)
    {
        printf("Distance = %d; Time = %d: ",dist[en],sj[en]);
        printf("%d",v[0]);
        for(int i=1;i<v.size();i++)
        {
            printf(" -> %d",v[i]);
        }
    }
    else
    {
        printf("Distance = %d: ",dist[en]);
        printf("%d",v[0]);
        for(int i=1;i<v.size();i++)
        {
            printf(" -> %d",v[i]);
        }
        printf("\n");
        printf("Time = %d: ",sj[en]);
        printf("%d",v1[0]);
        for(int i=1;i<v1.size();i++)
        {
            printf(" -> %d",v1[i]);
        }
    }


}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值