AcWing 340通信线路（dijkstra+二分法找答案）

题目

在郊区有 N 座通信基站，P 条 双向 电缆，第 i 条电缆连接基站 Ai 和 Bi。

特别地，1 号基站是通信公司的总站，N 号基站位于一座农场中。

现在，农场主希望对通信线路进行升级，其中升级第 i 条电缆需要花费 Li。

电话公司正在举行优惠活动。

农产主可以指定一条从 1 号基站到 N 号基站的路径，并指定路径上不超过 K 条电缆，由电话公司免费提供升级服务。

农场主只需要支付在该路径上剩余的电缆中，升级价格最贵的那条电缆的花费即可。

求至少用多少钱可以完成升级。

输入格式
第 1 行：三个整数 N，P，K。

第 2…P+1 行：第 i+1 行包含三个整数 Ai,Bi,Li。

输出格式
包含一个整数表示最少花费。

若 1 号基站与 N 号基站之间不存在路径，则输出 −1。

数据范围
0≤K<N≤1000,
1≤P≤10000,
1≤Li≤1000000
输入样例：
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

题意分析：
让你找到最短路径，在该路径中权值最大的一条边，因为K条边可以免费，那么就是第K+1大的边。
解题思路

那么我们怎么记录有多少个值大于这个mid的呢？
我们可以将权值分为两部分，如果这个权值大于mid那么权值变为1，否则权值变为0。

提交传送门

实现代码

#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;

typedef pair<int,int> PII;

const int inf=0x3f3f3f3f;
const int N=1005,M=1e4+5;

struct no{
	int to,w,next;
}e[M<<1];
int head[N],tot;
void add(int u,int v,int w)
{
	e[tot].to=v;
	e[tot].w=w;
	e[tot].next=head[u];
	head[u]=tot++;
}
int dis[N];
int book[N];
int n,m,k;

int dij(int w)
{
	for(int i=1; i<=n; i++) dis[i]=inf,book[i]=0;
	int u=1;
	dis[u]=0;
	priority_queue< PII,vector<PII>,greater<PII> > pque;//升序排列 也即是从小到大排列 
	pque.push({0,1});
	
	while(pque.size())
	{
		PII hd=pque.top();
		pque.pop();
		
		int ver=hd.second;
		if(book[ver]) continue;
		book[ver]=1;
		
		for(int i=head[ver]; i!=-1; i=e[i].next)
		{
			int go=e[i].to;
			if(dis[go]>dis[ver]+(e[i].w>w?1:0))//如果当前权值大于答案则把权值变为1也就是加1，相反同理 
			{
				dis[go]=dis[ver]+(e[i].w>w?1:0);
				pque.push({dis[go],go});	
			}	
		}
	}
	if(dis[n]<=k) return 1;
	else return 0; 
}
int main()
{
	memset(head,-1,sizeof head);
	cin>>n>>m>>k;
	while(m--)
	{
		int a,b,c;
		cin>>a>>b>>c;
		add(a,b,c);
		add(b,a,c);		
	}
	int l=0;
	int r=1e6+1;
	while(r>l)//二分找答案 //范围：0...1e6+1 
	{
		int mid=r+l>>1;
		if(dij(mid)) r=mid;
		else l=mid+1;
	}
	if(r!=1e6+1) cout<<r<<endl;
	else cout<<-1<<endl;
	
	return 0;
}