[JLOI2011]飞行路线

提交：bzoj2763
题目↑↑

这道题是由tyb师兄讲的一道spfa

只要记录f[i][j]表示到第i个点用j次机会的最短路径即可

if(f[x][c]+a[k].d<f[y][c])//这种情况直接进行最短路
{
    f[y][c]=f[x][c]+a[k].d;
    if(v[y][c]==false)
    {
        v[y][c]=true;
        list[tail][0]=y;
        list[tail][1]=c;
        tail++;if(tail==100000+1) tail=1;
    }
}
if(f[x][c]<f[y][c+1]&&c<kk)//如果可以使用免费机会就改变
{
    f[y][c+1]=f[x][c];
    if(v[y][c+1]==false)
    {
        v[y][c+1]=true;
        list[tail][0]=y;
        list[tail][1]=c+1;
        tail++;if(tail==100000+1) tail=1;
    }
}

代码↓↓

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
struct node
{
    int x,y,d,next;
}a[110000];int len,last[51000];
void ins(int x,int y,int d)
{
    len++;
    a[len].x=x;a[len].y=y;a[len].d=d;
    a[len].next=last[x];last[x]=len;
}
int f[110000][11];
bool v[110000][11];
int n,m,kk;int st,ed;
int list[110000][2];
void spfa()
{
    int head=1,tail=2;
    memset(list,0,sizeof(list));
    list[1][0]=st;list[1][1]=0;
    while(head!=tail)
    {
        int x=list[head][0],c=list[head][1];
        for(int k=last[x];k;k=a[k].next)
        {
            int y=a[k].y;
            if(f[x][c]+a[k].d<f[y][c])
            {
                f[y][c]=f[x][c]+a[k].d;
                if(v[y][c]==false)
                {
                    v[y][c]=true;
                    list[tail][0]=y;
                    list[tail][1]=c;
                    tail++;if(tail==100000+1) tail=1;
                }
            }
            if(f[x][c]<f[y][c+1]&&c<kk)
            {
                f[y][c+1]=f[x][c];
                if(v[y][c+1]==false)
                {
                    v[y][c+1]=true;
                    list[tail][0]=y;
                    list[tail][1]=c+1;
                    tail++;if(tail==100000+1) tail=1;
                }
            }
        }
        head++;if(head==100000+1) head=1;
        v[x][c]=false;
    }
}
int main()
{
    scanf("%d%d%d",&n,&m,&kk);
    scanf("%d%d",&st,&ed);
    for(int i=1;i<=m;i++)
    {
        int x,y,d;
        scanf("%d%d%d",&x,&y,&d);
        ins(x,y,d);ins(y,x,d);
    }
    for(int i=0;i<=n;i++) for(int j=0;j<=kk;j++) f[i][j]=999999999;
    memset(v,false,sizeof(v));v[st][0]=true;
    f[st][0]=0;
    spfa();
    int mmin=999999999;
    for(int i=0;i<=kk;i++) mmin=min(f[ed][i],mmin);
    printf("%d\n",mmin);
    return 0;
}