题意
思路
现在感觉树形DP全是套路题,但是有的时候就是想不出来。
对于某一个长度为p的路径,不能与他构成合法的四元组的长度为q的路径有哪些情况呢?首先,lca在他的路径上的;其次,经过他的lca并且lca比他浅的。
于是树形DP, O ( n 2 ) O(n^2) O(n2)。据说用什么点分治可以 O ( n l o g n ) O(nlogn) 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;
}