tree（树形DP）

题意

思路

现在感觉树形DP全是套路题，但是有的时候就是想不出来。

对于某一个长度为p的路径，不能与他构成合法的四元组的长度为q的路径有哪些情况呢？首先，lca在他的路径上的；其次，经过他的lca并且lca比他浅的。

于是树形DP，$O(n^2)$。据说用什么点分治可以$O(nlogn)$。

其实代码就是非常普通的4个DFS加上一个LCA，非常普通但是结合在一起题目就非常有趣了。DFS太强了

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#define ll long long 
using namespace std;
const int N = 3010;
const int E = 13;
int n, p, q, e, point[N];
struct EDGE{
	int nxt, v;
}edge[N<<1];
int dpt[N], f[N][E];
ll g[N], h[N], tot, ans;

void add_edge(int u, int v)
{
    edge[++e] = (EDGE){point[u], v};
    point[u] = e;
}

void Dfs1(int u, int fa) // 处理倍增数组
{
	dpt[u] = dpt[fa]+1;
	f[u][0] = fa;
	for (int i = 1; i < E; i++)
		f[u][i] = f[f[u][i-1]][i-1];
	for (int i = point[u]; i != -1; i = edge[i].nxt){
		int v = edge[i].v;
		if (v == fa) continue;
		Dfs1(v, u);
	}
}

int Lca(int x, int y)
{
	if (dpt[x] < dpt[y])
		swap(x, y);
	for (int i = E-1; i >= 0; i--)
		if (dpt[f[x][i]] >= dpt[y])
			x = f[x][i];
	if (x == y) return x;
	for (int i = E-1; i >= 0; i--)
		if (f[x][i] != f[y][i])
			x = f[x][i], y = f[y][i];
	return f[x][0];
}

void Dfs2(int u, int rt, int fa, int rem) // 找长度为p的边并记录DP数组
{
	if (rem == 0){
		int lca = Lca(u, rt);
		tot++;
		g[lca]++;
		h[u]++; h[rt]++;
		h[lca] -= 2;
		return;
	}
	for (int i = point[u]; i != -1; i = edge[i].nxt){
		int v = edge[i].v;
		if (v == fa) continue;
		Dfs2(v, rt, u, rem-1);
	}
}

void Dfs3(int u, int fa) // 差分
{
	for (int i = point[u]; i != -1; i = edge[i].nxt){
		int v = edge[i].v;
		if (v == fa) continue;
		g[v] += g[u];
		Dfs3(v, u);
		h[u] += h[v];
	}
}

void Dfs4(int u, int rt, int fa, int rem) // 找q并且统计答案
{
	if (rem == 0){
		int lca = Lca(u, rt);
		ans += tot-(g[u]+g[rt]-g[lca]-g[f[lca][0]])-h[lca];
		return;
	}
	for (int i = point[u]; i != -1; i = edge[i].nxt){
		int v = edge[i].v;
		if (v == fa) continue;
		Dfs4(v, rt, u, rem-1);
	}
}

int main()
{
	scanf("%d%d%d", &n, &p, &q);
	memset(point, -1, sizeof(point)); e = -1;
	for (int i = 1; i < n; i++){
		int x, y;
		scanf("%d%d", &x, &y);
		add_edge(x, y);
		add_edge(y, x);
	}
	dpt[0] = -1;
	Dfs1(1, 0);
	memset(g, 0, sizeof(g));
	memset(h, 0, sizeof(h));
	for (int i = 1; i <= n; i++)
		Dfs2(i, i, 0, p);
	Dfs3(1, 0);
	ans = 0;
	for (int i = 1; i <= n; i++)
		Dfs4(i, i, 0, q);
	printf("%lld\n", ans);
	return 0;
}