CF486D Valid Sets（枚举+树形dp）

我还在想没给定根怎么能树形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;
}