CF486D Valid Sets

最新推荐文章于 2024-10-09 09:53:23 发布

minato_yukina

最新推荐文章于 2024-10-09 09:53:23 发布

阅读量142

点赞数

文章标签：深度优先图论算法

本文链接：https://blog.csdn.net/qq_36018057/article/details/126461526

版权

本文介绍了一种巧妙的动态规划方法，通过dp[u]计算在给定约束条件下，点u作为最小元素时的连通子图数目。关键在于利用最大-最小约束的特性避免重复，并在dfs中处理相同节点的贡献防止冗余。通过实例代码展示了如何使用这种方法解决实际问题。

摘要由CSDN通过智能技术生成

在这里插入图片描述
思路:比较妙的一个思路,考虑 $dp(u)为点u作为最小的a_u情况下连通子图的数目$
为什么要这么做,因为最大-最小<= $d$ 这个约束十分麻烦,如果我们在枚举点的同时,认为它是作为集合中的最大/最小,就能不重复不遗漏地统计出合法的子图.
接下来就是dp过程，+1代表可以不选择该子树一种方案. $dp(u)=\prod dp(v)+1$
在dfs过程中,要特别注意 $a [u] == a [v]$ ,事实上对于这两种点,得到的连通子图事实上有重复的部分。当以 $u$ 为根的贡献算出来后,再次以 $v$ 为根时,如果统计了 $u$ 为根的子树,显然会造成重复部分的生成。因为这一部分已经在以 $u$ 为根的时候被统计出来过一次了,这要求在dfs(v)的时候我们不能再次访问 $u$ 节点.既然相同的点只能访问一次,不妨让数组编号小的访问大的.

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e6+5;
const int INF = 1e9+7;
typedef long long ll;
typedef pair<int,int> pii;
#define all(a) (a).begin(), (a).end()
#define pb(a) push_back(a)
const int mod = INF;
vector<int> G[maxn];
int a[maxn];ll dp[maxn];int d,n;
void dfs(int u,int fa,int x){
	if(a[x]<a[u]||(a[x]==a[u]&&x<u)||(abs(a[u]-a[x])>d)) return ;
	dp[u] = 1;
	for(auto v : G[u]){
		if(v==fa) continue;
		dfs(v,u,x);
		dp[u] = 1LL*dp[u]*(dp[v]+1)%mod;
	}
}
int main(){
    ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin>>d>>n;
	for(int i=1;i<=n;i++) cin>>a[i];
	for(int i=1;i<=n-1;i++){
		int u,v;cin>>u>>v;
		G[u].pb(v);G[v].pb(u);
	}
	ll ans = 0;
	for(int i=1;i<=n;i++){
		memset(dp,0,sizeof(dp));
		dfs(i,0,i);
		ans = (ans+dp[i])%mod;
	}
	cout<<ans<<"\n";
}