BZOJ1579: [Usaco2009 Feb]Revamping Trails 道路升级

1579: [Usaco2009 Feb]Revamping Trails 道路升级

Time Limit: 10 Sec  Memory Limit: 64 MB
Submit: 1251  Solved: 337
[Submit][Status]

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

HINT

K是1; 更新道路3->4使得从３到４的时间由１００减少到０. 最新最短路经是1->3->4,总用时为１单位. N<=10000

Source

Gold

题解：

没想到竟是如此暴力。。。直接把k加进状态里即可。。。我的思维还是不够暴力。。。

代码：

#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<vector>
#include<map>
#include<set>
#include<queue>
#define inf 1000000000
#define maxn 10000+100
#define maxm 50000+100
#define eps 1e-10
#define ll long long
#define pa pair<int,int>
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=10*x+ch-'0';ch=getchar();}
    return x*f;
}
int n,m,k,tot;
int d[maxn][25],head[maxn];
bool v[maxn][25];
struct edge{int go,next,w;}e[2*maxm];
void ins(int x,int y,int z)
{
    e[++tot].go=y;e[tot].w=z;e[tot].next=head[x];head[x]=tot;
}
void insert(int x,int y,int z)
{
    ins(x,y,z);ins(y,x,z);
}
void dijkstra()
{
    priority_queue<pa,vector<pa>,greater<pa> >q;
    for(int i=1;i<=n;i++)
     for(int j=0;j<=k;j++)
      d[i][j]=inf;
    memset(v,0,sizeof(v));
    d[1][0]=0;q.push(make_pair(0,1));
    while(!q.empty())
    {
        int x=q.top().second;q.pop();
        int z=x/(n+1);x=x%(n+1);
        if(v[x][z])continue;v[x][z]=1;
        for(int i=head[x],y;i;i=e[i].next)
          {
            if(d[x][z]+e[i].w<d[y=e[i].go][z])
            {
                d[y][z]=d[x][z]+e[i].w;
                q.push(make_pair(d[y][z],z*(n+1)+y));
            }
            if(z==k)continue;
            if(d[x][z]<d[y][z+1])
            {
                d[y][z+1]=d[x][z];
                q.push(make_pair(d[y][z+1],(z+1)*(n+1)+y));
            }
          } 
    }
}
int main()
{
    freopen("input.txt","r",stdin);
    freopen("output.txt","w",stdout);
    n=read();m=read();k=read();
    for(int i=1;i<=m;i++)
    {
        int x=read(),y=read(),z=read();
        insert(x,y,z);
    }
    dijkstra();
    printf("%d
",d[n][k]);
    return 0;
}