【bzoj1579】[Usaco2009 Feb]Revamping Trails 道路升级

Description

每天,农夫John需要经过一些道路去检查牛棚N里面的牛. 农场上有M(1<=M<=50,000)条双向泥土道路,编号为1..M. 道路i连接牛棚P1_i和P2_i (1 <= P1_i <= N; 1 <= P2_i<= N). John需要T_i (1 <= T_i <= 1,000,000)时间单位用道路i从P1_i走到P2_i或者从P2_i 走到P１_i 他想更新一些路经来减少每天花在路上的时间.具体地说,他想更新K (1 <= K <= 20)条路经，将它们所须时间减为０．帮助FJ选择哪些路经需要更新使得从１到N的时间尽量少.

Input

• 第一行: 三个空格分开的数: N, M, 和 K * 第2..M+1行: 第i+1行有三个空格分开的数：P1_i, P2_i, 和 T_i

Output

• 第一行: 更新最多Ｋ条路经后的最短路经长度．

Sample Input

4 4 1

1 2 10

2 4 10

1 3 1

3 4 100

Sample Output

1

题解
Dijkstra，不同的是对于dis[i]，增加一维记录走了几次特殊道路，转移即可。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<set>
#include<ctime>
#include<vector>
#include<cmath>
#include<algorithm>
#include<map>
#include<queue>
#define mod 1000000007 
#define ll long long  
#define inf 1e9
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
struct node{int d,now,k;};
struct cmp{
    bool operator()(node a,node b)
    {
        return a.d>b.d;
    }
};
priority_queue<node,vector<node>,cmp>q;
int tot,n,m,ans=100000000,k;
int ret[100005],len[100005],Next[100005],Head[10005];
int dis[10005][21];
bool vis[10005][21];
inline void ins(int u,int v,int l)
{
    ret[++tot]=v;len[tot]=l;
    Next[tot]=Head[u];Head[u]=tot;
}
void Dijkstra()
{
    memset(dis,127,sizeof(dis));
    dis[1][0]=0;q.push((node){0,1,0});
    while (!q.empty())
    {
        node t=q.top();q.pop();
        if (vis[t.now][t.k]) continue;vis[t.now][t.k]=1;
        for (int i=Head[t.now];i;i=Next[i])
        {
            int v=ret[i];
            if (dis[v][t.k]>dis[t.now][t.k]+len[i]) 
            {
                dis[v][t.k]=dis[t.now][t.k]+len[i];
                q.push((node){dis[t.now][t.k]+len[i],v,t.k});
            }
            if (dis[v][t.k+1]>dis[t.now][t.k]&&t.k<k)
            {
                dis[v][t.k+1]=dis[t.now][t.k];
                q.push((node){dis[t.now][t.k],v,t.k+1});
            }
        }
    }
}
int main()
{
    n=read();m=read();k=read();
    for (int i=1;i<=m;i++)
    {
        int u=read(),v=read(),l=read();
        ins(u,v,l);ins(v,u,l);
    }
    Dijkstra();
    for (int i=0;i<=k;i++) ans=min(dis[n][i],ans);
    printf("%d",ans);
    return 0;
}