其他分享
首页 > 其他分享> > 圆方树总结

圆方树总结

作者:互联网

解题:

洛谷模板题:

#include <cmath>
#include <queue>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;

const int maxn = 40000 + 10;
int n, m, q, head_g[maxn], head_t[maxn], st[17][maxn], edge_num_g, edge_num_t, dfn_num, col_num, top;
int dfn[maxn], low[maxn], sta[maxn];
long long deep[maxn], dis[maxn], sum[maxn], sccsum[maxn];

struct Edge { int v, nxt; long long w; } edge_g[maxn << 2], edge_t[maxn << 2];

inline void Add_edge_g(int u, int v, long long w) {
  edge_g[++edge_num_g].v = v, edge_g[edge_num_g].w = w, edge_g[edge_num_g].nxt = head_g[u], head_g[u] = edge_num_g;
}

inline void Add_edge_t(int u, int v, long long w) {
  edge_t[++edge_num_t].v = v, edge_t[edge_num_t].w = w, edge_t[edge_num_t].nxt = head_t[u], head_t[u] = edge_num_t;
}

inline void Tarjan(int x, int p) {
  dfn[x] = low[x] = ++dfn_num, sta[++top] = x;
  for(int i = head_g[x]; i; i = edge_g[i].nxt) if( edge_g[i].v != p ) {
    if( dfn[edge_g[i].v] == 0 ) {
      sum[edge_g[i].v] = sum[x] + edge_g[i].w, Tarjan(edge_g[i].v, x), low[x] = min(low[x], low[edge_g[i].v]);
      if( low[edge_g[i].v] > dfn[x] ) Add_edge_t(x, edge_g[i].v, edge_g[i].w), Add_edge_t(edge_g[i].v, x, edge_g[i].w);  // 树边,圆圆边加入
    }
    else if( dfn[edge_g[i].v] < low[x] ) {  // 返祖边,得环,建立方点及圆方边
      sccsum[++col_num] = sum[x] - sum[edge_g[i].v] + edge_g[i].w;  // 得环长,col_num 认为是方点编号
      for(int j = top; sta[j] != edge_g[i].v; --j) {
        int _w = min(sum[sta[j]] - sum[edge_g[i].v], sccsum[col_num] - sum[sta[j]] + sum[edge_g[i].v]); // 到此环的根的距离,即圆方边边权
        Add_edge_t(n + col_num, sta[j], _w), Add_edge_t(sta[j], n + col_num, _w);
      }
      Add_edge_t(n + col_num, edge_g[i].v, 0), Add_edge_t(edge_g[i].v, n + col_num, 0); // 环的根所对应的圆方边权为 0
      low[x] = dfn[edge_g[i].v];
    }
  }
  --top;
}

inline void Deep_fs(int x, int p) {
  for(int i = 1; i < 17; ++i) st[i][x] = st[i - 1][st[i - 1][x]];
  for(int i = head_t[x]; i; i = edge_t[i].nxt) if( edge_t[i].v != p ) {
    st[0][edge_t[i].v] = x;
    dis[edge_t[i].v] = dis[x] + edge_t[i].w, deep[edge_t[i].v] = deep[x] + 1, Deep_fs(edge_t[i].v, x);
  }
}

inline long long Querry(int x, int y) {
  int lca = 0, res = dis[x] + dis[y];
  if( deep[x] < deep[y] ) swap(x, y);
  for(int i = 16; i >= 0; --i) if( deep[st[i][x]] >= deep[y] ) x = st[i][x];
  if( x == y ) lca = x;
  else {
    for(int i = 16; i >= 0; --i) if( st[i][x] != st[i][y] ) x = st[i][x], y = st[i][y];
    lca = st[0][x];
  }
  if( lca <= n ) return res - (dis[lca] << 1);
  if( dfn[x] > dfn[y] ) swap(x, y);
  return res - dis[x] - dis[y] + min(sum[y] - sum[x], sccsum[lca - n] - sum[y] + sum[x]);
}

int main(int argc, char const *argv[])
{
  scanf("%d%d%d", &n, &m, &q);
  for(int u, v, w, i = 1; i <= m; ++i) scanf("%d%d%d", &u, &v, &w), Add_edge_g(u, v, w), Add_edge_g(v, u, w);
  deep[1] = 1, Tarjan(1, 0), Deep_fs(1, 0);
  for(int u, v, i = 1; i <= q; ++i) scanf("%d%d", &u, &v), printf("%lld\n", Querry(u, v));

  return 0;
}

标签:总结,num,int,edge,st,方点,圆方树,maxn
来源: https://www.cnblogs.com/nanjoqin/p/11293720.html