BC#65T4 ZYB's Tree

题目：http://acm.hdu.edu.cn/showproblem.php?pid=5593

点分治被卡了TAT。。。 正解是dp，可以按层数考虑dp，先预处理跑一边dfs得到子树各层数点数大小，再dfs一遍找出不在子树上的与当前点距离<=k的点。

设u为当前点，v为其一个儿子，有g[v][1]=1

g[v][i]+=g[u][i-1](i>=2)

g[v][i]+=f[u][i-1]-f[v][i-2](i>=2)

完全不难嘛比赛的时候人都傻了TAT，还是太弱了

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <cstdlib>
#define rep(i,l,r) for (int i=l;i<=r;i++)
#define down(i,l,r) for (int i=l;i>=r;i--)
#define clr(x,y) memset(x,y,sizeof(x))
#define maxn 500500
#define ll long long
using namespace std;
int f[maxn][],g[maxn][],head[maxn],n,k,ans,tot;
struct data{int obj,pre;
}e[maxn];
int read(){
    int x=,f=; char ch=getchar();
    while (!isdigit(ch)){ if (ch=='-') f=-; ch=getchar();}
    while (isdigit(ch)) {x=x*+ch-''; ch=getchar();}
    return x*f;
}
void insert(ll x,ll y){
    e[++tot].obj=y; e[tot].pre=head[x]; head[x]=tot;
}
void dfs(int u){
    f[u][]=;
    for (int j=head[u];j;j=e[j].pre){
        int v=e[j].obj;
        dfs(v);
        rep(i,,k) f[u][i]+=f[v][i-];
    }
}
void dfs2(int u){
    int sum=;
    rep(i,,k) sum+=f[u][i];
    for (int j=head[u];j;j=e[j].pre){
        int v=e[j].obj;
        g[v][]=;
        rep(i,,k) g[v][i]+=g[u][i-];
        rep(i,,k) g[v][i]+=f[u][i-]-f[v][i-];
        dfs2(v);
    }
    rep(i,,k) sum+=g[u][i];
    ans^=sum;
}
int main(){
    int t;
    scanf("%d",&t);
    while (t--){
        clr(g,); clr(f,); clr(head,); tot=;
        ll a,b;
        n=read(); k=read(); scanf("%lld%lld",&a,&b);
        rep(i,,n){
            ll now=(ll)i;
            ll f=(a*now+b)%(now-)+;
            insert(f,i);
        }
        ans=;
        dfs(); dfs2();
        printf("%d\n",ans);
    }
    return ;
}