小美的区域会议

n, k = list(map(int, input().split()))
graph = [[] for _ in range(n)]
for _ in range(n - 1):
    u, v = list(map(int, input().split()))
    u -= 1
    v -= 1
    graph[u].append(v)
    graph[v].append(u)
ranks = list(map(int, input().split()))
mod = int(1e9 + 7)

def search(root, node, parent, graph, ranks, minRank, maxRank):
    if ranks[node] > maxRank&nbs***bsp;ranks[node] < minRank:
        return 1
    else:
        global mod
        res = 1
        for anode in graph[node]:
            if anode != parent:
                if ranks[anode] > minRank:
                    res = (res * search(root, anode, node, graph, ranks, minRank, maxRank)) % mod
                elif ranks[anode] == minRank and anode > root:
                    res = (res * search(root, anode, node, graph, ranks, minRank, maxRank)) % mod
        res = (res + 1) % mod
        return res

res = 0
for i in range(n):
    root = i
    minRank = ranks[i]
    maxRank = ranks[i] + k
    res = (res + search(i, i, -1, graph, ranks, minRank, maxRank) - 1) % mod
print(res)

#include <iostream>
#include <vector>
#include <list>
using namespace std;

int n,k;
vector<int> level;
vector<list<int>> adj;
vector<int> visited;
long mod = 1000000007;

int mx_no,mx_level,min_level; // 此次搜索中 最大级别为mx_level（不超过），且级别为mx_level的节点中最大no为mx_no，

long search(int start){
    long count = 1;
    visited[start] = true;
    for(int to : adj[start]){
        if(visited[to])
            continue;
        if(level[to]>mx_level||level[to]<min_level)
            continue;
        if(level[to]==mx_level&&to>mx_no)
            continue;
        count += count * search(to);
        if(count>=mod)
            count %= mod;
    }
    visited[start] = false;
    return count;
}

int main() {
    cin>>n>>k;
    adj.resize(n+1);
    level.resize(n+1);
    visited.resize(n+1);
    for(int i=0;i<n-1;i++){
        int from,to;
        cin>>from>>to;
        adj[from].push_back(to);
        adj[to].push_back(from);
    }
    for(int i=1;i<=n;i++){
        cin>>level[i];
    }
    long count = 0;
    for(int i=1;i<=n;i++){  
        // 选择i：要求搜索结果中，i为级别最高的节点（其它节点级别不超过它），且要求与它同级别的节点中，i的编号最大
        mx_no = i;
        mx_level = level[i];
        min_level = mx_level - k;
        // cerr<<i<<" "<<search(i)<<endl;
        count += search(i);
        if(count>=mod)
            count %= mod;
    }
    cout<<count;
}
// 64 位输出请用 printf("%lld")

def main():
    def dfs(u, fa):
        nonlocal graph, a, k, p, i
        res = 1
        for v in graph[u]:
            if v != fa and (a[i], i) < (a[v], v) and a[i] + k >= a[v]:
                res *= (dfs(v, u) + 1)
                res %= p
        return res

    n, k = list(map(int, input().split()))
    graph = [[] for i in range(n + 1)]

    # 把边的关系存到图里
    for _ in range(n - 1):
        u, v = list(map(int, input().split()))
        graph[u].append(v)
        graph[v].append(u)
    # 负责人级别
    a = [0] + list(map(int, input().split()))

    ans = 0
    p = 10 ** 9 + 7
    for i in range(1, n + 1):
        ans += dfs(i, 0)
        ans %= p
    print(ans)


if __name__ == "__main__":
    main()

import java.util.*;
public class Main{
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int k = sc.nextInt();
        Set<Integer>[]tree = new HashSet[n];
        for(int i=0;i<n;i++){
            tree[i] = new HashSet<>();
        }
        for(int i=0;i<n-1;i++){
            int a = sc.nextInt()-1;
            int b = sc.nextInt()-1;
            tree[a].add(b);
            tree[b].add(a);
        }
        int[]r = new int[n];
        for(int i=0;i<n;i++){
            r[i] = sc.nextInt();
        }
        long ans = 0;
        for(int i=0;i<n;i++){
            boolean[]visited = new boolean[n];
            long a = dfs(tree,visited,i,r,k,r[i],i);
            ans  = (ans+a)%1000000007;
        }
        System.out.println(ans);
    }
    public static long dfs(Set<Integer>[]tree,boolean[]visited,int i,int[]r,int k,int k0,int i0){
        visited[i] = true;
        long a = 1;
        for(int j:tree[i]){
            if(!visited[j]&&r[j]>=k0&&r[j]-k0<=k&&(r[j]>k0||j<i0))
                a = (a*(1+dfs(tree,visited,j,r,k,k0,i0)))%1000000007;
        }
        return a;
    }
}