点分治维护dp+连通块上新型dp思路+乘积方面进行根号dp:0922T4

首先连通块,所以点分治肯定是

Trick1 钦定选根的连通块dp

对于钦定选根的连通块dp,有一种常见思路

先对原树求其dfn序,按dfn序倒序求解

具体的,对于当前点 i i i(注意这里都是指dfn序),我们可以钦定 i i i 是否选

如果 i i i 选,就由 i + 1 i+1 i+1,也就是 i i i 的第一个儿子转移过来(因为只有他选他子树才可能被选)

如果 i i i 不选,就由 i + w i i+w_i i+wi 转移过来,因为他的儿子必然不会被选

至于 i i i i + w i i+w_i i+wi 同时选的情况,我们在 i + 1 i+1 i+1 那里已经算了

对于 i i i i + w i i+w_i i+wi 是否连通的问题,当他们的lca都被选时,则他们必然也被选,这里一定会在他们祖先那里被算到

点分治维护dp+连通块上新型dp思路+乘积方面进行根号dp:0922T4_第1张图片

Trick 2 对于乘积类dp的根号优化方法

考虑直接 d p [ x ] [ i ] dp[x][i] dp[x][i] i i i 值域过大。

但我们可以拆分 f ( x , i ) , g ( x , i ) f(x,i),g(x,i) f(x,i),g(x,i),代表已选乘积为 i i i / 还可以选乘积为 i i i 的方案数

这样状态直接压成 O ( m ) O(\sqrt m) O(m )

其实也可以用整除分块的证明进行预处理

#include
using namespace std;
#define int long long
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<<1)+(x<<3)+(ch^48);ch=getchar();}return x*f;}
#define Z(x) (x)*(x)
#define pb push_back
//mt19937 rand(time(0));
//mt19937_64 rand(time(0));
//srand(time(0));
#define N 4010
#define M 1510
#define mo (int)(1e9+7)
int n, m, i, j, k, T;
int Rt, rt, f[N][M], g[N][M]; 
int mx[N], w[N], dfn[N], tot, sum,  u, v; 
int sq, p[N], v1, v2, a[N], ans; 
vector<int>G[N]; 

void dfs(int x, int fa) {
	w[x]=mx[x]=1; 
	for(int y : G[x]) {
		if(y==fa || p[y]) continue; 
		dfs(y, x); 
		w[x]+=w[y]; 
		mx[x]=max(mx[x], w[y]); 
	}
	mx[x]=max(mx[x], sum-w[x]); 
	if(mx[x]<mx[rt]) rt=x; 
}

void dfs2(int x, int fa) {
	dfn[++tot]=x;  
	for(int y: G[x]) 
		if(y!=fa && !p[y]) dfs2(y, x); 
}

void Add(int &a, int b) {
	a=(a+b)%mo; 
}

void dfz(int x) {
//	printf("> %lld\n", x); 
	int i, j, u; 
	tot=0; dfs(x, 0); dfs2(x, 0); 
//	for(i=1; i<=tot; ++i) printf("%lld ", dfn[i]); printf("\n"); 
	for(i=0; i<=tot+5; ++i)
		for(j=0; j<=sq+5; ++j) f[i][j]=g[i][j]=0; 
//	f[tot+1][1]=1; 
	
	for(i=tot; i>=1; --i) {
		u=dfn[i]; 
		if(a[u]>sq) Add(g[i][m/a[u]], 1); 
		else Add(f[i][a[u]], 1); 
		for(j=1; j<=sq; ++j) {
			v1=i+1; v2=i+w[u]; 
			if(j*a[u]>sq && j*a[u]<=m) Add(g[i][m/(j*a[u])], f[v1][j]); 
			else if(j*a[u]<=m) Add(f[i][j*a[u]], f[v1][j]); 
			if(j>=a[u]) Add(g[i][j/a[u]], g[v1][j]); 
//			
//			
//			
			Add(f[i][j], f[v2][j]); Add(g[i][j], g[v2][j]); 
		}
	}
	for(i=1; i<=sq; ++i) Add(ans, f[1][i]+g[1][i]); 
	
	
	
//	printf("# %lld : %lld\n", x, ans); 
	
	dfs(x, 0); p[x]=1; 
	for(int y : G[x]) 
		if(!p[y]) {
			dfs(y, x); 
			sum=w[y]; 
			mx[rt=0]=1e9; 
			dfs(y, x); 
			dfz(rt); 
		}
}

signed main()
{
//	freopen("in.txt", "r", stdin);
//	freopen("out.txt", "w", stdout);
		freopen("fn.in", "r", stdin);
	freopen("fn.out", "w", stdout);
//	T=read();
//	while(T--) {
//
//	}
	n=read(); m=read(); sq=sqrt(m); 
//	printf("# %lld\n", sq); 
	for(i=1; i<=n; ++i) a[i]=read(); 
	for(i=1; i<n; ++i) {
		u=read(); v=read(); 
		G[u].pb(v); G[v].pb(u); 
	}
	sum=n; mx[rt=0]=1e9; 
	dfs(1, 0); Rt=rt; 
//	printf("%lld\n", rt); 
	dfz(rt);
	printf("%lld", (ans%mo+mo)%mo); 
	return 0;
}


你可能感兴趣的:(点分治,dp优化,树形dp,连通块,dp)