HDU 4396

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

题意：在至少走k条边的前提下求最短路

思路：在原有最短路模板的基础上多加一维，dis[i][j]表示走到i点经过j条边的最短路，没有别的变化

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
using namespace std ;

const int INF=0xfffffff ;
struct node{
    int s,t,v,nxt ;
}e[] ;

int n,m,k,cnt,head[],vis[][],dis[][] ;

void add(int s,int t,int v)
{
    e[cnt].s=s ;
    e[cnt].t=t ;
    e[cnt].v=v ;
    e[cnt].nxt=head[s] ;
    head[s]=cnt++ ;
}

void spfa(int s)
{
    for(int i= ;i<=n ;i++)
        for(int j= ;j< ;j++)
            dis[i][j]=INF ;
    dis[s][]= ;
    memset(vis,,sizeof(vis)) ;
    vis[s][]= ;
    queue <pair<int,int> > q ;
    q.push(make_pair(s,)) ;
    while(!q.empty())
    {
        pair<int,int> u=q.front() ;
        q.pop() ;
        vis[u.first][u.second]= ;
        int step=u.second+ ;
        if(step>k)step=k ;
        for(int i=head[u.first] ;i!=- ;i=e[i].nxt)
        {
            int tt=e[i].t ;
            if(dis[tt][step]>dis[u.first][u.second]+e[i].v)
            {
                dis[tt][step]=dis[u.first][u.second]+e[i].v ;
                if(!vis[tt][step])
                {
                    vis[tt][step]= ;
                    q.push(make_pair(tt,step)) ;
                }
            }
        }
    }
}
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        cnt= ;
        memset(head,-,sizeof(head)) ;
        while(m--)
        {
            int a,b,c ;
            scanf("%d%d%d",&a,&b,&c) ;
            add(a,b,c) ;
            add(b,a,c) ;
        }
        int s,t ;
        scanf("%d%d%d",&s,&t,&k) ;
        k=k/+(k%!=) ;
        spfa(s) ;
        if(dis[t][k]==INF)puts("-1") ;
        else printf("%d\n",dis[t][k]) ;
    }
    return  ;
}