USACO 2009 Feb Gold 3.Revamping Trails

Description

Farmer John dutifully checks on the cows every day. He traverses so me of the M (1 <= M <= 50,000) trails conveniently numbered 1..M from pasture 1 all the way out to pasture N (a journey which is always possible for trail maps given in the test data). The N (1 <= N <= 10,000) pastures conveniently numbered 1..N on Farmer John’s farm are currently connected by bidirectional dirt trails. Each trail i connects pastures P1_i and P2_i (1 <= P1_i <= N; 1 <= P2_i <= N) and requires T_i (1 <= T_i <= 1,000,000) units of time to traverse.
He wants to revamp some of the trails on his farm to save time on his long journey. Specifically, he will choose K (1 <= K <= 20) trails to turn into highways, which will effectively reduce the trail’s traversal time to 0. Help FJ decide which trails to revamp to minimize the resulting time of getting from pasture 1 to N.

Input

• Line 1: Three space-separated integers: N, M, and K

• Lines 2..M+1: Line i+1 describes trail i with three space-separated
  integers: P1_i, P2_i, and T_i

Output

• Line 1: The length of the shortest path after revamping no more than
  K edges

Sample Input

4 4 1
1 2 10
2 4 10
1 3 1
3 4 100

Sample Output

1

Key To Problem

小小的图论+二维dp
dis[u][num]表示到点u，翻新num条道路所需要走的最短的时间
dis[to][num]=dis[u][num]+s;
dis[to][num+1]=dis[u][num];
需要用dijkstra+堆优化，据说SPFA超时，并没有尝试。

CODE

#include
#include
#include
#include
#include
#define N 10010
using namespace std;
vector<int>V[N];
vector<int>pre[N];
struct node
{
    int u,num,s;
    bool friend operator <(node x,node y)
    {
        return x.s>y.s;
    }
};
int n,m,l;
int dis[N][30];
bool used[N][30];

void dijkstra(int u)
{
    memset(dis,0x3f,sizeof(dis));
    priority_queueQ;
    node k;
    k.u=u,k.s=0,k.num=0;
    Q.push(k);
    dis[u][0]=0;
    while(Q.size())
    {
        k=Q.top();
        Q.pop();
        if(used[k.u][k.num])
            continue;
        used[k.u][k.num]=1;
        int num=k.num;
        for(int i=0;iint to=V[k.u][i];
            int s=pre[k.u][i];
            if(dis[to][num]>dis[k.u][num]+s)
            {
                dis[to][num]=dis[k.u][num]+s;
                node p;
                p.u=to,p.num=num,p.s=dis[to][num];
                Q.push(p);
            }
            if(num+1<=l&&dis[to][num+1]>dis[k.u][num])
            {
                dis[to][num+1]=dis[k.u][num];
                node p;
                p.u=to,p.num=num+1,p.s=dis[to][num+1];
                Q.push(p);
            }
        }
    }
}

int main()
{
//  freopen("revamp.in","r",stdin);
//  freopen("revamp.out","w",stdout);
    scanf("%d%d%d",&n,&m,&l);
    for(int i=1;i<=m;i++)
    {
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        V[x].push_back(y);
        pre[x].push_back(z);
        V[y].push_back(x);
        pre[y].push_back(z);
    }
    dijkstra(1);
    printf("%d\n",dis[n][l]);
    return 0;
}