Codeforce 486D(树形dp)

最新推荐文章于 2024-01-23 17:08:39 发布

原创最新推荐文章于 2024-01-23 17:08:39 发布 · 514 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#acm #dp #codeforce

动态规划同时被 2 个专栏收录

226 篇文章

订阅专栏

---------树形dp

33 篇文章

订阅专栏

本文介绍了一种基于树形结构的动态规划算法实现方法，旨在解决特定类型的问题：即在一个由n个节点组成的树中，根据每个节点的权值找出满足条件的节点集合数量。这些集合内的节点必须两两可达，且最大权值与最小权值之差不超过给定阈值d。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

链接：点击打开链接

题意：给出一个n个点的树，给出每个点的权值，问存在多少个集合，使得集合中两两可以互相到达，并且最大值减去最小值小于等于d

代码：

#include <math.h>
#include <vector>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
const long long MOD=1000000007;
long long d,dp[2005],val[2005];
vector<long long> G[2005];
void dfs(long long s,long long fa,long long rt){
    long long i,tmp;
    dp[s]=1;
    for(i=0;i<G[s].size();i++){
        tmp=G[s][i];
        if(tmp==fa)
        continue;
        dfs(tmp,s,rt);                          //保证的最大，并且差值小于d
        if((val[rt]>val[tmp]&&val[rt]-val[tmp]<=d)||(val[rt]==val[tmp]&&rt<tmp))
        dp[s]=(dp[s]+dp[s]*dp[tmp])%MOD;        //但是当值相等时，必定会出现重复的情况
    }                                           //所以值相等时，只能从编号小的点到大的点
}
int main(){
    long long n,i,j,u,v,ans;
    while(scanf("%I64d%I64d",&d,&n)!=EOF){
        for(i=1;i<=n;i++)
        G[i].clear();
        for(i=1;i<=n;i++)
        scanf("%I64d",&val[i]);
        for(i=1;i<n;i++){
            scanf("%I64d%I64d",&u,&v);
            G[u].push_back(v);
            G[v].push_back(u);
        }
        ans=0;
        for(i=1;i<=n;i++){                      //将每个点看做值最大的根节点进行dfs
            memset(dp,0,sizeof(dp));
            dfs(i,-1,i);
            ans=(ans+dp[i])%MOD;
        }
        printf("%I64d\n",ans);
    }
    return 0;
}