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;
}