codeforces 161D 树的点分治

D. Distance in Tree
time limit per test
3 seconds
memory limit per test
512 megabytes
input
standard input
output
standard output

A tree is a connected graph that doesn't contain any cycles.

The distance between two vertices of a tree is the length (in edges) of the shortest path between these vertices.

You are given a tree with n vertices and a positive number k. Find the number of distinct pairs of the vertices which have a distance of exactly k between them. Note that pairs (v, u) and (u, v) are considered to be the same pair.

Input

The first line contains two integers n and k (1 ≤ n ≤ 50000, 1 ≤ k ≤ 500) — the number of vertices and the required distance between the vertices.

Next n - 1 lines describe the edges as "ai bi" (without the quotes) (1 ≤ ai, bi ≤ n, ai ≠ bi), where ai and bi are the vertices connected by the i-th edge. All given edges are different.

Output

Print a single integer — the number of distinct pairs of the tree's vertices which have a distance of exactly k between them.

Please do not use the %lld specifier to read or write 64-bit integers in С++. It is preferred to use the cin, cout streams or the %I64d specifier.

Examples
Input
5 2
1 2
2 3
3 4
2 5
Output
4
Input
5 3
1 2
2 3
3 4
4 5
Output

2


#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<vector>
using namespace std;
#define ll long long
const int maxm = 100005;
vector<int>v[maxm];
int n, root, m, sn, tot;
ll ans;
int son[maxm], s[maxm], vis[maxm], flag[505], cnt[505];
void getroot(int k, int pre)
{
	son[k] = 1, s[k] = 0;
	for (int i = 0;i < v[k].size();i++)
	{
		if (v[k][i] == pre || vis[v[k][i]]) continue;
		getroot(v[k][i], k);
		son[k] += son[v[k][i]], s[k] = max(s[k], son[v[k][i]]);
	}
	s[k] = max(s[k], sn - s[k]);
	if (s[root] > s[k]) root = k;
}
void getdeep(int k, int pre, int deep)
{
	if (deep <= m)
	{
		ans += flag[m - deep];
		cnt[deep] ++;
	}
	//printf("%d %d %d\n", root, k, deep);
	for (int i = 0;i < v[k].size();i++)
	{
		if (v[k][i] == pre || vis[v[k][i]]) continue;
		getdeep(v[k][i], k, deep + 1);
	}
}
void query(int k)
{
	memset(flag, 0, sizeof(flag));
	flag[0] = 1;
	for (int i = 0;i < v[k].size();i++)
	{
		if (vis[v[k][i]]) continue;
		memset(cnt, 0, sizeof(cnt));
		getdeep(v[k][i], 0, 1);
		for (int j = 1;j <= m;j++)
			flag[j] += cnt[j];
	}
}
void dfs(int k)
{
	vis[k] = 1, query(k);
	for (int i = 0;i < v[k].size();i++)
	{
		if (vis[v[k][i]]) continue;
		sn = son[v[k][i]], root = 0;
		getroot(v[k][i], 0), dfs(root);
	}
}
int main()
{
	int i, j, sum, x, y;
	scanf("%d%d", &n, &m);
	for (i = 1;i < n;i++)
	{
		scanf("%d%d", &x, &y);
		v[x].push_back(y), v[y].push_back(x);
	}
	s[0] = 100000000, root = 0, sn = n;
	getroot(1, 0), dfs(root);
	printf("%lld\n", ans);
	return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值