Comet OJ

题意:给定一颗 $n$ 个节点的树,定义 $dis(x,y)$ 为树上点 $x$ 到 $y$ 的路径经过的边数.

定义一个点集 $S$ 的 $f(S)$ 为 $f(S)=maxleft {dis(x,y)|x,yin S ight }$ $,|S|geqslant2$

求:对于 $i$ ,有少个点集 $S$ 满足 $|S|geqslant 2$ 且 $f(S)=i$

题解:

上面那个 $f(S)$ 就是这个点集的直径.

考虑枚举直径,我们知道树上的直径有奇数条边/偶数条边两种情况,这里先讲一下直径为偶数条边的情况,奇数条边同理.

假设当前直径为 $i$,那么我们可以枚举直径的中心点 $p$,也可以看作是我们要枚举的点集的中心点.

我们先让 $p$ 为这颗树的树根.

假设当前枚举的半径的半径为 $j$,那么显然 $p$ 的子树中深度为 $(j-1)$ 的点都是可以随便选的(可选可不选).

令这部分方案数为 $re$,则 $re=2^{dep[j-1]}$ 其中 $dep[i]$ 表示当前根的子树中所有深度小于等于 $i$ 的节点数量.

枚举完可以随便选的部分,再枚举一下深度恰好为 $j$ 的部分:令 $sum[j]$ 表示所有儿子中深度恰好为 $j$ 的数量.

那么我们只需保证在这么多点中选大于等于 $2$ 个点即可.

这个的方案数为 $2^{sum[j]}-1$,然后减掉只有一个的情况,就是 $sum_{vin son[p]} 2^{cnt[v][j]}-1$

因为直径可能是奇数,所以将每条边拆成一个点连两条边即可.

#include <bits/stdc++.h>  
#define N 4010 
#define ll long long 
#define mod 998244353    
#define setIO(s) freopen(s".in","r",stdin) 
using namespace std;  
int edges,now,n;  
int hd[N<<1],to[N<<2],nex[N<<2],cnt[N<<1][N],sum[N],bin[N],ans[N];      
void add(int u,int v) 
{
    nex[++edges]=hd[u],hd[u]=edges,to[edges]=v;    
}
void dfs(int u,int ff,int d) 
{
    if(u<=n) ++sum[d], ++cnt[now][d];        
    for(int i=hd[u];i;i=nex[i]) if(to[i]!=ff) dfs(to[i], u, d+1); 
}
int main() 
{  
    // setIO("input");  
    int i,j;  
    scanf("%d",&n);   
    bin[0]=1; 
    for(i=1;i<=n;++i) bin[i]=bin[i-1]*2%mod; 
    for(i=1;i<n;++i) 
    {
        int u,v; 
        scanf("%d%d",&u,&v);    
        add(u,i+n),add(i+n,u);
        add(i+n,v),add(v,i+n);           
    }  
    for(i=1;i<=2*n;++i) 
    {
        now=0;   
        for(j=hd[i];j;j=nex[j]) ++now, dfs(to[j],i,1);         
        int re=(i<=n);     
        for(j=1;j<n;++j) 
        {
            int mdl=bin[sum[j]]-1;   
            for(int k=1;k<=now;++k) 
            {
                (mdl+=mod-bin[cnt[k][j]]+1)%=mod;   
            }    
            (ans[j]+=(ll)mdl*bin[re]%mod)%=mod;    
            re+=sum[j];   
        }
        memset(sum,0,sizeof sum);   
        for(j=1;j<=now;++j) memset(cnt[j], 0, sizeof cnt[j]);    
    }
    for(i=1;i<n;++i) 
        printf("%d
",(ans[i]+mod)%mod);    
    return 0;   
}

  

原文地址:https://www.cnblogs.com/guangheli/p/11656139.html