F 简单图上操作
题意:在一个无向图中,每条边都有权值,我们有k次跳过权值的车票超过v的边,问你从起点s到终点t,v的最小值是多少。
题解:二分答案,我们维护一个dp[u]代表从起点s到节点u最少要花费的车票数量,那么我们只要判断dp[t]最后是否大于k就能知道答案是否合法。
#include<bits/stdc++.h>
using namespace std;
const int maxn = 2e5+10,inf = 1e9+7;
struct edge{
int to,next,w;
}e[maxn*4];
int cnt,head[maxn],d[maxn];
int n,m,k,s,t,mx;
void addEdge(int u,int v,int w)
{
e[++cnt].to = v ;
e[cnt].next = head[u];
e[cnt].w = w;
head[u] = cnt;
}
void bfs()
{
for(int i=1;i<=n;i++) d[i] = inf;
queue<int> q;
d[s]= 0;
q.push(s);
while(!q.empty())
{
int u = q.front();
q.pop();
if(u==t) return ;
for(int i=head[u];i!=-1;i=e[i].next)
{
int v = e[i].to;
int tmp = 0;
if(e[i].w>mx) tmp = 1;
if(d[u]+tmp<d[v]) {
d[v] = d[u]+tmp;
if(d[v]<=k) q.push(v);
}
}
}
}
int ok(int x)
{
mx = x;
bfs();
if(d[t]<=k) return 1;
return 0;
}
int main()
{
scanf("%d%d%d%d%d",&n,&m,&k,&s,&t);
memset(head,-1,sizeof(head));
for(int i=1;i<=m;i++){
int u,v,w; scanf("%d%d%d",&u,&v,&w);
addEdge(u,v,w);
addEdge(v,u,w);
}
int l = 0,r = inf;
while(l<=r)
{
int m = (l+r)/2;
if(ok(m)) r = m-1;
else l = m+1;
}
printf("%d\n",l);
}