P4197 Peaks [克鲁斯卡尔重构树 + 主席树]

部分kruskal重构树内容摘抄于bzt神仙的blog

Problem

\(Bytemountains\)\(n\)座山峰,每座山峰有他的高度\(h_i\) 。有些山峰之间有双向道路相连,共\(M\)条路径,每条路径有一个困难值,这个值越大表示越难走,现在有\(Q\)组询问,每组询问询问从点\(v\)开始只经过困难值小于等于\(x\)的路径所能到达的山峰中第\(k\)高的山峰,如果无解输出\(-1\)

kruskal重构树是用来解决一些诸如“查询从某个点开始 经过边权不超过\(val\)所能到达的节点”的问题
所以很显然 在最小生成树上是最优的。。其他多余的边可以不需要考虑。。
那我们在跑kruskal的时候重新建边。。
克鲁斯卡尔重构树的思想就是在建最小生成树的时候不是直接连边 而是新建一个节点 并把这个节点的值设为边权 然后令两个连通块的代表点分别作为它的左右儿子 然后令这个新节点成为整个连通块的代表点
即如果\(u\) \(v\)不连通
\(u\) \(v\)两点连到新建节点上。。顺便连边。。然后就变成了一棵树。。

sort(q + 1 , q + m + 1 , cmp) ;
  for( int i = 1 ; i <= n ; i ++ ) fa[i] = i ;
  tot = n ;
  for( int i = 1 ; i <= m ; i ++ ) {
    int u = find(q[i].u) , v = find(q[i].v) ;
    if(u == v) continue ;
    val[++ tot] = q[i].w ; fa[tot] = fa[u] = fa[v] = tot ;
    add(tot , u) ; add(tot , v) ;
  }

以下内容来自于 bztminamoto 的 blog

假设现在有四个节点,要求他们的克鲁斯卡尔重构树

我们按最小生成树的方法找,先把边按权值从小到大排序。

然后设第一条边权值为4,连接1和2这两个连通块

然后新建一个节点5,点权设为4,并把1和2作为他的左右儿子

第二条边权值为6,连接3和4这两个连通块

然后新建一个节点6,点权设为6,并把3和4作为他的左右儿子

第三条边权值为7,连接1和2,那么我们就是要把4和6的连通块相连了(这两个是连通块的代表点)

然后新建一个节点7,点权设为7,并把5和6作为他的左右儿子

然后这一棵克鲁斯卡尔重构树就建好了

然后这道题求的是 \(k\)
\(k\) 大 问题一般都是主席树解决。。
所以把新建的树变成dfs序放到主席树上解决就可以了(在线)。。

#include<bits/stdc++.h>
using namespace std ;
const int N = 2e5 + 5 ;
const int M = 5e5 + 5 ;
const int inf = INT_MAX ;
struct node { int v , nxt ; } e[N << 1] ;
struct Node { int u , v , w ; } q[M] , t[N << 1] ;
int n , m , Q ;
int rt[N] , h[N] ;
int ls[N << 5] , rs[N << 5] , sum[N << 5] ;
int val[N] ;
int head[N] , cnt = 0 ;
inline void add(int u , int v) { e[++ cnt] = { v , head[u]} ; head[u] = cnt ; }
inline bool cmp(Node x , Node y) { return x.w < y.w ; }
int fa[N] ; int tot ;
inline int find(int x) { return fa[x] == x ? x : fa[x] = find(fa[x]) ; }
inline void kruskal() {
  sort(q + 1 , q + m + 1 , cmp) ;
  for( int i = 1 ; i <= n ; i ++ ) fa[i] = i ;
  tot = n ;
  for( int i = 1 ; i <= m ; i ++ ) {
    int u = find(q[i].u) , v = find(q[i].v) ;
    if(u == v) continue ;
    val[++ tot] = q[i].w ; fa[tot] = fa[u] = fa[v] = tot ;
    add(tot , u) ; add(tot , v) ;
  }
}
int f[N][22], L[N] , R[N] ;
int b[N] , pop = 0 ;
inline void dfs(int u , int fa) {
  f[u][0] = fa ;
  for( int i = 1 ; i <= 19 ; i ++ ) f[u][i] = f[f[u][i - 1]][i - 1] ;
  L[u] = pop ;
  if(! head[u]) { b[++ pop] = u ; R[u] = pop ; return ; }
  for( int i = head[u] ; i ; i = e[i].nxt) {
    int v = e[i].v ; dfs(v , u) ;
  } R[u] = pop ;
}
inline void build(int l , int r , int & o) {
  o = ++ cnt ;
  if(l == r) return ;
  int mid = l + r >> 1 ;
  build(l , mid , ls[o]) ; build(mid + 1 , r , rs[o]) ;
}
inline void upd(int l , int r , int pre , int & o , int x) {
  o = ++ cnt ;
  ls[o] = ls[pre] ; rs[o] = rs[pre] ; sum[o] = sum[pre] ;
  if(l == r) { sum[o] ++ ; return ; }
  int mid = l + r >> 1 ;
  if(x <= mid) upd(l , mid , ls[pre] , ls[o] , x) ;
  else upd(mid + 1 , r , rs[pre] , rs[o] , x) ;
  sum[o] = sum[ls[o]] + sum[rs[o]] ;
}
inline int find_fa(int x , int k) {
  for(register int i = 19 ; ~ i ; i --)
    if(val[f[x][i]] <= k) x = f[x][i] ;
  return x ;
}
inline int query(int a , int b , int l , int r , int k) {
  if(l == r)
    if(k == sum[b] - sum[a]) return l ;
    else return 0 ;
  int x = sum[rs[b]] - sum[rs[a]] , mid = l + r >> 1 ;
  if(x >= k) return query(rs[a] , rs[b] , mid + 1 , r , k) ;
  else return query(ls[a] , ls[b] , l , mid , k - x) ;
}
signed main() {
  ios :: sync_with_stdio(false) ;
  cin.tie(nullptr) ;
  cout.tie(nullptr) ;
  cin >> n >> m >> Q ;
  val[0] = inf ;
  for( int i = 1 ; i <= n ; i ++ ) cin >> h[i] ;
  for( int i = 1 ; i <= n ; i ++ ) t[i].u = i , t[i].w = h[i] ;
  sort(t + 1 , t + n + 1 , cmp) ;
  for( int i = 1 ; i <= n ; i ++ ) h[t[i].u] = i ;
  for( int i = 1 ; i <= m ; i ++ ) { cin >> q[i].u >> q[i].v >> q[i].w ; }
  kruskal() ;
  dfs(tot , tot) ; cnt = 0 ;
  build(1 , n , rt[0]) ;
  for( int i = 1 ; i <= pop ; i ++)
    upd(1 , n , rt[i - 1] , rt[i] , h[b[i]]) ;
  t[0].w = -1 ;
  for( int i = 1 ; i <= Q ; i ++) {
    int v , x , k ; cin >> v >> x >> k ;
    int fa = find_fa(v , x) ;
    int ans = query(rt[L[fa]] , rt[R[fa]] , 1 , n , k) ;
    cout << t[ans].w << '\n' ;
  }
  return 0 ;
}

猜你喜欢

转载自www.cnblogs.com/Isaunoya/p/11780762.html