poj 3662 Telephone Lines

/*
* @Author: Lyucheng
* @Date:   2017-07-24 11:08:55
* @Last Modified by:   Lyucheng
* @Last Modified time: 2017-07-24 17:36:49
*/
/*
 题意：题意很迷，有n个点，p条边，现在要修一条从1到n的线路，你可以将其中的k条线权值变为0，并且你修的线路中
    最长的边，一定要是最小的

 思路：二分答案，将大于mid的边权值为1，小于等于的为0，找出最少耗费的1，如果小于k那么就可以通过
*/
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>

#define MAXN 1005
#define INF 0x3f3f3f3f
#define MAXR 1000000

using namespace std;

int n,p,k;
int u,v,w;
struct HeapNode //Dijkstra算法用到的优先队列的节点
{
    int d,u;
    HeapNode(int d,int u):d(d),u(u){}
    bool operator < (const HeapNode &rhs)const
    {
        return d > rhs.d;
    }
};

struct Edge     //边
{
    int from,to,dist;
    Edge(int f,int t,int d):from(f),to(t),dist(d){}
};

struct Dijkstra
{
    int n,m;            //点数和边数,编号都从0开始
    vector<Edge> edges; //边列表
    vector<int> G[MAXN];//每个节点出发的边编号(从0开始编号)
    bool done[MAXN];    //是否已永久标号
    int d[MAXN];        //s到各个点的距离
    int p[MAXN];        //p[i]为从起点s到i的最短路中的最后一条边的编号

    void init(int n)
    {
        this->n=n;
        for(int i=0;i<n;i++) G[i].clear();//清空邻接表
        edges.clear();  //清空边列表
    }

    void AddEdge(int from,int to,int dist)
    {//如果是无向图，每条无向边调用两次AddEdge
        edges.push_back(Edge(from,to,dist) );
        m = edges.size();
        G[from].push_back(m-1);
    }

    bool dijkstra(int s,int pos)//求s到所有点的距离
    {
        priority_queue<HeapNode> Q;
        for(int i=0;i<n;i++) d[i]=INF;
        d[s]=0;
        memset(done,0,sizeof(done));
        Q.push(HeapNode(0,s) );

        while(!Q.empty())
        {
            HeapNode x=Q.top(); Q.pop();
            int u=x.u;
            if(done[u]) continue;
            done[u]= true;

            for(int i=0;i<G[u].size();i++)
            {
                Edge& e= edges[G[u][i]];
                int new_dist=e.dist>pos?1:0;
                if(d[e.to]> d[u]+new_dist)
                {
                    d[e.to] = d[u]+new_dist;
                    p[e.to] = G[u][i];
                    Q.push(HeapNode(d[e.to],e.to) );
                }
            }
        }
        if(d[n-1]>k) return false;
        else return true;
    }
}DJ;

int main(){
    // freopen("in.txt", "r", stdin);
    // freopen("out.txt", "w", stdout);
    while(scanf("%d%d%d",&n,&p,&k)!=EOF){
        DJ.init(n);
        for(int i=0;i<p;i++){
            scanf("%d%d%d",&u,&v,&w);
            DJ.AddEdge(u-1,v-1,w);
            DJ.AddEdge(v-1,u-1,w);
        }
        int res=-1;
        int l=0,r=MAXR,m;
        while(l<=r){
            m=(l+r)/2;
            if(DJ.dijkstra(0,m)==true){//如果这个长度可以完成的话
                r=m-1;
                res=m;
            }else{//如果不可以完成的话
                l=m+1;
            }
        }
        printf("%d\n",res);
    }
    return 0;
}