【2022省选模拟】过路费——贪心、最短路

此题没有链接，随便提供一个

题目描述

题解

这题其实是搬的Gym - 101630J。

观察数据范围，发现同时记录两维的 $O(n^3\log n)$ DIJ做法过不了，需要想一个 $O(n^2\log n)$ 以下的做法。

首先，原先的记录两维的DIJ可以变为枚举第 $k$ 大的边 $w_k$，然后跑DIJ只记录一维，即过程中选取的边的数量。

考虑怎么把这一维去掉。这里我们不妨用类似WQS二分中的思想（这题只是容易让人联想到WQS二分，但并不是用它做的），在DIJ的过程中仍然只选边权 $\ge w_k$ 的边，但是计算的时候，我们把每条边减去 $w_k$。

考虑这样直接跑出来的最短路 $f$，假设选了 $x$ 条边：

• 若 $x\ge k$，那么我们把选的边中最小的那 $x-k$ 条剔除后，一定有一个更优的解，即答案对应的 $w_k$ 一定 $\ge$ 此时枚举的 $w_k$。此时在所有枚举的 $w_k$ 不小于此时的 $w_k$ 的解中，一定有一个最小的 $f+w_k\ast k$ 是等于我们要求的最优解的。
• 若 $x<k$，那么这条路一定会再选几个 $<w_k$ 的边才合法。这时一定存在一个 $w_k<此时枚举的w_k$ 的解更优，并且解的大小是 $<f+w_k\ast k$ 的。

所以我们只需要枚举所有 $m$ 条边作为 $w_k$ 来跑DIJ，然后取 $f+w_k\ast k$ 最小的那个作为答案即可，时间是 $O(n^2\log n)$。

代码

代码中还记录了选的边的数量，但其实是多此一举。

#include<bits/stdc++.h>//JZM yyds!!
#define ll long long
#define uns unsigned
#define IF (it->first)
#define IS (it->second)
#define END putchar('\n')
using namespace std;
const int MAXN=2005;
const ll INF=1e17;
inline ll read(){
	ll x=0;bool f=1;char s=getchar();
	while((s<'0'||s>'9')&&s>0){if(s=='-')f^=1;s=getchar();}
	while(s>='0'&&s<='9')x=(x<<1)+(x<<3)+(s^48),s=getchar();
	return f?x:-x;
}
int ptf[50],lpt;
inline void print(ll x,char c='\n'){
	if(x<0)putchar('-'),x=-x;
	ptf[lpt=1]=x%10;
	while(x>9)x/=10,ptf[++lpt]=x%10;
	while(lpt)putchar(ptf[lpt--]^48);
	if(c>0)putchar(c);
}
inline ll lowbit(ll x){return x&-x;}

int n,m,k,S,T;
struct itn{
	ll a;int b;itn(){}
	itn(ll A,int B){a=A,b=B;}
	inline itn operator+(const ll&w)const{return itn(a+w,b+1);}
	inline bool operator<(const itn&y)const{
		if(a^y.a)return a>y.a;
		else return b>y.b;
	}
}f[MAXN];
#define pitn pair<itn,int>
#define fi first
#define se second
priority_queue<pitn>q;
struct edge{
	int v,to;ll w;edge(){}
	edge(int V,int T,ll W){v=V,to=T,w=W;}
}e[MAXN];
int EN,G[MAXN];
inline void addedge(int u,int v,ll w){
	e[++EN]=edge(v,G[u],w),G[u]=EN;
}
inline itn solve(ll w){
	while(!q.empty())q.pop();
	for(int i=1;i<=n;i++)f[i]=itn(INF,0);
	f[S]=itn(0,0),q.push(pitn(f[S],S));
	while(!q.empty()){
		itn pf=q.top().fi;
		int u=q.top().se;q.pop();
		if(pf<f[u])continue;
		for(int i=G[u];i;i=e[i].to){
			int v=e[i].v;
			itn to=f[u];
			if(e[i].w>=w)to=to+(e[i].w-w);
			if(f[v]<to)f[v]=to,q.push(pitn(f[v],v));
		}
	}return f[T];
}
signed main()
{
	freopen("fee.in","r",stdin);
	freopen("fee.out","w",stdout);
	n=read(),m=read(),k=read();
	S=read(),T=read();
	for(int i=1;i<=m;i++){
		int u=read(),v=read();
		addedge(u,v,read());
	}
	ll ans=solve(0).a;
	for(int i=1;i<=EN;i++){
		itn g=solve(e[i].w);
		if(g.b>=k)ans=min(ans,g.a+k*e[i].w);
	}
	print(ans);
	return 0;
}