Description
\(n\) 个点,\(m\) 条边,\(k\) 个充电站,\(Q\) 次询问,每次询问给出两个点 \(p_1\) , \(p_2\) ,两个点一定在充电站上,问从 \(p_1\) 到 \(p_2\) 的电池容量最少需要多少
前置知识
- 最短路算法
- 最小生成树算法
- 倍增求LCA / 树链剖分
Solution
考虑如何求出任意一个点到离它最近的充电站的距离?
暴力的做法是对每一个充电站跑一边单源最短路,但复杂度实在承受不了
有一个很巧妙的处理方法是,我们可以建一个超级源点,连向所有充电站,权值设为 \(0\),然后在新图上跑一个最短路,就能求出所有点到离它最近的充电站得距离了
有什么用?别急,让我们接着往下看
设走到一个点 \(i\),剩余电量为 \(x\), 那么一定有 \(x > dis_i\) (不管是中途充电还是到达终点均成立,因为保证了终点也是充电站)
假设我们要求的电容量为 \(c\),那么三者关系满足:
\[c \ge x \ge dis_i \]因为走到 \(i\) 点至少消耗了 \(dis_i\) 的电(最少的情况是以对 \(i\) 来说它的起点刚好是最近的充电站来说的),那么上面的不等式可以改为:
\[c - dis_i \ge x \ge dis_i \]设下一个点为 \(j\),边权为 \(w_{i,j}\),同理有:
\[c - dis_j \ge x - w_{i,j} \ge dis_j \]结合上面两个式子,进行合并得到 \(dis_j \le x - w_{i,j}\),即:
\[dis_j \le c - dis_i - w_{i,j} \]交换一下位置:
\[c \ge dis_i + dis_j + w_{i,j} \]这个不等式的含义为,从 \(i\) 点到 \(j\) 点所需最小的电池容量为 \(dis_i + dis_j + w_{i,j}\),也就是说没条边的最小限制已经求出来了,用这个最小限制就可以更新图上所有的边
那么每次询问一个起点 \(a\) 到 \(b\) 的最小电池容量,就是找一条从 \(a\) 点到 \(b\) 点的路径,使得这个路径上的最大值最小
很像二分? 但用不着那么麻烦
因为这条路径一定在最小生成树上,同时在这个最小生成树上每对点都有一个简单路径,最小生成树的构建过程也保证了这条简单路径上的最大值就是答案
那么用LCA+倍增法求出路径上的最大值即可(也可以用树链剖分来实现,前提是码力足够强)
具体实现来看代码
Code
/*
Work by: Suzt_ilymics
Knowledge: ??
Time: O(??)
思路极其鬼畜:
1、最短路
2、生成树
3、倍增+LCA
4、查询
*/
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
//#include<time.h>
#define LL long long
#define int long long
#define orz cout<<"lkp AK IOI!"<<endl
using namespace std;
const int MAXN = 4e5+5;
const int MAXM = 6e5+5;
const int INF = 1e9+7;
const int mod = 1e9+7;
struct edge{
int from, to, w, nxt;
}e[MAXM << 2], E[MAXM], E2[MAXM << 1];
int head[MAXN], num_edge = 0;
int Head[MAXN], Num_edge = 0;
struct node{
int bh, val;
bool operator < (const node &b) const { return val > b.val; }
};
int n, m, k, Q, cnt;
int dis[MAXN], fa[MAXN], f[MAXN][22], dep[MAXN], maxm[MAXN][22];
bool vis[MAXN];
priority_queue<node> q;
int read(){
int s = 0, f = 0;
char ch = getchar();
while(!isdigit(ch)) f |= (ch == '-'), ch = getchar();
while(isdigit(ch)) s = (s << 1) + (s << 3) + ch - '0' , ch = getchar();
return f ? -s : s;
}
bool cmp(edge x, edge y){ return x.w < y.w; }
void add_edge(int from, int to, int w){ e[++num_edge] = (edge){from, to, w, head[from]}, head[from] = num_edge; }
void Add_edge(int from, int to, int w){ E2[++Num_edge] = (edge){from, to, w, Head[from]}, Head[from] = Num_edge; }
int find(int x) { return fa[x] == x ? x : fa[x] = find(fa[x]); }
void dij(){//跑一边dij求出所有点到最近的充电站的距离
memset(dis, 0x3f, sizeof dis);
dis[0] = 0;
q.push((node){0, 0});
while(!q.empty()){
node u = q.top(); q.pop();
if(vis[u.bh]) continue;
vis[u.bh] = true;
for(int i = head[u.bh]; i; i = e[i].nxt){
int v = e[i].to;
if(dis[v] > dis[u.bh] + e[i].w){
dis[v] = dis[u.bh] + e[i].w;
if(!vis[v]) q.push((node){v, dis[v]});
}
}
}
}
void kruskal(){//建出更新权值后的图的最小生成树
for(int i = 1; i <= n; ++i) fa[i] = i;//重置父亲
for(int i = 1; i <= m; ++i){
// cout<<E[i].from<<" "<<E[i].to<<" "<<E[i].w<<"lkp"<<endl;
int uf = find(E[i].from), vf = find(E[i].to);
if(uf != vf){
// orz;
// if(rand() % 2) fa[uf] = vf;//随机连父亲,随机化优化复杂度
// else
fa[vf] = uf;
Add_edge(E[i].from, E[i].to, E[i].w), Add_edge(E[i].to, E[i].from, E[i].w);//建出最小生成树来
cnt++;
if(cnt == n - 1) return ;//如果建了n - 1条边,就结束
}
}
}
void dfs(int u, int fa){//dfs预处理lca ,注意这里跑的是新图
f[u][0] = fa;
for(int i = Head[u]; i; i = E2[i].nxt){
int v = E2[i].to;
if(v == fa) continue;
dep[v] = dep[u] + 1;
maxm[v][0] = E2[i].w;
// cout<<maxm[v][0]<<endl;
dfs(v, u);
}
}
void init(){//倍增法预处理lca,顺便维护一个最大值
for(int i = 1; i <= 20; ++i){
for(int j = 1; j <= n; ++j){
f[j][i] = f[f[j][i - 1]][i - 1];
maxm[j][i] = max(maxm[j][i - 1], maxm[f[j][i - 1]][i - 1]);
// cout<<maxm[j][i]<<endl;
}
}
}
int get_max(int x, int y){//可以直接在求两点lca的过程中求出两点简单路径的最大值
int ans = 0;
if(dep[x] < dep[y]) swap(x, y);
for(int i = 20; i >= 0; --i){
if(dep[f[x][i]] < dep[y]) continue;
ans = max(ans, maxm[x][i]);
x = f[x][i];
}
if(x == y) return ans;
for(int i = 20; i >= 0; --i){
if(f[x][i] == f[y][i]) continue;
ans = max(ans, max(maxm[x][i], maxm[y][i]));
// cout<<ans<<endl;
x = f[x][i];
y = f[y][i];
}
ans = max(ans, max(maxm[x][0], maxm[y][0]));
return ans;
}
signed main()
{
// srand(time(0));
n = read(), m = read(), k = read(), Q = read();
for(int i = 1, u, v, w; i <= m; ++i){
u = read(), v = read(), w = read();
add_edge(u, v, w), add_edge(v, u, w);
E[i].from = u, E[i].to = v, E[i].w = w;
}
for(int i = 1; i <= k; ++i){
add_edge(0, i, 0), add_edge(i, 0, 0);
}
dij();
// for(int i = 1; i <= n; ++i) cout<<i<<" "<<dis[i]<<" lkp"<<endl;
for(int i = 1; i <= m; ++i){//更新边的权值
E[i].w += dis[E[i].from] + dis[E[i].to];
}
sort(E + 1, E + m + 1, cmp);//按边权大小由小到大排序
kruskal();
// memset(minn, 0x7f, sizeof minn);
dep[1] = 1;
dfs(1, -1);
init();
for(int i = 1, u, v; i <= Q; ++i){
u = read(), v = read();
cout<<get_max(u, v)<<endl;//直接输出所求结果即可
}
return 0;
}