记录编号 |
468801 |
评测结果 |
AAAAAAAAAA |
题目名称 |
树集 |
最终得分 |
100 |
用户昵称 |
Dedsec |
是否通过 |
通过 |
代码语言 |
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;
}