【codeforces 486D】Valid Sets

題目：Valid Sets

題意：

       給出一棵樹，求這棵樹的滿足最大點與最小點之差小於d的連通子圖的個數。

解析：

       計數類DP+樹形DP。

       由於n只有2000，所以可以枚舉每個點作爲根且是最大權值進行DFS，遍歷子節點根據乘法原理每個子節點分可選可不選。

       但是如果節點權值相同會有重複情況，這時候規定這時候子節點編號必須小於或大於根節點編號才能記錄答案就不會重複。

 

代碼：

#include <bits/stdc++.h>
using namespace std;

const int mod=1e9+7;
const int Max=2010;
int n,m,size,d;
long long ans,f[Max];
int first[Max],num[Max];
struct shu{int to,next;}edge[Max<<1];

inline void build(int x,int y)
{
	edge[++size].next=first[x],first[x]=size,edge[size].to=y;
	edge[++size].next=first[y],first[y]=size,edge[size].to=x;
}

inline void dfs(int p,int fa,int id)
{
	f[p]=1;
	for(int u=first[p];u;u=edge[u].next)
	{
	  int to=edge[u].to;
	  if(to==fa||num[to]>num[id]||(num[to]==num[id]&&id<to)||num[id]-num[to]>d) continue;
	  dfs(to,p,id);
	  f[p]*=f[to]+1;
	  if(f[p]>=mod) f[p]%=mod;
	}
}

int main()
{
	scanf("%d%d",&d,&n);
	for(int i=1;i<=n;i++) scanf("%d",&num[i]);
	for(int i=1;i<n;i++)
	{
	  int x,y;
	  scanf("%d%d",&x,&y),build(x,y);
	}
	for(int i=1;i<=n;i++)
	{
	  memset(f,0,sizeof(f));
	  dfs(i,0,i);
	  ans+=f[i];
	  if(ans>=mod) ans%=mod;
	}
	cout<<ans;
	return 0;
}