算法竞赛Telephone Lines POJ_3662

// 2012t.cpp : 定义控制台应用程序的入口点。

// 算法竞赛Telephone Lines POJ_3662

#include "stdafx.h"

#include <cstdio>

#include <cstring>

#include <string>

#include <cmath>

#include <iostream>

#include <algorithm>

#include <queue>

#include <cstdlib>

#include <vector>

#define inf 0x3f3f3f3f

#define N 100

using namespace std;

struct edge

{

    int to,w;

    edge(){}

    edge(int to,int w):to(to),w(w){}

    bool operator<(const edge &a)const{

        return w>a.w;

    }

};

vector<edge>g[N];

int n,m,k;

int dis[N],vis[N];

int ok(int limit)

{

    for(int i=1;i<=n;i++)dis[i]=inf,vis[i]=0;

   priority_queue <edge>que;

    dis[1]=0;

    que.push(edge(1,0));

    while(!que.empty())

    {

       edge now=que.top();que.pop();

 

        int x=now.to;

        if(vis[x])continue;

        vis[x]=1;

        for(int i=0,sz=g[x].size();i<sz;i++)

        {

            int v=g[x][i].to,w=g[x][i].w>limit?1:0;

            if(dis[v]>dis[x]+w)

            {

                dis[v]=dis[x]+w;

                que.push(edge(v,dis[v]));

            }

        }

    }

//printf("\n\n");

  for(int i=1;i<=n;i++) printf("%d    ",dis[i]);

    return dis[n]<=k;

}

int solve()

{

int count=0;

    int l=0,r=N,ans=-1;

    while(l<=r)

    {

count++;

printf("\n count= %d,  %d,  %d ... %d\t",count, l, r , (l+r)>>1);

        int mid=(l+r)>>1;

        if(ok(mid))r=mid-1,ans=mid;

        else l=mid+1;

    }

    return ans;

}

int main()

{

    //while(scanf("%d%d%d",&n,&m,&k)>0)

    {

n=5;m=7;k=1;

        for(int i=1;i<=n;i++)g[i].clear();

      

            int u,v,w;

/*

1 2 5

3 1 4

2 4 8

3 2 3

5 2 9

3 4 7

4 5 6

*/

u=1;v=2;w=5;

            g[u].push_back(edge(v,w));

            g[v].push_back(edge(u,w));

u=3;v=1;w=4;

            g[u].push_back(edge(v,w));

            g[v].push_back(edge(u,w));

u=2;v=4;w=8;

            g[u].push_back(edge(v,w));

            g[v].push_back(edge(u,w));

u=3;v=2;w=3;

            g[u].push_back(edge(v,w));

            g[v].push_back(edge(u,w));

u=5;v=2;w=9;

            g[u].push_back(edge(v,w));

            g[v].push_back(edge(u,w));

u=3;v=4;w=7;

            g[u].push_back(edge(v,w));

            g[v].push_back(edge(u,w));

u=4;v=5;w=6;

            g[u].push_back(edge(v,w));

            g[v].push_back(edge(u,w));

        printf("\n\n%d\n",solve());

    }

}