CodeChef：Company and Club Hierarchies（树形dp & 技巧）

题意：一棵树，每个节点有个权值k[i]和颜色c[i]，定义以节点u为根节点的序列为“好的序列”条件是：能够找到一条有序链，该链以u节点开始，链的长度为k[i]+1，链的成员颜色均为c[i]，对于链上第j个元素，j-1和j+1个元素必须是其祖先和子孙，问对每个节点求出它的好的序列数。

思路：官方题解 树形dp，dp[i][j]为i颜色j长度的链数，ans[u]为u节点的好序列数，那么ans[u] = dp[c[u]][k[i]-1]，这里dp没有保存当前结点，因为搜索到节点v时，先记录s = dp[c[v]][k[v]-1]，回溯时减回这个s就能保证结果是从子树得到了。

//reference: Pawel# include <bits/stdc++.h>using namespace std;const int maxn = 5e5+30;const int mod = 1e9+7;vector<int>dp[maxn], v[maxn];int club[maxn], cnt[maxn], level[maxn];int ans[maxn];void dfs(int u){    int s = (level[u]>0&&level[u]<cnt[club[u]])?dp[club[u]][level[u]-1]:0;    for(auto to : v[u])        dfs(to);    if(level[u] >= cnt[club[u]]) return;    if(level[u] == 0)        ans[u] = 1;    else        ans[u] = (dp[club[u]][level[u]-1] - s + mod)%mod;    dp[club[u]][level[u]] +=  ans[u];    dp[club[u]][level[u]] %= mod;}int main(){    int t, x, n;    scanf("%d",&t);    while(t--)    {        scanf("%d%d",&n,&x);        for(int i=0; i<n; ++i)        {            v[i].clear();            dp[i].clear();            ans[i] = cnt[i] = 0;        }        for(int p, i=1; i<n; ++i)        {            scanf("%d",&p);            v[p].push_back(i);        }        for(int i=0; i<n; ++i)        {            scanf("%d",&club[i]);            ++cnt[club[i]];        }        for(int i=0; i<n; ++i)            scanf("%d",&level[i]);        for(int i=0; i<n; ++i)            dp[i] = vector<int>(cnt[i]);        dfs(0);        for(int i=0; i<n; ++i)            printf("%d\n",ans[i]);    }    return 0;};