poj1986 Distance Queries(lca又是一道模版题)
生活随笔
收集整理的這篇文章主要介紹了
poj1986 Distance Queries(lca又是一道模版题)
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
題目鏈接:http://poj.org/problem?id=1986
題意:就是老問題求val[u]+val[v]-2*val[root]就行。還有這題沒有給出不聯(lián)通怎么輸出那么題目給出的數據一定
是聯(lián)通的。
?
題解:就是單純的lca。
?
#include <iostream> #include <cstring> #include <vector> #include <cstdio> using namespace std; const int M = 8e4 + 10; vector<pair<int , int> >vc[M]; int p[M][20] , val[M] , deep[M]; void dfs(int pos , int pre , int d) {p[pos][0] = pre;deep[pos] = d;int len = vc[pos].size();for(int i = 0 ; i < len ; i++) {int u = vc[pos][i].first;if(u != pre) {val[u] += val[pos] + vc[pos][i].second;dfs(u , pos , d + 1);}} } void init(int n) {for(int i = 0 ; i < 18 ; i++) {for(int j = 1 ; j <= n ; j++) {if(p[j][i] == -1) {p[j][i + 1] = -1;}else {p[j][i + 1] = p[p[j][i]][i];}}} } int lca(int a , int b) {if(deep[a] < deep[b]) {swap(a , b);}int d = deep[a] - deep[b];for(int i = 0 ; i < 18 ; i++) {if(d & (1 << i)) {a = p[a][i];}}if(a == b) {return a;}for(int i = 17 ; i >= 0 ; i--) {if(p[a][i] != p[b][i] && p[a][i] != -1) {a = p[a][i];b = p[b][i];}}return p[a][0]; } int main() {int n , m , u , v , w , k;char cp[10];while(scanf("%d%d" , &n , &m) != EOF) {for(int i = 1 ; i <= n ; i++) {val[i] = 0;vc[i].clear();}for(int i = 1 ; i < n ; i++) {scanf("%d%d%d%s" , &u , &v , &w , cp);vc[u].push_back(make_pair(v , w));vc[v].push_back(make_pair(u , w));}memset(p , -1 , sizeof(p));dfs(1 , -1 , 1);init(n);scanf("%d" , &k);while(k--) {scanf("%d%d" , &u , &v);printf("%d\n" , val[u] + val[v] - 2 * val[lca(u , v)]);}}return 0; }轉載于:https://www.cnblogs.com/TnT2333333/p/6737644.html
創(chuàng)作挑戰(zhàn)賽新人創(chuàng)作獎勵來咯,堅持創(chuàng)作打卡瓜分現金大獎總結
以上是生活随笔為你收集整理的poj1986 Distance Queries(lca又是一道模版题)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 装饰器,迭代器,生成器
- 下一篇: Mybayis的项目使用的Mapping