Codeforces 486D D. Valid Sets

http://codeforces.com/contest/486/problem/D

题意:给定一棵树，点上有权值，以及d，要求有多少种联通块满足最大值减最小值小于等于d。

思路：枚举i作为最大的点权，然后dfs树规一下，就能得出以这个点为最大值的方案数，因为有权值相等的点，所以我们规定一下，只能从标号小的拓展到标号大的，就不会重复了。

#include<algorithm>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<iostream>
#define ll long long
const ll Mod=1000000007;
int tot,go[200005],first[200005],next[200005],a[200005],d,n;
ll f[200005];
int read(){
    int t=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while ('0'<=ch&&ch<='9'){t=t*10+ch-'0';ch=getchar();}
    return t*f;
}
void insert(int x,int y){
    tot++;
    go[tot]=y;
    next[tot]=first[x];
    first[x]=tot;
}
void add(int x,int y){
    insert(x,y);insert(y,x);
}
void dfs(int x,int fa,int fi){
    f[x]=1;
    for (int i=first[x];i;i=next[i]){
        int pur=go[i];
        if (pur==fa) continue;
        if (a[pur]>a[fi]) continue;
        if (a[fi]-d>a[pur]) continue;
        if (a[fi]==a[pur]&&fi>pur) continue;
        dfs(pur,x,fi);
        f[x]*=(1LL+f[pur]);
        f[x]%=Mod;
    }
}
int main(){
    d=read();n=read();
    for (int i=1;i<=n;i++) 
     a[i]=read();
    for (int i=1;i<n;i++){
        int x=read(),y=read();
        add(x,y);
    } 
    ll ans=0;
    for (int i=1;i<=n;i++){
     for (int j=1;j<=n;j++) f[j]=0;
     dfs(i,0,i);
     ans=(ans+f[i])%Mod;
    }
    printf("%I64d
",ans);
}