ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

#805. 【模板】最小瓶颈生成树(数据加强版)

2022-04-26 21:34:38  阅读:143  来源: 互联网

标签:dep return 加强版 int res void fa 805 模板


考虑最小瓶颈生成树的性质

因此我们可以直接在最小生成树上求 \(LCA\) 以及 路径上的边的最大值

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

template <typename T> void chkmax(T &x, T y) { x = x >= y ? x : y; }
template <typename T> void chkmin(T &x, T y) { x = x <= y ? x : y; }

const int N = 100001, M = 1000001;

struct Edg {
  int u, v, w;
  bool operator <(const Edg &T) const {
    return w < T.w;
  }
}edge[M];

int n, m, Q;
int h[N], e[N * 2], ne[N * 2], w[N * 2], idx;
int p[N];
int q[N], dep[N], fa[N][20], mx[N][20];

int find(int x) {
  if(p[x] != x) p[x] = find(p[x]);
  return p[x];
}

void add(int a, int b, int c) {
  e[idx] = b, ne[idx] = h[a], w[idx] = c, h[a] = idx ++;
}

void bfs(int root) {
  memset(dep, 0x3f, sizeof dep);
  dep[0] = 0; dep[root] = 1;
  int hh = 0, tt = 0; q[0] = root;
  while(hh <= tt) {
    int t = q[hh ++];
    for(int i = h[t]; ~i; i = ne[i]) {
      int j = e[i];
      if(dep[j] <= dep[t] + 1) continue;
      dep[j] = dep[t] + 1;
      fa[j][0] = t;
      mx[j][0] = w[i];
      q[++ tt] = j;
      for(int k = 1; k < 20; k ++ ) {
        fa[j][k] = fa[fa[j][k - 1]][k - 1];
        mx[j][k] = max(mx[j][k - 1], mx[fa[j][k - 1]][k - 1]);
      }
    }
  }
}

int lca(int a, int b) {
  if(dep[a] < dep[b]) swap(a, b);
  for(int k = 19; k >= 0; k -- ) {
    if(dep[fa[a][k]] >= dep[b]) {
      a = fa[a][k];
    }
  }
  if(a == b) return a;
  for(int k = 19; k >= 0; k -- ) {
    if(fa[a][k] != fa[b][k]) {
      a = fa[a][k];
      b = fa[b][k];
    }
  }
  return fa[a][0];
}

int query (int u, int v) {
  int res = -1e9;
  for(int i = 19; i >= 0; i -- ) {
    if(dep[fa[u][i]] >= dep[v]) {
      res = max(res, mx[u][i]);
      u = fa[u][i];
    }
  }
  return res;
}

void solve() {

  scanf("%d%d", &n, &m);

  for(int i = 0; i <= n; i ++ ) h[i] = -1, p[i] = i;
  
  for(int i = 0; i < m; i ++ ) {
    int u, v, w; scanf("%d%d%d", &u, &v, &w);
    edge[i] = {u, v, w};
  }

  sort(edge, edge + m);

  for(int i = 0; i < m; i ++ ) {
    int u = edge[i].u, v = edge[i].v, w = edge[i].w; 
    int fu = find(u), fv = find(v);
    if(fu == fv) continue;
    add(u, v, w); add(v, u, w);
    p[fu] = fv; 
  }

  bfs(1);

  scanf("%d", &Q);

  while(Q -- ) {
    int u, v; scanf("%d%d", &u, &v);
    int p = lca(u, v);
    printf("%d\n", max(query(u, p), query(v, p)));
  }

}

int main() {
  solve();
  return 0;
}

标签:dep,return,加强版,int,res,void,fa,805,模板
来源: https://www.cnblogs.com/c972937/p/16196599.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有