记录编号 468801 评测结果 AAAAAAAAAA
题目名称 树集 最终得分 100
用户昵称 GravatarDedsec 是否通过 通过
代码语言 C++ 运行时间 0.030 s
提交时间 2017-11-02 08:40:54 内存使用 0.40 MiB
显示代码纯文本
#include<iostream>
#include<cstring>
#include<cstdio>
#include<string>
#include<queue>
#include<vector>
#include<algorithm>
#include<cmath>
#include<stack>
#include<map>
using namespace std;
#define R register
#define ll long long
#define fo(i,a,b) for(R int (i)=(a);(i)<=(b);++(i))
#define debug(x) cout<<#x<<"="<<x<<endl
#define mod 1000000007
int n,d,a[2010],head[2010<<1],to[2010<<1],nxt[2010<<1],q,tot=1;
ll f[2010],ans,sum[2010];
void add(int x,int y)
{
	nxt[++tot]=head[x];head[x]=tot;
	to[tot]=y;
}
void dfs(int x,int fa)
{
	sum[x]=1;
	for(int i=head[x];i;i=nxt[i])
	{
		if(a[to[i]]>a[q]+d||to[i]==fa||a[to[i]]<a[q])continue;
		if(a[to[i]]==a[q]&&to[i]<q)continue;
		dfs(to[i],x);
		sum[x]*=(sum[to[i]]+1);
		sum[x]%=mod;
	}
}
int main()
{
	freopen("treeset.in","r",stdin);
	freopen("treeset.out","w",stdout);
	scanf("%d%d",&d,&n);
	fo(i,1,n)scanf("%d",&a[i]);
	fo(i,2,n)
	{
		R int x,y;
		scanf("%d%d",&x,&y);
		add(x,y);add(y,x);
	}
	fo(i,1,n)
	{
		q=i;
		dfs(i,0);
		ans+=sum[i];ans%=mod;
	}
	cout<<ans;
	return 0;
}