组合

题目大意:
给定一棵n个点的树，每个点有点权。
现在要求你统计满足如下条件的点集S的数量：
点集S非空
点集S中的点在树上是一个连通块，即如果u和v在集合S中，则从u到v经过的所有点均在S中
点集S中最大点权与最小点权的差值不超过d
输出满足上面条件的点集数量求余1e9+7。

思路:

枚举每一个可能的最小的点权值i,并因此的得到一棵新的树.一颗以点权值i为的数的子集个数等于它所有儿子的值*(ans+1)

Code;

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<vector>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fod(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=2e3+10,MOD=1e9+7;
typedef long long LL;
struct Edge{
    int to,next;
    Edge(int to=0,int next=0):to(to),next(next){} 
}e[N*2];
int len=0,last[N];
void add(int from,int to) {
    e[++len]=Edge(to,last[from]); last[from]=len;
}
LL sum=0;
int d,n,val[N],List[N];
bool visible[N];
LL dfs(int u,int fa) {
    LL ans=1;
    for(int i=last[u];i;i=e[i].next) {
        int v=e[i].to;
        if(visible[v]&&v!=fa) {
            LL temp=dfs(v,u);
            ans=(ans*(temp+1))%MOD;
        }
    }
    return ans;
}
LL solve(int minv) {
    for(int i=1;i<=n;i++)
        if(val[i]>=minv&&val[i]<=minv+d) visible[i]=1;
        else visible[i]=0;
    LL ans=0;
    for(int i=1;i<=n;i++) {
        if(val[i]==minv&&visible[i]) {
            ans+=dfs(i,0);
            visible[i]=0; 
        }
    }
    return ans;
} 
int main() {
    scanf("%d%d",&d,&n);
    fo(i,1,n) scanf("%d",&val[i]);
    for(int u,v,i=1;i<n;i++) {
        scanf("%d%d",&u,&v);
        add(u,v);
        add(v,u);
    }
    fo(i,1,n) List[i]=val[i];
    sort(List+1,List+1+n);
    List[0]=List[1]-1;
    for(int i=1;i<=n;i++) 
        if(List[i]!=List[i-1])
            sum=(sum+solve(List[i]))%MOD;
    printf("%lld\n",sum);
    return 0;
}