Codeforces 954H Path Counting 动态规划

题意

给出一棵深度为n的树，其中深度为i的节点有a[i]个儿子。问对于每个k，有多少条简单路径满足其长度恰好为k。
n<=5000

分析

一开始的想法是枚举lca，然后对其每棵子树算贡献。
想一下发现会很难做。
那么我们考虑枚举路径的端点。
设d[i,j]表示从某个深度为i的节点开始，只往下走且长度为j的路径条数。那么d[i,j]显然等于i的子树中深度为i+j的点数。
设u[i,j]表示从某个深度为i的节点开始，第一步必须往上走，且路径长度为j的方案。
有两种转移，一种是走到父亲后继续往上，贡献就等于u[i-1,j-1]。另一种转移是走到父亲后就开始往下走，贡献就等于u[i,j-2]*(a[i-1]-1)。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long LL;

const int N=5001;
const