codevs 1036 商務旅行 題解報告

繼續我的刷題之路；；；

題目描述 Description

某首都城市的商人要經常到各城鎮去做生意，他們按自己的路線去做，目的是為了更好的節約時間。

假設有N個城鎮，首都編号為1，商人從首都出發，其他各城鎮之間都有道路連接配接，任意兩個城鎮之間如果有直連道路，在他們之間行駛需要花費機關時間。該國公路網絡發達，從首都出發能到達任意一個城鎮，并且公路網絡不會存在環。

你的任務是幫助該商人計算一下他的最短旅行時間。

輸入描述 Input Description

輸入檔案中的第一行有一個整數N，1<=n<=30 000，為城鎮的數目。下面N-1行，每行由兩個整數a 和b (1<=a, b<=n; a<>b)組成，表示城鎮a和城鎮b有公路連接配接。在第N+1行為一個整數M，下面的M行，每行有該商人需要順次經過的各城鎮編号。

輸出描述 Output Description

在輸出檔案中輸出該商人旅行的最短時間。

樣例輸入 Sample Input

5

1 2

1 5

3 5

4 5

4

1

3

2

5

樣例輸出 Sample Output

7

資料範圍及提示 Data Size & Hint

讀題可知，在一棵樹上以一定的順序走，求最短路徑。

可以肯定，在樹上兩點之間的最短路徑就是兩點各自走到他們的LCA 的距離和；

是以，對于m個點；

隻需要求出第i，i-1 個點的LCA

過程中維護距離，

最後加和即可。

代碼如下：：：、

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<stack>
#include<cstdlib>
#include<string>
#include<bitset>
#include<iomanip>
#include<deque>
#define INF 1000000000
#define fi first
#define se second
#define N 100005
#define P 1000000007
#define debug(x) cerr<<#x<<"="<<x<<endl
#define MP(x,y) make_pair(x,y)
using namespace std;
int n,m;
int h[N],v[N],next[N],p=0,d[N],f[N],w[N],l[N],j[N][15],ww[N][15],deep[N];
bool vv[N];
void add(int a,int b)
{
    p++;
    v[p]=b;
    next[p]=h[a];
    h[a]=p;
}
inline int get_num()
{
int num = 0;
char c;
bool flag = false;
while ((c = getchar()) == ' ' || c == '\n' || c == '\r');
if (c == '-') flag = true;
else num = c - '0';
while (isdigit(c = getchar()))
num = num * 10 + c - '0';
return (flag ? -1 : 1) * num;
}  //忽略上面這一大堆沒用的東西。。。。
void dfs(int x,int s)
{
    deep[x]=s;
    for(int i=1;(1<<i)<s;i++)
    {
        j[x][i]=j[j[x][i-1]][i-1];
    }
    int pp=h[x];
    while(pp)
    {
        if(vv[v[pp]])
        {
            pp=next[pp];
            continue;
        }
        vv[v[pp]]=1;
        j[v[pp]][0]=x;
        dfs(v[pp],s+1);
        pp=next[pp];
    }
}
int LCA(int x,int y)
{
    int ans=0; 
    if(deep[y]>deep[x])
    {
        int kk=y;
        y=x;
        x=kk;
    }
    for(int i=15;i>=0;i--)
    {
        if((deep[x]-(1<<i))>=deep[y])
        {
            ans+=(1<<i);
            x=j[x][i];
        }
    }
    if(x==y)return ans;
    for(int i=15;i>=0;i--)
    {
        if(j[x][i]!=j[y][i]&&deep[x]-(1<<i)>=1)
        {
            ans+=(1<<i);
            ans+=(1<<i);
            x=j[x][i];
            y=j[y][i];
        }
    }
    ans+=2;
    return ans;
}
int main()
{
    cin>>n;
    for(int i=1;i<n;i++)
    {
        int q,w;
        cin>>q>>w;
        add(q,w);
        add(w,q);
    } 

    vv[1]=1;
    dfs(1,1);
    cin>>m;
    int l=1;
    int sum=0;
    for(int i=1;i<=m;i++)
    {
        int q;
        cin>>q;
        sum+=LCA(l,q);
        l=q;
    }
    cout<<sum;
}
           

噫，，水題。。