BestCoder Round #65 HDU 5593 ZYB's Tree （树形DP）

题目链接：点击打开链接

题意：有一棵n（n<=500000）个节点的树，ans[i]为树上到i点距离不超过k的点的个数，求出所有ans[i]的异或和

解法：树形DP，分步求解。具体如下：

1.dp[i][j]为以i为根的子树上到i的距离不超过j的点的个数，可先求距离恰好为j的点的个数，再求前缀和即可。

2.在求dp[i][j]时，可使在第一轮循环中按 j 从小到大顺序更新，第二轮循环中对于每个点 i ，dp[i][j]=sigma（dp[son[i]][j-1]），这样的好处是可以避免记忆化搜索。

3.在求出dp[i][j]后，求解到i距离不超过j的点的个数可从两方面考虑，一是在i的子树上的点，二是在i子树外的点，这部分点的个数为 sigma（dp[e][k-len]-dp[v][k-len-1]）,其中节点e、v均为i的祖先节点，且e为v的父节点，len为e到i的距离。

AC代码如下：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
#include <cmath>
#include <vector>
#include <set>
#include <map>
#include <stack>
#include <queue>
using namespace std;

int fa[500005],dp[500005][11],ans[500005];
vector<int>  son[500005];
int t,n,k,A,B;

void init(){
    for(int i=1;i<=n;i++)
        son[i].clear();
    fa[1]=0;
    for(int i=2;i<=n;i++){
        int tt=((long long)A*i+B)%(i-1)+1;
        fa[i]=tt;
        son[tt].push_back(i);
    }
}

void dfs(int x){
    int cur=x,pre=fa[x];
    int len=k;
    ans[x]=dp[x][len];
    while(pre!=0&&len>=1){
        ans[x]+=dp[pre][len-1];
        if(len>=2)
            ans[x]-=dp[cur][len-2];
        len--;
        int mem=pre;
        pre=fa[pre];
        cur=mem;
    }
}

int main (){
    scanf("%d",&t);
    while(t--){
        scanf("%d%d%d%d",&n,&k,&A,&B);
        init();
        for(int i=1;i<=n;i++)
            dp[i][0]=1;
        for(int j=1;j<=k;j++){
            for(int i=1;i<=n;i++){
                dp[i][j]=0;
                for(int tt=0;tt<son[i].size();tt++)
                    dp[i][j]+=dp[son[i][tt]][j-1];
            }
        }
        for(int i=1;i<=n;i++)
            for(int j=1;j<=k;j++)
                dp[i][j]+=dp[i][j-1];
        int tmp=0;
        for(int i=1;i<=n;i++){
            dfs(i);
            tmp=tmp^ans[i];
        }
        printf("%d\n",tmp);
    }
    return 0;
}