标签：rt int ans dfs CF486D Valid eid Sets dp

https://www.luogu.com.cn/problem/CF486D

考虑树形DP
强制每个联通块的根为最大的那个数（如果有多个那么就是编号最大的那个）
然后枚举根就可以转移了
d p [ u ] = d p [ u ] + d p [ u ] ∗ d p [ v ] dp[u]=dp[u]+dp[u]*dp[v] dp[u]=dp[u]+dp[u]∗dp[v]

code:

#include<bits/stdc++.h>
#define N 2050
#define ll long long
#define mod 1000000007
using namespace std;
struct edge {
    int v, nxt;
} e[N << 1];
int p[N], eid;
void init() {
    memset(p, -1, sizeof p);
    eid = 0;
}
void insert(int u, int v) {
    e[eid].v = v;
    e[eid].nxt = p[u];
    p[u] = eid ++;
}
ll f[N];
int n, d, a[N];
void dfs(int u, int fa, int rt) {
    f[u] = 1;
    for(int i = p[u]; i + 1; i = e[i].nxt) {
        int v = e[i].v;
        if(v == fa) continue;
        if((a[rt] > a[v] || (a[rt] == a[v] && rt > v)) && a[rt] - a[v] <= d) {
            dfs(v, u, rt);
            f[u] = (f[u] + f[u] * f[v]) % mod;
        }
    }
}
int main() {
    init();
    scanf("%d%d", &d, &n);
    for(int i = 1; i <= n; i ++) scanf("%d", &a[i]);
    for(int i = 1; i < n; i ++) {
        int u, v;
        scanf("%d%d", &u, &v);
        insert(u, v), insert(v, u);
    }
    ll ans = 0;
    for(int i = 1; i <= n; i ++) {
        dfs(i, i, i);
        ans = (ans + f[i]) % mod;
    }
    printf("%lld", ans);
    return 0;
}

标签：rt,int,ans,dfs,CF486D,Valid,eid,Sets,dp
来源： https://blog.csdn.net/qq_38944163/article/details/120693305

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。