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=;
 int tot,go[],first[],next[],a[],d,n;
 ll f[];
 int read(){
     int t=,f=;char ch=getchar();
     while (ch<''||ch>''){if (ch=='-') f=-;ch=getchar();}
     while (''<=ch&&ch<=''){t=t*+ch-'';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]=;
     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=;i<=n;i++)
      a[i]=read();
     for (int i=;i<n;i++){
         int x=read(),y=read();
         add(x,y);
     }
     ll ans=;
     for (int i=;i<=n;i++){
      for (int j=;j<=n;j++) f[j]=;
      dfs(i,,i);
      ans=(ans+f[i])%Mod;
     }
     printf("%I64d\n",ans);
 }