我还在想没给定根怎么能树形dp呢。。。我们枚举每一个i来当跟,然后每次统计过根且a[i]为最大值的方案数。考虑到如果a[i]!=a[j],则一定不会重复计数。那可能重复的就是a[i]==a[j]的情况,那我们强制规定一个方向,让标号小的能访问到相等的大的即可。这样就不会重复计数了。
#include <bits/stdc++.h>
#define ll long long
#define mod 1000000007
#define N 2010
#define inf 0x3f3f3f3f
using namespace std;
inline int read(){
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
return x*f;
}
int d,n,a[N],h[N],num=0,ans=0;
bool vis[N];
struct edge{
int to,next;
}data[N<<1];
int dfs(int x,int rt){
ll res=1;vis[x]=1;
for(int i=h[x];i;i=data[i].next){
int y=data[i].to;
if(vis[y]||a[y]>a[rt]||a[y]<a[rt]-d) continue;
if(a[y]==a[rt]&&y<rt) continue;
res+=(ll)res*dfs(y,rt)%mod;res%=mod;
}return res;
}
int main(){
// freopen("a.in","r",stdin);
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();
data[++num].to=y;data[num].next=h[x];h[x]=num;
data[++num].to=x;data[num].next=h[y];h[y]=num;
}
for(int i=1;i<=n;++i){//枚举以i为根,且a[i]为最大值,过根的方案数
memset(vis,0,sizeof(vis));ans=(ans+dfs(i,i))%mod;
}printf("%d\n",ans);
return 0;
}