*浙大PAT甲级 1111

来源:互联网 发布:欧美番号推荐知乎 编辑:程序博客网 时间:2024/06/07 17:56

通过两边的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 999999999using 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]);        }    }}


0 0
原创粉丝点击