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 0x3f3f3f3fusing 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;}