bzoj千题计划226：bzoj2763: [JLOI2011]飞行路线

http://www.lydsy.com/JudgeOnline/problem.php?id=2763

这也算分层图最短路？

dp[i][j]到城市i，还剩k次免费次数的最短路

#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>

using namespace std;

#define N 10001
#define M 50001

int n,k;
int S,T;

int tot;
int front[N],nxt[M<<1],to[M<<1],val[M<<1];

int dp[N][11];
bool vis[N][11];

struct node
{
    int pos,rest;
    int dis;

    node(int Pos=0,int Rest=0,int Dis=0):pos(Pos),rest(Rest),dis(Dis){}
        
    bool operator < (node p) const
    {
        return dis>p.dis;
    }
}now;
priority_queue<node>q;

void read(int &x)
{
    x=0; char c=getchar();
    while(!isdigit(c)) c=getchar();
    while(isdigit(c)) { x=x*10+c-'0'; c=getchar(); }
}

void add(int u,int v,int w)
{
    to[++tot]=v; nxt[tot]=front[u]; front[u]=tot; val[tot]=w;
    to[++tot]=u; nxt[tot]=front[v]; front[v]=tot; val[tot]=w;
}

void dijkstra()
{
    memset(dp,63,sizeof(dp));
    dp[S][k]=0;
    now.pos=S;
    now.rest=k;
    q.push(now);
    int u,v,cnt;
    while(!q.empty())
    {
        now=q.top();
        q.pop();
        u=now.pos;
        cnt=now.rest;
        if(vis[u][cnt]) continue;
        vis[u][cnt]=true;
        for(int i=front[u];i;i=nxt[i])
        {
            v=to[i];
            if(dp[u][cnt]+val[i]<dp[v][cnt])
            {
                dp[v][cnt]=dp[u][cnt]+val[i];
                q.push(node(v,cnt,dp[v][cnt]));
            }
            if(cnt && dp[u][cnt]<dp[v][cnt-1])
            {
                dp[v][cnt-1]=dp[u][cnt];
                q.push(node(v,cnt-1,dp[v][cnt-1]));
            }
        }
    }
}

int main()
{
    int m;
    read(n); read(m); read(k);
    read(S); read(T);
    int u,v,w;
    while(m--)
    {
        read(u); read(v); read(w);
        add(u,v,w);
    }
    dijkstra();
    int ans=1e9;
    for(int i=0;i<=k;++i) ans=min(ans,dp[T][i]);
    printf("%d",ans);
}

Description

Alice和Bob现在要乘飞机旅行，他们选择了一家相对便宜的航空公司。该航空公司一共在n个城市设有业务，设这些城市分别标记为0到n-1，一共有m种航线，每种航线连接两个城市，并且航线有一定的价格。Alice和Bob现在要从一个城市沿着航线到达另一个城市，途中可以进行转机。航空公司对他们这次旅行也推出优惠，他们可以免费在最多k种航线上搭乘飞机。那么Alice和Bob这次出行最少花费多少？

Input

数据的第一行有三个整数，n,m,k，分别表示城市数，航线数和免费乘坐次数。

第二行有两个整数，s,t，分别表示他们出行的起点城市编号和终点城市编号。(0<=s,t<n)

接下来有m行，每行三个整数，a,b,c，表示存在一种航线，能从城市a到达城市b，或从城市b到达城市a，价格为c。(0<=a,b<n,a与b不相等，0<=c<=1000)

Output

只有一行，包含一个整数，为最少花费。

Sample Input

5 6 1
0 4
0 1 5
1 2 5
2 3 5
3 4 5
2 3 3
0 2 100

Sample Output

8

HINT

对于30%的数据,2<=n<=50,1<=m<=300,k=0;

对于50%的数据,2<=n<=600,1<=m<=6000,0<=k<=1;

对于100%的数据,2<=n<=10000,1<=m<=50000,0<=k<=10.