hdu (4396：多校第十场）：又见spfa求各种状态的最短路

   本题题意：为找到至少走K条边的最短路。

#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
#include <queue>
#include <map>
#include <set>
#include <stack>
#include <cmath>
#include <string>
#define inf 1<<30
using namespace std;
const int N=5050;
struct Node
{
   int v;
   int len;
};
struct Que
{
   int num;
   int tim;
};
vector<Node>edge[N];
queue<Que>myqueue;

int n,m,s,t,k;
int dis[N][60];//dis[i][j]表示走了到达第i个节点走了j条路的最短路
int vis[N][60];//是否在队列中

int spfa()
{
    for(int i=1;i<=n;i++)
      for(int j=0;j<=k+1;j++)
        dis[i][j]=inf;
    while(!myqueue.empty())
       myqueue.pop();
    memset(vis,0,sizeof(vis));
    Que top;
    top.num=s;
    top.tim=0;
    dis[s][0]=0;
    vis[s][0]=1;
    myqueue.push(top);
    Que net;
    while(!myqueue.empty())
    {
      top=myqueue.front();
      myqueue.pop();
      int tim=top.tim;
      int u=top.num;
      vis[u][tim]=0;
      for(int i=0;i<edge[u].size();i++)
      {
          int v=edge[u][i].v;
          int le=edge[u][i].len;
          int nx;
          if(tim>k)
            nx=tim;
          else
            nx=tim+1;
          if(dis[v][nx]>dis[u][tim]+le)
          {
              dis[v][nx]=dis[u][tim]+le;
              if(!vis[v][nx])
              {
                  vis[v][nx]=1;
                  net.num=v;
                  net.tim=nx;
                  myqueue.push(net);
              }
          }
      }
    }
    if(dis[t][k]>=inf&&dis[t][k+1]>=inf)
    return -1;
    else
    return min(dis[t][k],dis[t][k+1]);
}

int main()
{
  int u,v,len;
  Node ne;
  while(scanf("%d%d",&n,&m)!=EOF)
  {
      for(int i=1;i<=n+2;i++)
        edge[i].clear();
       for(int i=1;i<=m;i++)
       {
          scanf("%d%d%d",&u,&v,&len);
           ne.len=len;
           ne.v=v;
           edge[u].push_back(ne);
           ne.v=u;
           edge[v].push_back(ne);
       }
       scanf("%d%d%d",&s,&t,&k);
       k=k%10==0?k/10:k/10+1;
       int ans=spfa();
       printf("%d\n",ans);
  }
  return 0;
}