Codeforces Round #511 (Div. 2) E. Region Separation(思维,性质,dp)

LINK

题意

每一轮可以砍掉若干条边,满足每个连通块权值相等,求砍树方案数.

两种方案不同,当且仅当至少存在一轮,剩余的连通块不完全相同.

考虑如果把树分成$k$份,那么每个连通块的权值为$\frac{S}{k}$

我们$dfs$这棵树,若存在一颗子树的权值等于$\frac{S}{k}$,显然需要砍掉这个儿子和父亲的边

这样重复做下去,发现需要砍掉所有满足$\frac{S}{k}|s_u$的$u$与父亲的边

我们定义$f[k]$表示

$$f[k]=\sum _{i=1}^n[\frac{S}{k}|s_i]$$

一种划分要么无解,有解必然只有唯一解,且满足$f[k]==k$

若我们处理了这样的$f$数组就知道哪些划分是有解的

定义$dp[i]$表示最后一轮的结果是$k$划分的方案数,有转移

$dp[k]=\sum _{k|j}dp[j]$

问题变成求解$f$数组来给$dp$数组赋上初值

求$f$数组,对于每个固定的$k$可以花$O(n)$的时间检验

由于$S$很大,枚举因子的想法也不可行,复杂度达到$O(n^2)$

$$f[k]=\sum _{i=1}^n[\frac{S}{k}|s_i]$$

观察到$\frac{S}{k}|s_i$的条件为

$$s_i=a\ast \frac{S}{k}$$

也就是$\frac{S}{k}|s_i$,$\frac{S}{k}|S$,则有$\frac{S}{k}|gcd(s_i,S)$

有$\frac{S}{gcd(S,s_i)}|k$

于是合法的$k$容易求,只需要枚举$\frac{S}{gcd(S,s_i)}$的倍数即可

#include <iostream>
#include <bits/stdc++.h>
#define int long long
using namespace std;
const int maxn = 1e6+10;
const int mod = 1e9+7;
int n,f[maxn],s[maxn],dp[maxn],c[maxn];
vector<int>vec[maxn];
void dfs(int u,int fa)
{
	for( auto v:vec[u] )
	{
		if( v==fa )	continue;
		dfs( v,u );
		s[u] += s[v];
	}
}
int gcd(int a,int b){ return b==0?a:gcd(b,a%b); }
void initF()
{
	for(int i=1;i<=n;i++)
	{
		int x = s[1]/gcd( s[i],s[1] );
		if( x<=n )	c[x]++;
	}	
	for(int i=1;i<=n;i++)
	{
		for(int j=i;j<=n;j+=i)	f[j] += c[i];	
	}
}
signed main()
{
	cin >> n;
	for(int i=1;i<=n;i++)	scanf("%lld",&s[i] );
	for(int i=1;i<n;i++)
	{
		int x;	scanf("%lld",&x );
		vec[x].push_back( i+1 );
		vec[i+1].push_back( x );
	}
	dfs( 1,0 );
	initF();
	int ans = 0;
	dp[1] = 1;
	for(int i = 1;i <= n;++ i) 
		if( f[i] == i )
		{
			for(int j = i << 1;j <= n;j += i)
				dp[j] = ( dp[j] + dp[i] )%mod;
			ans = ( ans + dp[i] )%mod;
		}
	cout << ans;
}