10.30 T2 樹上的戰爭(倍增lca)

               b

                                         題目來源:zhx(侵刪)

【問題描述】

                                     你是能看到第二題的 friends 呢。
                                                                  ——laekov

Hja 和 Yjq 爲了搶男主角打了起來, 現在他們正在一棵樹上決鬥。 Hja 在 A點, Yjq 在 B 點, Hja 先發制人開始移動。 每次他們可以沿着一條邊移動, 但一旦一條邊被對方走過了自己就不能再走這條邊了。 每條邊上都有權值, 他們都希望自己的權值儘量多。 現在給你這棵樹以及他們倆開始的位置, 問 Hja 能夠獲得的最大權值。

【輸入格式】

第一行兩個整數N, M, 代表樹的點數和詢問的個數。
接下來N− 1行每行三個整數a, b, c, 代表從a到b有一條權值爲c的邊。
接下來M行, 每行兩個整數A, B代表一次詢問。

【輸出格式】

對於每次詢問, 輸出一個整數代表答案。

【樣例輸入 1】

2 1
1 2 3
1 2

【樣例輸出 1】

3

【樣例輸入 2】

3 2
1 2 3
1 3 1
2 3
1 3

【樣例輸出 2】

3
4

【數據範圍與規定】

對於30%的數據, 1 ≤N,M ≤ 1000。
對於另外30%的數據, M= 1。
對於100%的數據, 1 ≤ N,M≤ 105, 0 ≤c≤ 103, 1 ≤a,b,A,B≤N。

思路:

因爲他們都希望自己的權值儘量多,所以必定都會按照最優的方案走,也就是走他們之間的路徑,儘可能快的堵住對方。

所以我們要求兩個人之間的路徑,再在中間找他們相遇的地方,這裏注意hja是先手,他所經過的道路數爲(L+1)/2,其中L表示路徑長度。

找出路徑和相遇點後就可以求他們現在各自所擁有的權值。令sum[i]表示以i爲根節點的子樹的權值和。

路徑可以用倍增lca來求,sum數組可以在dfs時順便處理出來。

但是還要注意一個問題:

兩個人在一個節點相遇後,仍舊會朝着一個方向走,直到所有能走的邊都被走過了爲止。

所以我們要把相遇點所連接的各子樹(除去兩個人走來的子樹)的答案記下來,從大到小排序,按照最優的原則和兩個人的先後順序進行分配。

解題步驟

  1. dfs處理fa和sum

  2. lca求路徑

  3. 找相遇點

  4. 對相遇點的子樹答案進行分配

特判:X等於Y

代碼

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

int N,M,cnt,A,B,C,X,Y,ans;
int first[100010],next[200010],deep[100010];
int fa[100010][30],sum[100010],f[100010],v[100010];

struct maple{
    int f,t,d;
}Rode[200010];

void Build(int f,int t,int d)
{
   Rode[++cnt]=(maple){f,t,d};  
   next[cnt]=first[f];
   first[f]=cnt;
}

int Dfs(int f,int t)
{
    deep[t]=deep[f]+1;
    fa[t][0]=f;
    for(int i=first[t];i;i=next[i])
       if(Rode[i].t!=f)
       {
           sum[t]=sum[t]+Dfs(t,Rode[i].t)+Rode[i].d;
       }
    return sum[t];   
}

void Done()
{
    for(int i=1;i<=log2(N);++i)
       for(int j=1;j<=N;++j)
          fa[j][i]=fa[fa[j][i-1]][i-1];
}
int lca(int x,int y)
{
    if(deep[x]<deep[y]) swap(x,y);
    for(int i=log2(N);i>=0;--i)
    {
        if(deep[fa[x][i]]>=deep[y])
            x=fa[x][i];
    }   
    if(x==y) return x;
    for(int i=log2(N);i>=0;--i)
       if(fa[x][i]!=fa[y][i])
          x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}

int ask(int x,int d)
{
    for(int i=log2(N);i>=0;--i)
        if(deep[fa[x][i]]>=d)
           x=fa[x][i];
    return x;
}
bool cmp(int a,int b)
{
    return a>b;
}
void ask_ans(int kx,int ky,int Lca,int L)
{
    if(kx!=ky)
    {   
        if(deep[kx]>=deep[ky]) ans=sum[kx];  
        else ans=sum[1]-sum[Lca];
    }
    else ans=0;
    int n=0;
    for(int i=first[Lca];i;i=next[i])
    {
         if(Rode[i].t!=kx&&Rode[i].t!=ky)
         {
              ++n;
              if(Rode[i].t!=fa[Lca][0])
                   v[n]=sum[Rode[i].t]+Rode[i].d;
              else v[n]=sum[1]-sum[Lca];
         }
         if(Rode[i].t==kx&&deep[kx]>=deep[ky]) ans+=Rode[i].d; 
    }
    int k=1;
    if(L%2!=0) ++k;
    sort(v+1,v+n+1,cmp); 
    for(int i=k;i<=n;i+=2)  ans+=v[i];
    return ;
}
int main()
{
    freopen("b.in","r",stdin);
    freopen("b.out","w",stdout);
    scanf("%d%d",&N,&M);
    for(int i=1;i<N;++i)
    {
        scanf("%d%d%d",&A,&B,&C);
        Build(A,B,C);
        Build(B,A,C);
    }
    sum[1]=Dfs(1,1);
    Done();
    while(M--)
    {
        scanf("%d%d",&X,&Y);
        int Lca=lca(X,Y),kx,ky,k;
        int L=deep[X]+deep[Y]-2*deep[Lca];
        int ll=L;
        if(L%2==0) L/=2;
        else L=L/2+1;
        if(deep[X]-deep[Lca]==L) 
        {
            kx=ask(X,deep[Lca]+1);
            ky=ask(Y,deep[Lca]+1);
            ask_ans(kx,ky,Lca,ll);
            //cout<<"*";
        }
        else
        {
            if(X==fa[Y][0]) ask_ans(X,Y,Y,1);//printf("****");
            else if(deep[X]-deep[Lca]>L) 
            {
                k=ask(X,deep[X]-L+1);
                ask_ans(k,fa[k][1],fa[k][0],ll);
                //cout<<"**";
            } 
            else if(deep[X]-deep[Lca]<L)
            {
                L=L-deep[X]+deep[Lca];
                int l=deep[Lca]+L+1;
                k=ask(Y,l);
                ask_ans(fa[k][1],k,fa[k][0],ll);
                //cout<<"***";
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}

打的比較醜QAQ

另 附上陪我調了一下午的數據,X,Y請自己出

15 100
1 2 1
2 4 1
2 5 0
5 9 1
5 10 3
2 6 2
6 11 4
11 12 6
11 13 5
1 3 2
3 7 2
3 8 4
8 14 5
14 15 7
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章