【SPFA】【二分】架设电话线

$Link$

$luoguP1948$

$Description$

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最少需要在电话线上花多少钱。

$Input$

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

$Output$

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

$Sample$ $Input$

5 7 1
1 2 5
3 1 4
2 4 8
3 2 3
5 2 9
3 4 7
4 5 6

$Sample$ $Output$

4

$Hint$

输入说明:
一共有5根废弃的电话线杆。电话线杆1不能直接与电话线杆4、5相连。电话线杆5不能直接与电话线杆1、3相连。其余所有电话线杆间均可拉电话线。电信公司可以免费为FJ连结一对电话线杆。

输出说明:
FJ选择如下的连结方案：1->3；3->2；2->5，这3对电话线杆间需要的电话线的长度分别为4、3、9。FJ让电信公司提供那条长度为9的电话线，于是，他所需要购买的电话线的最大长度为4。
数据点1~4：N<=10,P<=20;
数据点5~6：N<=50,P<=200;
数据点7~9：N<=1000,P<=1000;
数据点10~11：N<=1000,P<=5000;
数据点12~13：N<=1000,P<=10000;

$Train$ $of$ $Thought$

我们二分最终FJ要付的钱$mid$，那么大于这个$mid$的电路就是我们要给公司接的电路，所以就设边权为1
然后小于的就设为0，然后跑SPFA，如果$dis[n]$>$k$的话，那么就是不可行的，然后就一直二分就好了

$Code$

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>

using namespace std;

int n, t, p, k, flag, ans;
int h[10001], dis[10001], c[10001];

struct node
{
	int to, next, val, q;
}w[20001];

void add(int x, int y, int z)
{
	w[++t] = (node){y, h[x], z, 0}; h[x] = t;
	w[++t] = (node){x, h[y], z, 0}; h[y] = t;
	
}

void spfa()
{
	memset(c, 0, sizeof(c));
	for (int i = 1; i <= n; ++i)
		dis[i] = 1e9;
	dis[1] = 0;
	c[1] = 1;
	queue<int>Q;
	Q.push(1);
	while(Q.size())
	{
		int pq = Q.front();
		Q.pop();
		for (int i = h[pq]; i; i = w[i].next)
		{
			int y = w[i].to;
			if (dis[y] > dis[pq] + w[i].q)
			{
				dis[y] = dis[pq] + w[i].q;
				if (!c[y]) {
					c[y] = 1;
					Q.push(y);
				}
			}
		}
		c[pq] = 0;
	}
}

bool check(int x)
{
	for (int i = 1; i <= t; ++i)
		if (w[i].val > x) w[i].q = 1;
		else w[i].q = 0;
	spfa();
	if (dis[n] == 1e9) flag = 1;
	if (dis[n] <= k) return 1;
	else return 0; 
}

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);
	}	
	int l = 0, r = 1000000;
	while (l <= r)
	{
		int mid = (l + r) / 2;
		if (check(mid)) r = mid - 1, ans = mid;
		else l = mid + 1;
		if (flag) break;
	}
	if (flag) printf("-1");
	else printf("%d", ans);
}