架设电话线

题目描述

   Farmer John打算将电话线引到自己的农场，但电信公司并不打算为他提供免费服务。于是，FJ必须为此向电信公司支付一定的费用。 FJ的农场周围分布着N(1 <= N <= 1,000)根按1..N顺次编号的废弃的电话线杆，任意两根电话线杆间都没有电话线相连。一共P(1<= P <= 10,000)对电话线杆间可以拉电话线，其余的那些由于隔得太远而无法被连接。第i对电话线杆的两个端点分别为A_i、B_i，它们间的距离为L_i(1 <= L_i <= 1,000,000)。数据中保证每对{A_i，B_i}最多只出现1次。编号为1的电话线杆已经接入了全国的电话网络，整个农场的电话线全都连到了编号为N的电话线杆上。也就是说，FJ的任务仅仅是找一条将1号和N号电话线杆连起来的路径，其余的电话线杆并不一定要连入电话网络。经过谈判，电信公司最终同意免费为FJ连结K(0 <= K < N)对由FJ指定的电话线杆。对于此外的那些电话线，FJ需要为它们付的费用，等于其中最长的电话线的长度（每根电话线仅连结一对电话线杆）。如果需要连结的电话线杆不超过K对，那么FJ的总支出为0。请你计算一下，FJ最少需要在电话线上花多少钱。

输入   第1行: 3个用空格隔开的整数：N，P，以及K 
       第2..P+1行: 第i+1行为3个用空格隔开的整数：A_i，B_i，L_i

输出  第1行: 输出1个整数，为FJ在这项工程上的最小支出。如果任务不可能完成，输出-1

 样例输入 

5 7 1

1 2 5

3 1 4

2 4 8

3 2 3

5 2 9

3 4 7

4 5 6

样例输出 

4

解题思路：

      先找到所有可搭电话线中的最大值，然后，二分0～max之间的答案。进行SPFA，如果碰到那些比预计值mid大的电线，即让电信公司来，如果1～n的间的电信公司要修的电线>k的话，即返回false，否则返回true。

代码：（请不要直接拷贝哦）

#include <cstdio>
#include <queue>
#include <cstring>
int first[20005],next[20005],v[20005],w[20005],dis[1005];//SPFA的标配，这里要记录边的值，所以多了一个w数组
bool vis[1005];
int n,p,k,t,mx;
using namespace std;
inline void add(int x,int y,int z)//为SPFA服务
{
	next[++t]=first[x];
	first[x]=t;
	v[t]=y;
	w[t]=z;
}
bool spfa(int x)
{
	memset(vis,0,sizeof(vis));
	memset(dis,0x3f3f3f3f,sizeof(dis));//初始化
	queue<int> q;
	vis[1]=1;
	q.push(1);
	dis[1]=0;
	while (!q.empty())//队列不为空
	{
		int y=q.front();//取第一个
		q.pop();//弹出
		vis[y]=0;
		for (int i=first[y];i;i=next[i])//SPFA开始啦
		{
			int j=v[i],now=dis[y];
			if (w[i]>x) now++;//超出预计，让电信公司来修
			if (now<dis[j])//判断是否可以更新答案
			{
				dis[j]=now;
				if (!vis[j])
				{
					vis[j]=1;
					q.push(j);//加入至队列中
				}
			}
		}
	}
	return dis[n]<=k;
}
int ef()
{
	int s=-1,l=0,r=mx;//s赋-1是因为待会就不用加特判了，如果没有解，即返回-1
	while (l<=r)
	{
		int mid=l+r>>1;//取中值
		if (spfa(mid)) s=mid,r=mid-1; else l=mid+1;//s记录答案
	}
	return s;
}
int main()
{
    scanf("%d%d%d",&n,&p,&k);
	for (int i=1;i<=p;i++)
	{
		int x,y,z;
		scanf("%d%d%d",&x,&y,&z);
		add(x,y,z);
		add(y,x,z);
		if (z>mx) mx=z;//记待会要进行二分的最大值
	}
    printf("%d\n",ef());
    return 0;
}