[HDU多校]Ridiculous Netizens

[HDU多校]Ridiculous Netizens

• 点分治
  • 分成两个部分：对某一点P，连通块经过P或不经过P．
  • 经过P采用树形依赖背包
  • 不经过P的部分递归计算
• 树型依赖背包
  • v点必须由其父亲u点转移过来
  • 即必须经过P点
  • \(dp[v][s*a[v]]+=dp[u][s]\)
  • 第二维代表连通块的乘积
  • 第一维代表经过该点并且一定经过P点的方案数
  • 所以最后父节点还要加上子节点的方案数
• 空间优化
  • 第二维不能开这么大
  • 稍稍转变含义，改成还能＂装下＂多少体积
  • \(\lfloor \frac{m}{s}\rfloor\)就可以划分很多个等价的状态

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn = 2010;
const int mod = 1e9+7;
struct edge{
    int to,next;
}e[maxn<<1];
ll val[maxn],kuai[maxn];
ll dp[maxn][maxn];
int nowN,rt,maxP,head[maxn],tol,cnt,n;
int Size[maxn];
bool vis[maxn<<1];
ll ans;
int m;
inline void Mod(ll &a,ll b){
   a=a+b<mod?a+b:a+b-mod;
}
void init(int n){
    tol=1;
    for(int i=0;i<=n;++i){
        head[i]=0;
    }
    rt=0;
    cnt=ans=0;
}
void add(int u,int v){
    tol++;
    vis[tol]=0;
    e[tol].to=v;
    e[tol].next=head[u];
    head[u]=tol;
}
void get_rt(int u,int fa){
   int max_part=0;
   Size[u]=1;
   for(int i=head[u],v;i;i=e[i].next){
      v=e[i].to;
      if(vis[i]||v==fa)continue;
      get_rt(v,u);
      Size[u]+=Size[v];
      max_part=max(max_part,Size[v]);
   }
   max_part=max(nowN-Size[u],max_part);
   if(max_part<maxP){
     maxP=max_part;
     rt=u;
   }
}
void dfs(int u,int fa){
   for(int i=1;i<=cnt;++i) dp[u][i]=0;
   ll k=val[u],t;
   for(int i=1,j=1;i<=cnt;++i){
        t=kuai[i]/k;
        if(t==0)break;
        while(kuai[j]>t)++j;
        Mod(dp[u][j],dp[fa][i]);
   }
   for(int i=head[u],v;i;i=e[i].next){
      if(vis[i]) continue;
      v=e[i].to;
      if(v==fa) continue;
      dfs(v,u);
      for(int j=1;j<=cnt;++j) Mod(dp[u][j],dp[v][j]);
   }
}
void calc(){
   dp[0][1]=1;
   dfs(rt,0);
   for(int i=1;i<=cnt;++i) Mod(ans,dp[rt][i]);
   for(int i=head[rt],v;i;i=e[i].next){
       if(vis[i])continue;
       v=e[i].to;
       vis[i^1]=1;
       nowN=Size[v];
       maxP=n+5;
       rt=0;
       get_rt(v,0);
       calc();
   }
}
int main(){
    int _;
    scanf("%d",&_);
    while(_--){
        scanf("%d%d",&n,&m);
        init(n);
        for(int i=1;i<=n;++i)scanf("%lld",&val[i]);
        for(int i=1;i<=m;i=m/(m/i)+1){
            kuai[++cnt]=m/i;
        }
        for(int i=1,u,v;i<n;++i){
            scanf("%d%d",&u,&v);
            add(u,v);
            add(v,u);
        }
        nowN=n;
        maxP=n+5;
        get_rt(1,1);
        calc();
        printf("%lld\n",ans);
    }
    return 0;
}

代码量还是集中在点分治，不过套班子即可

DP还是好理解的.