Valid Sets

Codeforces Round #277 (Div. 2)   D. Valid Sets

链接：http://codeforces.com/contest/486/problem/D

解题思路：1.首先，考虑没有树的节点value情况，那就简化成了求一颗树的所有子树，用dp很容易就可以得到。

　　　　　　即：dp[u]=∏(dp[v]+1)    ,其中u是父节点，v是子节点。

　　　　　2.需要寻找节点value之差不超过d的子树，考虑到找到的子树一定会存在一个value最小的点，所以把每个节点做为根节点，

　　　　　　形成n颗所有节点满足 a[v]>=a[root]&&a[v]<=a[root]+d的树，把每颗树的根节点的dp[root]加起来就是总数。

　　　　　3.注意中间过程的取模，同时注意出现节点value一样的情况，避免重复计算。

代码如下：

#include <cstdio>
#include <cstring>
#include <vector>
#include <cmath>
#include <algorithm>

using namespace std;

const int maxn = 2000 + 10;
const int MOD = 1e9 + 7;

vector<int> node[maxn];
int a[maxn];
long long dp[maxn];
bool f[maxn];
int d, n;

void dfs(int u, int root)
{
    for (int i = 0; i < node[u].size(); i++)
    {
        f[u] = 1;
        int v = node[u][i];
        if(f[v] == 0 && ((a[v] > a[root] && a[v] <= a[root] + d) || (a[v] == a[root] && v > root)))
        {
            dfs(v, root);
            dp[u] = (dp[u] * (dp[v] + 1)) % MOD;
        }
    }
}

int main()
{
    while(scanf("%d%d", &d, &n) != EOF)
    {
        for (int i = 1; i <= n; i++)
        {
            scanf("%d", &a[i]);
        }
        for (int i = 0; i < n - 1; i++)
        {
            int u, v;
            scanf("%d%d", &u, &v);
            node[u].push_back(v);
            node[v].push_back(u);
        }
        int sum = 0;
        for (int i = 1; i <= n; i++)
        {
            memset(f, 0, sizeof(f));
            for (int j = 1; j <= n; j++)
            {
                dp[j] = 1;
            }
            dfs(i, i);
            sum = (sum + dp[i]) % MOD;
        }
        printf("%d
", sum);
    }    
    
    

    return 0;
}

AC如下：