1. 程式人生 > >[bzoj4543][POI2014]Hotel加強版【樹形dp】【長鏈剖分】

[bzoj4543][POI2014]Hotel加強版【樹形dp】【長鏈剖分】

【題目連結】
  https://www.lydsy.com/JudgeOnline/problem.php?id=4543
【題解】
  列舉中點的方式行不通了,需要換一種思路。
  想辦法dp一下:
  記f[i][j]表示以i為根的子樹,到i距離為j的點的數目。
  g[i][j]表示以i為根的子樹,在其中有多少對點可以與在子樹i外,且到i的距離為j的點組成滿足題意的三元組的數目。
  dp時:每次加入i的一個兒k子後,先更新答案:
  

ans=ans+j=0nf[i][j]g[k][j+1]+j=0ng[i][j]f[k][j+1]
  再更新fg:
  g[i][j]=g[i][j]+f[i][j]f[k][j1]
  f[i][j]=f[i][j]+f[k][j1]
  g[i][j]=g[i][j]+f[k][j+1]
  *注意順序。
  然而這樣複雜度還是O(N2)的,需要優化轉移。
  考慮ison[i]i只有一個兒子,那麼f[i][j]=f[i][j1],g[i][j]=g[i][j+1]
  所以可以通過指標移動O(1)解決。
  所以我們想到了一種優化方式:每個節點通過指標移動繼承可以延伸最長的的兒子的答案,其他兒子暴力計算。
  這個演算法的複雜度是:i=1ndep[i]i=1ndep[i]1=O(N)
  每個兒子往上轉移的複雜度是dep的,由於一個節點的深度一定是長鏈所指向的兒子的深度+1,所以可以省下
dep1
次轉移。
  

# include <bits/stdc++.h>
# define    N       1000100
# define    ll      long long
using namespace std;
int read(){
    int tmp=0, fh=1; char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-') fh=-1; ch=getchar();}
    while (ch>='0'&&ch<='9'){tmp=tmp*10+ch-'0'; ch=getchar();}
    return tmp*fh;
}
struct Edge{
    int data,next;
}e[N*2];
int dep[N],per[N],head[N],place,n;
ll space[N*10];
ll *f[N],*g[N],ans,*now=space+N;
void build(int u, int v){
    e[++place].data=v; e[place].next=head[u]; head[u]=place;
}
void create(int id){
    f[id]=now; now=now+dep[id]*2+1; 
    g[id]=now; now=now+dep[id]*2+1;
}
void dep_cnt(int x, int fa){
    for (int ed=head[x]; ed!=0; ed=e[ed].next)
        if (e[ed].data!=fa){
            dep_cnt(e[ed].data,x);
            if (dep[e[ed].data]>dep[per[x]])
                per[x]=e[ed].data;
        }
    dep[x]=dep[per[x]]+1;
}
void solve(int x, int fa){
    f[x][0]=1;
    if (per[x]!=0){
        f[per[x]]=f[x]+1;
        g[per[x]]=g[x]-1;
        solve(per[x],x);
        ans=ans+g[per[x]][1];
    }
    for (int ed=head[x]; ed!=0; ed=e[ed].next)
        if (e[ed].data!=fa&&e[ed].data!=per[x]){
            create(e[ed].data);
            solve(e[ed].data,x);
            int y=e[ed].data;
            for(int j=dep[y];j>=0;j--){
                if(j) ans+=f[x][j-1]*g[y][j];
                ans+=g[x][j+1]*f[y][j];
                g[x][j+1]+=f[x][j+1]*f[y][j];
            }
            for(int j=0;j<=dep[y];j++){
                if(j)   g[x][j-1]+=g[y][j];
                f[x][j+1]+=f[y][j];
            }
        }
}
int main(){
    n=read();
    for (int i=1; i<n; i++){
        int u=read(), v=read();
        build(u,v); build(v,u);
    }
    dep_cnt(1,0);
    create(1);
    solve(1,0);