[CF486D]有效集合-树形dp
Posted SkyLYnf
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了[CF486D]有效集合-树形dp相关的知识,希望对你有一定的参考价值。
Problem 有效集合
题目大意
给出一棵树,求出这棵树的不同联通子节点集合的数量,这些集合必须满足最大权值点减最小权值点小于等于d。
Solution
再一次树d乱搞。
因为数据范围贼小,所以我们对于每一个点为根的情况进行一次dfs.
对于以节点root为根的情况,我们认为root是最小值。
对于root的某一棵子树,其需要算的情况一定包含root,否则前面已经算过了。
于是我们就对这个root进行dfs。dfs过程中,我们始终保证root为最小值,不遍历大于它的点。
因为有了这个性质,我们只要不走加起来大于d的点,接下来所有的点都不属于被计算范围内,可以不走。
在这个以root为根的树中,现在我们遍历到u,选中它的一颗子树v。
对于这一串东西上,我们每次可以选择以前已遍历过的子树中选择某些点,再从v该棵子树下选择某些点组成新情况。
我们增加的情况是已经遍历过的子树答案*(该子树被遍历到的子树总答案+1)
加一是因为空集也算是一种情况。
也就是f[u]*=f[v]+1;
对于每个叶子结点,我们只需要将f设为1即可。
dfs完以后,我们将以每个点为根(最小值)的情况进行累加。
于是最终的ans便是答案。
需要注意的是,这一道题MOD1000000007
所以f数组、ans变量需要long long类型
而我一直爆炸,因为没注意到,dfs函数也要为long long类型。
所以请注意这个问题。
AC Code
#include <iostream> #include <cstdio> #include <cstring> #define MOD 1000000007 using namespace std; struct node{ int to,next; }e[4010]; int d,n,u,v,h[2010],a[2010],tot=0; long long f[2010],ans=0; void add(int u,int v){ e[++tot].to=u;e[tot].next=h[v];h[v]=tot; e[++tot].to=v;e[tot].next=h[u];h[u]=tot; } long long dfs(int x,int last,int root){ long long ans=1; f[x]=1; for(int i=h[x];~i;i=e[i].next){ if(last!=e[i].to&&(a[e[i].to]>a[root]||(a[e[i].to]==a[root]&&root>e[i].to)) &&a[e[i].to]-a[root]<=d){ ans=(ans*(dfs(e[i].to,x,root)+1)%MOD)%MOD; } } return ans; } int main(){ // freopen("cf486d.in","r",stdin); memset(h,-1,sizeof(h)); scanf("%d%d",&d,&n); for(int i=1;i<=n;i++)scanf("%d",&a[i]); for(int i=1;i<n;i++){ scanf("%d%d",&u,&v); add(u,v); } for(int i=1;i<=n;i++)ans+=dfs(i,0,i); printf("%lld",ans%MOD); }
以上是关于[CF486D]有效集合-树形dp的主要内容,如果未能解决你的问题,请参考以下文章