poj 3662 二分答案或二维dij

二维dij dist[v][k] 表示到达v，使用k条免费边后所经过路径最大值

dist[v][k]=min(max(u->v,dist[u][k]),dist[v][k])  

dist[v][k]=dist[u][k-1]

 

我用二分做的

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
using namespace std;
int vis[1005][1005],N,P,K,flag;
int first[1005],size;
struct Edge
{
    int v,l,next;
}edge[20005];
void insert(int u,int v,int l)
{
    edge[size].v=v;
    edge[size].l=l;
    edge[size].next=first[u];
    first[u]=size++;
}
void dfs(int u,int L,int r)
{
    vis[u][r]=1;
    if(u==N) flag=1;
    if(flag) return ;
    for(int e=first[u];e!=-1;e=edge[e].next)
    {
        int v=edge[e].v,l=edge[e].l;
        if(l<=L)
        {
            if(!vis[v][r])
            {
                 vis[v][r]=1;dfs(v,L,r);
            }
        }
        else if(r>0 && !vis[v][r-1])
        {
            vis[v][r-1]=1;dfs(v,L,r-1);
        }
    }
}
int main()
{
 //   freopen("test.txt","r",stdin);
    scanf("%d%d%d",&N,&P,&K);
    memset(first,-1,sizeof(first));size=0;
    for(int i=0;i<P;i++)
    {
        int a,b,l;
        scanf("%d%d%d",&a,&b,&l);
        insert(a,b,l);insert(b,a,l);
    }
    int l=0,r=1000000,ret=-1;
    while(l<=r)
    {
        int mid=(l+r)>>1;
        memset(vis,0,sizeof(vis));
        flag=0; dfs(1,mid,K);
        if(flag)
        {
            ret=mid;
            r=mid-1;
        }
        else l=mid+1;
    }
    printf("%d\n",ret);
    return 0;
}