Codeforces 486 D Valid Sets 记忆化搜索

最新推荐文章于 2019-09-21 23:13:58 发布

Winchester_

最新推荐文章于 2019-09-21 23:13:58 发布

阅读量739

点赞数

CC 4.0 BY-SA版权

分类专栏： Codeforces 动态规划专项练习文章标签： codeforces dp 搜索

本文链接：https://blog.youkuaiyun.com/Winchester_/article/details/51871861

Codeforces 动态规划专项练习专栏收录该内容

11 篇文章

订阅专栏

本文解析了一道Codeforces D级难度题目，该题要求求解一棵树中满足特定条件的连通子图数量。通过枚举每个节点作为最大值节点进行深度优先搜索，最终统计出所有符合条件的子图数目。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

http://codeforces.com/contest/486/problem/D

题目大意：给出一棵树，求这棵树的满足最大点与最小点之差小于等于d的连通子图的个数。

题目分析：枚举每个点，假设当前该点就是该联通子图的最大点，进行遍历，最后将每个点的方案总数累加起来即可。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<functional>
#include<cmath>
#include<cctype>
#include<cassert>
#include<climits>
using namespace std;
#define For(i,n) for(int i=1;i<=n;i++)
#define Rep(i,n) for(int i=0;i<n;i++)
#define Fork(i,k,n) for(int i=k;i<=n;i++)
#define ForD(i,n) for(int i=n;i;i--)
#define Forp(x) for(int p=first[x];p;p=e[p].next)
#define RepD(i,n) for(int i=n;i>=0;i--)
#define MEM(a) memset(a,0,sizeof(a))
#define MEMI(a) memset(a,127,sizeof(a))
#define MEMi(a) memset(a,128,sizeof(a))
#define INF (2139062143)
#define phiF (1000000006)
#define MAXN (1000000+10)
typedef long long ll;

const ll mo=1000000007;
struct info{
	int to,next;
}e[8005];

ll f[2005];
int tot,n,d,a[2005],first[10005],x,y,ans;

void add(int x,int y){
	tot++;
	e[tot].to=y;
	e[tot].next=first[x];
	first[x]=tot;
}

void dfs(int mx,int u,int t){
	f[u]=1;
	Forp (u){
		int v=e[p].to;
		if (v==t) continue;//u 防止返回；
		if (a[v]>a[mx]) continue;
		if ((a[mx]-a[v])>d) continue;
		if (a[v]==a[mx]&&v<mx) continue;//避免多种情况 
		dfs(mx,v,u); 
		f[u]*=f[v]+1ll;
        f[u]%=mo;
	}
	
}

int main(){
	scanf("%d%d",&d,&n);
	For (i,n) scanf("%d",&a[i]);
	For (i,n-1) {
		scanf("%d%d",&x,&y);
		add(x,y);
		add(y,x);
	}
	
	For (i,n){
		dfs(i,i,-1);
		ans=(f[i]+ans)%mo;
		
	}
	printf("%d",ans);
}