1. 程式人生 > >樹講解——牧場行走( lca )

樹講解——牧場行走( lca )

dfs ++ turn www col 整數 string pan mem

大視野 1602: [Usaco2008 Oct]牧場行走

Time Limit: 5 Sec Memory Limit: 64 MB
Submit: 1947 Solved: 1021
[Submit][Status][Discuss]

Description

N頭牛(2<=n<=1000)別人被標記為1到n,在同樣被標記1到n的n塊土地上吃草,第i頭牛在第i塊牧場吃草。 這n塊土地被n-1條邊連接。 奶牛可以在邊上行走,第i條邊連接第Ai,Bi塊牧場,第i條邊的長度是Li(1<=Li<=10000)。 這些邊被安排成任意兩頭奶牛都可以通過這些邊到達的情況,所以說這是一棵樹。 這些奶牛是非常喜歡交際的,經常會去互相訪問,他們想讓你去幫助他們計算Q(1<=q<=1000)對奶牛之間的距離。

Input

*第一行:兩個被空格隔開的整數:N和Q

*第二行到第n行:第i+1行有兩個被空格隔開的整數:AI,BI,LI

*第n+1行到n+Q行:每一行有兩個空格隔開的整數:P1,P2,表示兩頭奶牛的編號。

Output

*第1行到第Q行:每行輸出一個數,表示那兩頭奶牛之間的距離。

Sample Input

4 2
2 1 2
4 3 2
1 4 3
1 2
3 2

Sample Output

2
7
思路: 對於這道題,我們可以發現,兩各節點之間的距離就等於這兩點的深度之和減去這兩點的lca的深度的兩倍 so,這道題我們就轉化成求lca的問題了
好廢話少說,我們來看看代碼吧! 代碼:
#include<vector>
#include<cstdlib>
#include<stdio.h>
#include<cstring>
#include<iostream>
#include<algorithm>
#define N 10001
using namespace std;
vector<pair<int,int> >vec[N];
int n,m,x,y,z,fa[N],deep[N],top[N],size[N],dis[N];
int lca(int x,int y) { for(;top[x]!=top[y];) { if(deep[top[x]]<deep[top[y]]) swap(x,y); x=fa[top[x]]; } if(deep[x]>deep[y]) swap(x,y); return x; } void dfs(int x) { size[x]=1; deep[x]=deep[fa[x]]+1; for(int i=0;i<vec[x].size();i++) { if(fa[x]!=vec[x][i].first) { fa[vec[x][i].first]=x; dis[vec[x][i].first]=dis[x]+vec[x][i].second; dfs(vec[x][i].first); size[x]+=size[vec[x][i].first]; } } } void dfs1(int x) { int t=0; if(!top[x]) top[x]=x; for(int i=0;i<vec[x].size();i++) if(vec[x][i].first!=fa[x]&&size[vec[x][i].first]>size[t]) t=vec[x][i].first; if(t) { top[t]=top[x]; dfs1(t); } for(int i=0;i<vec[x].size();i++) if(vec[x][i].first!=fa[x]&&vec[x][i].first!=t) dfs1(vec[x][i].first); } int main() { scanf("%d%d",&n,&m); for(int i=1;i<n;i++) { scanf("%d%d%d",&x,&y,&z); vec[x].push_back(make_pair(y,z)); vec[y].push_back(make_pair(x,z)); } dfs(1); dfs1(1); for(int i=1;i<=m;i++) { scanf("%d%d",&x,&y); printf("%d\n",dis[x]+dis[y]-2*dis[lca(x,y)]); } }

樹講解——牧場行走( lca )