poj 3662 二分+最短路，求第k大最小权值

//poj 3662找一条从1到n的路线让这条路线中，第k大线最小
//先对所有的线段排序，二分枚举+最短路
#include<cstdio>
#include<cstring>
#include<vector>
#include<iostream>
#include<algorithm>
#include<queue>
#define N 1000+10
#define clr(a,b) memset(a,b,sizeof(a))
#define inf 0xfffffff
using namespace std;
struct node{
   int u,v,w;
}edge[N*N];
int dis[N];
bool used[N];
int g[N][N];
int n,p,k;
int cmp(node a,node b)
{
        return a.w<b.w;
}
void init(int l)
{
     for(int i=1;i<=p;i++)
     {
             int a=edge[i].u;
             int b=edge[i].v;
             if(edge[i].w>l) g[a][b]=g[b][a]=1;
             else g[a][b]=g[b][a]=0;
     }
}
void dij(int u)
{
        for(int i=1;i<=n;i++)
                dis[i]=g[u][i];
        memset(used,false,sizeof(used));
        used[u]=true;
        for(int i=1;i<n;i++)
        {
                int minn=inf,tep=-1;
                for(int j=1;j<=n;j++)
                {
                        if(minn>dis[j]&&!used[j])
                                minn=dis[tep=j];

                }
                if(tep==-1) break;
                used[tep]=true;
                for(int j=1;j<=n;j++)
                {

                      if(dis[tep]+g[tep][j]<dis[j]&&!used[j])
                                 dis[j]=dis[tep]+g[tep][j];

                }
        }
}
int er_find()
{
        int low=1;
        int high=p;
        while(low<=high)
        {
                int mid=(low+high)/2;
                init(edge[mid].w);
                dij(1);
                if(dis[n]<=k)
                        high=mid-1;
                else
                        low=mid+1;

        }
        return low;
}
int main()
{

        while(scanf("%d%d%d",&n,&p,&k)!=EOF)
        {
                for(int i=1;i<=n;i++)
                {
                        for(int j=1;j<=n;j++)
                        {
                                if(i==j) g[i][j]=0;
                                else g[i][j]=inf;
                        }
                }
                for(int i=1;i<=p;i++)
                        scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
                sort(edge+1,edge+p+1,cmp);
                init(0);
                dij(1);
                if(dis[n]>=inf)
                        printf("-1\n");
                else
                {
                        if(dis[n]<=k) printf("0\n");
                        else
                                        printf("%d\n",edge[er_find()].w);

                }
        }
        return 0;
}