首先连通块,所以点分治肯定是
对于钦定选根的连通块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都被选时,则他们必然也被选,这里一定会在他们祖先那里被算到
考虑直接 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;
}