ZYB‘s Tree（树上每个点距离为k的点)

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

---

思路：经典，以前碰过一次。

但是其中一个思路也对。自己调了一小时还是不知道哪里的问题。样例没过。干脆换一种，思路是一样的。

首先距离为k来自u的子树，dp[u][j]+=dp[v][j-1]

但是答案还有来自父亲以上的一块和u的兄弟，也就是父亲以下的一块。

对于来自父亲以上的，换根后第二次dfs的时候直接转移过来。对于父亲以下的,算为dp[u][j-1]-dp[v][j-2]，去除本身子树的那部分距离。

注意算A的时候开longlong

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=5e5+1000;
typedef int LL;
inline LL read(){LL x=0,f=1;char ch=getchar();	while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;}
LL dp[maxn][13],f[maxn][13],sum[maxn][13];
vector<LL>g[maxn];
LL n,k,A,B;
void dfs1(LL u,LL fa){
     dp[u][0]=1;///初始化到自身距离为0的点为自身
     for(LL i=0;i<g[u].size();i++){
         LL v=g[u][i];
         if(v==fa) continue;
         dfs1(v,u);
         for(LL j=0;j<=k;j++){
             if(j>0) dp[u][j]+=dp[v][j-1];
             ///sum[u][j]+=dp[v][j];
         }
     }
}
void dfs2(LL u,LL fa){
//     f[u][0]=1;
//     if(fa){
//        for(LL j=1;j<=k;j++){
//            f[u][j]+=f[fa][j-1];
//        }
//     }
//     for(LL j=2;j<=k;j++){
//         if(j>=2&&fa){
//             f[u][j]+=sum[fa][j-2]-dp[u][j-2];
//         }
//     }
     for(LL i=0;i<g[u].size();i++){
         LL v=g[u][i];
         if(v==fa) continue;
         for(LL j=2;j<=k;j++){
             f[v][j]=dp[u][j-1]-dp[v][j-2]+f[u][j-1];
         }
         f[v][1]=dp[u][0];
         dfs2(v,u);
     }
}
int main(void){
   cin.tie(0);std::ios::sync_with_stdio(false);
   LL t;cin>>t;
   while(t--){
      memset(dp,0,sizeof(dp));
      memset(g,0,sizeof(g));
      memset(sum,0,sizeof(sum));
      for(LL i=0;i<maxn;i++) g[i].clear();
      cin>>n>>k>>A>>B;
      for(LL i=2;i<=n;i++){
          long long id=((long long int)A*i+B)%(i-1)+1;
          g[id].push_back(i);
          g[i].push_back(id);
      }
      dfs1(1,0);

      dfs2(1,0);
      LL ans=0;
      for(LL i=1;i<=n;i++){
          LL tot=0;
          for(LL j=0;j<=k;j++){
            tot+=(dp[i][j]+f[i][j]);
          }
          ans^=tot;
      }
      cout<<ans<<"\n";
   }
   return 0;
}