hdu 5593 ZYB's Tree

ZYB's Tree

 

 Accepts: 77

 

 Submissions: 513

 Time Limit: 3000/1500 MS (Java/Others)

 

 Memory Limit: 131072/131072 K (Java/Others)

问题描述

$ZYB$有一颗$N$个节点的树,现在他希望你对于每一个点,求出离每个点距离不超过$K$的点的个数.

两个点$(x,y)$在树上的距离定义为两个点树上最短路径经过的边数,

为了节约读入和输出的时间,我们采用如下方式进行读入输出:

读入:读入两个数$A,B$,令fa_ifa​i​​为节点$i$的父亲,fa_1=0fa​1​​=0;fa_i=(A*i+B)\%(i-1)+1fa​i​​=(A∗i+B)%(i−1)+1 $i\in [2,N]$ .

输出:输出时只需输出$N$个点的答案的$xor$和即可。

输入描述

第一行一个整数$T$表示数据组数。

接下来每组数据：

 一行四个正整数$N,K,A,B$.

 最终数据中只有两组$N\ge 100000$。

$1\le T\le 5$,$1\le N\le 500000$,$1\le K\le 10$,$1\le A,B\le 1000000$

输出描述

$T$行每行一个整数表示答案.

输入样例

1
3 1 1 1

输出样例

3

思路：
dp[i][j]表示以i为根的子树中，和i距离为j的点有多少个

/*======================================================
# Author: whai
# Last modified: 2015-12-07 21:13
# Filename: d.cpp
======================================================*/
#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
#include <cstring>
#include <string>
#include <cmath>
#include <set>
#include <map>
#include <queue>
#include <stack>

using namespace std;

#define LL __int64
#define PB push_back
#define P pair<int, int>
#define X first
#define Y second

const int N = 5 * 1e5 + 5;

vector<int> G[N];
int dp[N][15];

void dfs1(int u, int k) {
	dp[u][0] = 1;
	for(int i = 0; i < G[u].size(); ++i) {
		int v = G[u][i];
		dfs1(v, k);
		for(int j = 0; j < k; ++j) {
			dp[u][j + 1] += dp[v][j];
		}
	}
}

void dfs2(int u, int k) {
	for(int i = 0; i < G[u].size(); ++i) {
		int v = G[u][i];
		for(int j = k; j >= 2; --j) {
			dp[v][j] += dp[u][j - 1] - dp[v][j - 2];
		}
		dp[v][1] += dp[u][0];
		dfs2(v, k);
	}
}

void init(int n) {
	for(int i = 0; i <= n; ++i) {
		G[i].clear();
	}
	memset(dp, 0, sizeof(dp));
}

int main() {
	int T;
	cin>>T;
	while(T--) {
		int n, k, A, B;
		cin>>n>>k>>A>>B;
		init(n);
		for(int i = 2; i <= n; ++i) {
			int fa = ((LL)A * i + B) % (i - 1) + 1;
			G[fa].PB(i);
		}

		dfs1(1, k);
		dfs2(1, k);

		int ans = 0;
		for(int i = 1; i <= n; ++i) {
			int tmp = 0;
			for(int j = 0; j <= k; ++j) {
				tmp += dp[i][j];
			}
			ans ^= tmp;
		}
		cout<<ans<<endl;
	}
	return 0;
}