逃亡的贝贝（二分+最短路）

D-逃亡的贝贝_牛客练习赛104 (nowcoder.com)

 

 题意：

给一个图，k瓶药水，每瓶药水可以使边权降低为原来的114/514向上取整，但一瓶只能用在一条边，询问从起点到终点经过的最大边权的最小值。

思路：

二分边权最大值，对于每一条边，若本来就小于mid则建边为0，若使用药水后小于mid建边为1，否则不建边（设为inf无法通过），跑最短路的最短距离就是使用药水的次数，若<=k，则该mid符合，继续二分。

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<ll,ll> PII;
#define inf 0x3f3f3f3f3f
const ll N=2e6+10;
ll n,m,S,T,k;
ll nextt[N],head[N],to[N],W[N];
ll vis[N],dis[N];
ll cnt;
void add(ll u,ll v,ll w)
{
	to[cnt]=v;
	W[cnt]=w;
	nextt[cnt]=head[u];
	head[u]=cnt++;
}
bool check(ll mid)
{
	//for(int i=0;i<n;i++)head[i]=-1;
	for(int i=0;i<=n;i++)dis[i]=inf;
	for(int i=0;i<=n;i++)vis[i]=0;
	priority_queue<PII,vector<PII>,greater<PII> > q;
	dis[S]=0;//
	q.push({0,S});
	while(q.size())
	{
		PII p=q.top();q.pop();
		ll t=p.second;
		if(vis[t]==1)continue;
		vis[t]=1;
		for(int i=head[t];i!=0;i=nextt[i])
		{
			ll x;//新建的权值 
			ll hao=to[i];
			if(W[i]<=mid)x=0;
			else if(ceil((double)114*W[i]/514)<=mid)x=1;
            //else if(ceil((double)(114*W[i]/514))<=mid)x=1;//不对
		//else if((W[i]*114+513)/514<=mid) x=1;
			else x=inf;
			if(dis[hao]>dis[t]+x)
			{
				dis[hao]=dis[t]+x;
				q.push({dis[hao],hao});
			}
		}
	}
	return dis[T]<=k;
}
int main()
{
	ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	cin>>n>>m>>S>>T>>k;
	ll maxx=-1;

	for(int i=0;i<m;i++)
	{
		ll u,v,w;
		cin>>u>>v>>w;
		//maxx=max(maxx,w);
		add(u,v,w);
		add(v,u,w);
	}
	ll l=0,r=1e9,mid,ans=inf;
	while(l<=r)
	{
		mid=(l+r)/2;
		if(check(mid))
		{
			ans=mid;
			r=mid-1;
		}
		else
			l=mid+1;
	}
	if(ans==inf) cout<<"I really need TS1's time machine again!"<<'\n';
	else cout<<ans<<'\n';
	return 0;
}