【LCA转RMQ ST】 HDU 2874 Connections between cities
题意 :给出一个森林 求两点间的最短距离
并查集把一棵树搞在一起
再将每棵树的根搞在root结点上
用LCA转 RMQ 搞
DFS 记录 到 root 的距离
输出是 LEN [ A ] + LEN [ B ] - 2 * LEN [ LCA( A , B ) ];(两个点到根节点的距离再减去公共祖先到根的距离);
注意 : RMQ 中存坐标
#include <cstdio> #include <cstring> #include <cstdlib> #include <string> #include <iostream> #include <algorithm> #include <sstream> #include <cmath> using namespace std; #include <queue> #include <stack> #include <vector> #include <deque> #include <set> #include <map> #include <time.h>; #define cler(arr, val) memset(arr, val, sizeof(arr)) #define FOR(i,a,b) for(int i=a;i<=b;i++) #define IN freopen ("in.txt" , "r" , stdin); #define OUT freopen ("out.txt" , "w" , stdout); typedef long long LL; const int MAXN = 100000+5; const int MAXM = 100000; const int INF = 0x3f3f3f3f; const int mod = 1000000007; int rmq[2*MAXN];//rmq数组,就是欧拉序列对应的深度序列 struct ST { int mm[2*MAXN]; int dp[2*MAXN][20];//最小值对应的下标 void init(int n) { mm[0] = -1; for(int i = 1; i <= n; i++) { mm[i] = ((i&(i-1)) == 0)?mm[i-1]+1:mm[i-1]; dp[i][0] = i; } for(int j = 1; j <= mm[n]; j++) for(int i = 1; i + (1<<j) - 1 <= n; i++) dp[i][j] = rmq[dp[i][j-1]] < rmq[dp[i+(1<<(j-1))][j-1]]?dp[i][j-1]:dp[i+(1<<(j-1))][j-1]; } int query(int a,int b)//查询[a,b]之间最小值的下标 { if(a > b)swap(a,b); int k = mm[b-a+1]; return rmq[dp[a][k]] <= rmq[dp[b-(1<<k)+1][k]]?dp[a][k]:dp[b-(1<<k)+1][k]; } }; //边的结构体定义 struct Edge { int to,next,val; }; Edge edge[MAXN*2]; int tot,head[MAXN],fa[MAXN],len[MAXN]; int F[MAXN*2];//欧拉序列,就是dfs遍历的顺序,长度为2*n-1,下标从1开始 int P[MAXN];//P[i]表示点i在F中第一次出现的位置 int cnt; ST st; bool flag[MAXN],have[MAXN]; void init() { tot = 0; memset(head,-1,sizeof (head)); memset(have,false,sizeof(have)); } void addedge(int u,int v,int w)//加边,无向边需要加两次 { edge[tot].to = v; edge[tot].val = w; edge[tot].next = head[u]; head[u] = tot++; } void dfs(int u,int pre,int dep) { F[++cnt] = u; rmq[cnt] = dep; P[u] = cnt; for(int i = head[u]; i != -1; i = edge[i].next) { int v = edge[i].to; if(v == pre) continue; len[v]=len[u]+edge[i].val; dfs(v,u,dep+1); F[++cnt] = u; rmq[cnt] = dep; } } void LCA_init(int root,int node_num)//查询LCA前的初始化 { cnt = 0; dfs(root,root,0); st.init(2*node_num-1); } int query_lca(int u,int v)//查询u,v的lca编号 { return F[st.query(P[u],P[v])]; } int find(int x) { if(fa[x]==x) return x; return fa[x]=find(fa[x]); } void merge(int x,int y) { int fx=find(x),fy=find(y); if(fx<fy) fa[fx]=fy; else fa[fy]=fx; } int main() { #ifndef ONLINE_JUDGE freopen("in.txt", "r", stdin); // freopen("out.txt", "w", stdout); #endif int n,m,c,a,b,w; while(scanf("%d%d%d",&n,&m,&c)!=EOF) { init(); for(int i=0;i<=n;i++) fa[i]=i; for(int i=0; i<m; i++) { scanf("%d%d%d",&a,&b,&w); addedge(a,b,w); addedge(b,a,w); merge(a,b); } int root=0; for(int i=1; i<=n; i++) find(i); for(int i=1; i<=n; i++) have[fa[i]]=true; for(int i=1; i<=n; i++) if(have[i]) addedge(root,i,0); len[0]=0; LCA_init(root,n+1); for(int i=0; i<c; i++) { scanf("%d%d",&a,&b); if(find(a)!=find(b)) puts("Not connected"); else if(a==b) puts("0"); else printf("%d\n",len[a]+len[b]-2*len[query_lca(a,b)]); } } return 0; }