hdu5593ZYB's Tree 树形dp

#include<cstdio>
#include<cstring>
#include<iostream>
#include<vector>
using namespace std ;
const int maxn = 5e5+10;
typedef long long ll ; 
int dp[maxn][12];
int ans[maxn][12];
int n , k ;
vector<int> vec[maxn] ;
void dfs(int u){
    for(int i = 0;i < vec[u].size();i++){
        int v = vec[u][i] ;
        dfs(v) ;
        for(int j = 1;j <= k;j++){
            dp[u][j] += dp[v][j-1] ;
        }
    }
    dp[u][0] += 1 ;
}
void dfss(int u){
    for(int i = 0;i < vec[u].size();i++){
        int v = vec[u][i] ;
        ans[v][0] = dp[v][0] ;
        for(int j = 1;j <= k;j++){
            ans[v][j] = ans[u][j-1] + dp[v][j] ;
            if(j >= 2) ; 
            ans[v][j] -= dp[v][j-2] ; 
        }
        dfss(v) ;
    }
}
int main()
{
    int t ;
    int a , b  ;
    scanf("%d" , &t) ;
    while(t--){
        scanf("%d%d%d%d" , &n , &k , &a , &b) ;
        for(int i = 1;i <= n;i++){
           vec[i].clear() ;
        }
        for(int i = 2;i <= n;i++){
            int u = ((ll)a*i + b)%(i-1) + 1 ;
            vec[u].push_back(i) ;
        }
        memset(dp , 0 , sizeof(dp)) ;
        memset(ans , 0 , sizeof(ans)) ;
        dfs(1) ;
        memcpy(ans[1] , dp[1] , sizeof(dp[1])) ;
        dfss(1) ;
        int sum = 0 ;
        for(int i = 1;i <= n;i++){
            int tmp = 0 ;
            for(int j = 0;j <= k;j++){
               tmp += ans[i][j] ;
            }
            sum ^= tmp;
        }
        printf("%d\n" , sum) ;
    }
    return 0 ;
}