最短路 hdu-4522 湫湫系列故事——过年回家

题目链接：

http://acm.hdu.edu.cn/showproblem.php?pid=4522

题目意思：

求出从已知起点到已知终点，两种方式的最小不舒服度。

注意如果某条路为1的话，既可以选硬座又可以选卧铺。

解题思路:

迪杰斯特拉算法。

代码：

 

#include<iostream>

#include<cmath>

#include<cstdio>

#include<cstdlib>

#include<string>

#include<cstring>

#include<algorithm>

#include<vector>

#include<map>

#include<stack>

#include<list>

#include<queue>

#include<ctype.h>

#define eps 1e-6

#define INF (1<<30)

#define PI acos(-1.0)

using namespace std;



#define Maxn 207

bool edge[Maxn][Maxn][2]; //edge[i][j][0]表示i和j之间可以坐硬座，1表示可以做卧铺

int vis[Maxn][2],dis[Maxn][2],A[2];

char save[10000+7];

int n;



void dijs(int start,int kind)

{

   memset(vis,false,sizeof(vis));

   for(int i=1;i<=n;i++)

   {

      if(edge[start][i][kind])

         dis[i][kind]=A[kind];

      else

         dis[i][kind]=INF;

   }

   dis[start][kind]=0;

   vis[start][kind]=true;

   for(int i=1;i<=n-1;i++)

   {

      int Min=INF,flag;



      for(int j=1;j<=n;j++)

      {

         if(vis[j][kind]==false)

         {

            if(dis[j][kind]<Min)

            {

               Min=dis[j][kind];

               flag=j;

            }

         }

      }

      if(Min==INF)

            return ;

      vis[flag][kind]=true;

      for(int k=1;k<=n;k++)

      {

         if(vis[k][kind])

            continue;

         if(edge[flag][k][kind]&&A[kind]+Min<dis[k][kind])

            dis[k][kind]=A[kind]+Min;

      }



   }

}



int main()

{

   int q,t,k,start,end;



   scanf("%d",&q);

   while(q--)

   {

      scanf("%d%d",&n,&t);

      memset(edge,false,sizeof(edge));

      for(int j=1;j<=t;j++)

      {

         scanf("%s%d",save,&k);

         int len=strlen(save);

         save[len++]='\0';

         int last=0,cur=0;



         int i=0;

         for(;save[i]!='+';i++)

            last=last*10+save[i]-'0';

         for(i++;i<len;i++)

         {

            if(!isdigit(save[i]))

            {

               edge[last][cur][k]=true;

               if(k)

                  edge[last][cur][0]=true;

               last=cur;

               //printf("%d\n",cur);

               cur=0;

            }

            else

               cur=cur*10+save[i]-'0';

         }

      }

      scanf("%d%d",&A[0],&A[1]);

      scanf("%d%d",&start,&end);

      dijs(start,0); //硬座的情况

      dijs(start,1);//卧铺的情况

      int ans=min(dis[end][0],dis[end][1]);



      if(ans==INF)

         printf("-1\n");

      else

         printf("%d\n",ans);





      }



   return 0;

}