标签：个性 数组 计算 node ace fine logs codeforce printf
给出一棵树，求出这棵树的不同联通子节点集合的数量，这些集合必须满足最大权值点减最小权值点小于等于d。
再一次树d乱搞。
因为数据范围贼小，所以我们对于每一个点为根的情况进行一次dfs.
对于以节点root为根的情况，我们认为root是最小值。
对于root的某一棵子树，其需要算的情况一定包含root，否则前面已经算过了。
于是我们就对这个root进行dfs。dfs过程中，我们始终保证root为最小值，不遍历大于它的点。
因为有了这个性质，我们只要不走加起来大于d的点，接下来所有的点都不属于被计算范围内，可以不走。
在这个以root为根的树中，现在我们遍历到u，选中它的一颗子树v。
对于这一串东西上，我们每次可以选择以前已遍历过的子树中选择某些点，再从v该棵子树下选择某些点组成新情况。
我们增加的情况是已经遍历过的子树答案*（该子树被遍历到的子树总答案+1）
加一是因为空集也算是一种情况。
也就是f[u]*=f[v]+1;
对于每个叶子结点，我们只需要将f设为1即可。
dfs完以后，我们将以每个点为根（最小值）的情况进行累加。
于是最终的ans便是答案。
需要注意的是，这一道题MOD1000000007
所以f数组、ans变量需要long long类型
而我一直爆炸，因为没注意到，dfs函数也要为long long类型。
所以请注意这个问题。
#include <iostream> #include <cstdio> #include <cstring> #define MOD 1000000007 using namespace std; struct node{ int to,next; }e[4010]; int d,n,u,v,h[2010],a[2010],tot=0; long long f[2010],ans=0; void add(int u,int v){ e[++tot].to=u;e[tot].next=h[v];h[v]=tot; e[++tot].to=v;e[tot].next=h[u];h[u]=tot; } long long dfs(int x,int last,int root){ long long ans=1; f[x]=1; for(int i=h[x];~i;i=e[i].next){ if(last!=e[i].to&&(a[e[i].to]>a[root]||(a[e[i].to]==a[root]&&root>e[i].to)) &&a[e[i].to]-a[root]<=d){ ans=(ans*(dfs(e[i].to,x,root)+1)%MOD)%MOD; } } return ans; } int main(){ // freopen("cf486d.in","r",stdin); memset(h,-1,sizeof(h)); scanf("%d%d",&d,&n); for(int i=1;i<=n;i++)scanf("%d",&a[i]); for(int i=1;i<n;i++){ scanf("%d%d",&u,&v); add(u,v); } for(int i=1;i<=n;i++)ans+=dfs(i,0,i); printf("%lld",ans%MOD); }
标签：个性 数组 计算 node ace fine logs codeforce printf
原文地址：http://www.cnblogs.com/skylynf/p/7143389.html
