bzoj 5093 图的价值 —— 第二类斯特林数+NTT

题目:https://www.lydsy.com/JudgeOnline/problem.php?id=5093

每个点都是等价的,从点的贡献来看,得到式子:

\( ans = n * \sum\limits_{d=0}^{n-1} d^{k} * 2^{C_{n-1}^{2}} * C_{n-1}^{d} \)

使用 \( n^{k} = \sum\limits_{i=0}^{k} S(k,i) * i! *C_{n}^{i} \)

得到 \( ans = n * \sum\limits_{d=0}^{n-1} 2^{C_{n-1}^{2}} * C_{n-1}^{d} * \sum\limits_{j=0}^{k} S(k,j) * j! * C_{d}^{j} \)

此时不要把组合数拆成阶乘!虽然拆成阶乘可以消去 \( d! \),但如果不消去,放在一起可以得到新的组合意义;

\( ans = n * 2^{C_{n-1}^{2}} * \sum\limits_{j=0}^{k} S(k,j) * j! * \sum\limits_{d=0}^{n-1} C_{n-1}^{d} * C_{d}^{j} \)

而 \( \sum\limits_{d=0}^{n-1} C_{n-1}^{d} * C_{d}^{j} \) 表示从 \( n-1 \) 个人里选 \( d \) 个人,再从 \( d \) 个人里选 \( j \) 个人;

其实就是从 \( n-1 \) 个人里选 \( j \) 个人,剩下的人随便选,即 \( C_{n-1}^{j} * 2^{n-1-j} \)

所以 \( ans = n * 2^{C_{n-1}^{2}} * \sum\limits_{j=0}^{k} S(k,j) * j! * C_{n-1}^{j} * 2^{n-1-j} \)

而通过 \( S(n,m) = \frac{1}{m!} \sum\limits_{k=0}^{m} C_{m}^{k} * (m-k)^{n} * (-1)^{k} \) (枚举 \( k \) 个空组,最后除去 \( m \) 组的排列)

即 \( S(n,m) = \sum\limits_{k=0}^{m} \frac{(m-k)^{n}}{(m-k)!} * \frac{(-1)^{k}}{k!} \)

可以用NTT求出一行的第二类斯特林数,也就是求出 \( S(k,i) \)

然后把 \( C_{n-1}^{j} \) 拆开约分,上下都只有 \( k \) 级别,预处理即可;

还是要注意次数是对 \( mod-1 \) 取模。

代码如下:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
int const xn=2e5+5,xm=(1<<19),mod=998244353;
int n,m,lim,a[xm],b[xm],rev[xm],jc[xn],jcn[xn],jd[xn];
int upt(int x){while(x>=mod)x-=mod; while(x<0)x+=mod; return x;}
ll pw(ll a,ll b)
{
  ll ret=1; a=a%mod; b=b%(mod-1);
  for(;b;b>>=1,a=(a*a)%mod)if(b&1)ret=(ret*a)%mod;
  return ret;
}
void init()
{
  jc[0]=1;
  for(int i=1;i<=m;i++)jc[i]=(ll)jc[i-1]*i%mod;
  jcn[m]=pw(jc[m],mod-2);
  for(int i=m-1;i>=0;i--)jcn[i]=(ll)jcn[i+1]*(i+1)%mod;
  jd[0]=1;
  for(int j=1;j<=m;j++)jd[j]=(ll)jd[j-1]*(n-j)%mod;
}
void ntt(int *a,int tp)
{
  for(int i=0;i<lim;i++)
    if(i<rev[i])swap(a[i],a[rev[i]]);
  for(int mid=1;mid<lim;mid<<=1)
    {
      int len=(mid<<1),wn=pw(3,tp==1?(mod-1)/len:(mod-1)-(mod-1)/len);
      for(int j=0;j<lim;j+=len)
    for(int k=0,w=1;k<mid;k++,w=(ll)w*wn%mod)
      {
        int x=a[j+k],y=(ll)w*a[j+mid+k]%mod;
        a[j+k]=upt(x+y); a[j+mid+k]=upt(x-y);
      }
    }
  if(tp==1)return; int inv=pw(lim,mod-2);
  for(int i=0;i<lim;i++)a[i]=(ll)a[i]*inv%mod;
}
int main()
{
  scanf("%d%d",&n,&m); init();
  lim=1; int l=0;
  while(lim<=m+m)lim<<=1,l++;
  for(int i=0;i<lim;i++)rev[i]=((rev[i>>1]>>1)|((i&1)<<(l-1)));
  for(int i=0;i<=m;i++)a[i]=(ll)pw(i,m)*jcn[i]%mod;
  for(int i=0;i<=m;i++)b[i]=upt((i&1?-1:1)*jcn[i]);
  ntt(a,1); ntt(b,1);
  for(int i=0;i<lim;i++)a[i]=(ll)a[i]*b[i]%mod;
  ntt(a,-1);
  int ans=0;
  for(int j=0;j<=m;j++)
    ans=(ans+(ll)a[j]*jc[j]%mod*jd[j]%mod*jcn[j]%mod*pw(2,n-1-j))%mod;
  printf("%lld\n",(ll)n*pw(2,((ll)(n-1)*(n-2)/2))%mod*ans%mod);
  return 0;
}

原文地址:https://www.cnblogs.com/Zinn/p/10070115.html

时间: 2024-10-07 18:33:29

bzoj 5093 图的价值 —— 第二类斯特林数+NTT的相关文章

bzoj5093:图的价值(第二类斯特林数+NTT)

传送门 首先,题目所求为\[n\times 2^{C_{n-1}^2}\sum_{i=0}^{n-1}C_{n-1}^ii^k\] 即对于每个点\(i\),枚举它的度数,然后计算方案.因为有\(n\)个点,且关于某个点连边的时候剩下的边都可以随便连,所以有前面的两个常数 所以真正要计算的是\[\sum_{i=0}^{n-1}C_{n-1}^ii^k\] 根据第二类斯特林数的性质,有\[i^k=\sum_{j=0}^iS(k,j)\times j!\times C_i^j\] 然后带入,得\[\s

BZOJ4555 [Tjoi2016&amp;Heoi2016]求和 【第二类斯特林数 + NTT】

题目 在2016年,佳媛姐姐刚刚学习了第二类斯特林数,非常开心. 现在他想计算这样一个函数的值: S(i, j)表示第二类斯特林数,递推公式为: S(i, j) = j ? S(i ? 1, j) + S(i ? 1, j ? 1), 1 <= j <= i ? 1. 边界条件为:S(i, i) = 1(0 <= i), S(i, 0) = 0(1 <= i) 你能帮帮他吗? 输入格式 输入只有一个正整数 输出格式 输出f(n).由于结果会很大,输出f(n)对998244353(7

bzoj 5093 [Lydsy1711月赛]图的价值 NTT+第二类斯特林数

[Lydsy1711月赛]图的价值 Time Limit: 30 Sec  Memory Limit: 256 MBSubmit: 245  Solved: 128[Submit][Status][Discuss] Description “简单无向图”是指无重边.无自环的无向图(不一定连通). 一个带标号的图的价值定义为每个点度数的k次方的和. 给定n和k,请计算所有n个点的带标号的简单无向图的价值之和. 因为答案很大,请对998244353取模输出. Input 第一行包含两个正整数n,k(

【BZOJ5093】图的价值(第二类斯特林数,组合数学,NTT)

[BZOJ5093]图的价值(第二类斯特林数,组合数学,NTT) 题面 BZOJ 题解 单独考虑每一个点的贡献: 因为不知道它连了几条边,所以枚举一下 \[\sum_{i=0}^{n-1}C_{n-1}^i·i^k·2^{\frac{n(n-1)}{2}}\] 因为有\(n\)个点,所以还要乘以一个\(n\) 所以,我们真正要求的就是: \[\sum_{i=0}^{n-1}C_{n-1}^i·i^k\] 怎么做? 看到了\(i^k\)想到了第二类斯特林数 \[m^n=\sum_{i=0}^{m}

【bzoj5093】[Lydsy1711月赛]图的价值(NTT+第二类斯特林数)

题意: 给定\(n\)个点,一个图的价值定义为所有点的度数的\(k\)次方之和. 现在计算所有\(n\)个点的简单无向图的价值之和. 思路: 将式子列出来: \[ \sum_{i=1}^n\sum_{j=0}^{n-1}{n-1\choose j}2^{\frac{(n-1)(n-2)}{2}}j^k \] 表示分别考虑每个点的贡献,我们只需要枚举其度数即可,其余的边任意连. 然后我们将后面的\(j^k\)用第二类斯特林数展开: \[ \begin{aligned} &\sum_{i=1}^{n

【BZOJ 4555】[Tjoi2016&amp;Heoi2016]求和 NTT+第二类斯特林数

用到第二类斯特林数的性质,做法好像很多,我打的是直接ntt,由第二类斯特林数的容斥公式可以推出,我们可以对于每一个i,来一次ntt求出他与所有j组成的第二类斯特林数的值,这个时候我们是O(n^2logn)的,还不如暴力,但是我们发现,对于刚刚提到的容斥的式子,将其化为卷积形式后,其一边的每一项对于每一个i都相同,另一边的每一项是对于所有的i形成一个n项的等比数列,这样我们可以把成等比数列的一边求和,用固定的一边去卷他们的和,这时候的答案的每一项就是所有的i的这一项的和,然后我们再O(n)乘上阶乘

bzoj 4555 [Tjoi2016&amp;Heoi2016]求和 NTT 第二类斯特林数 等比数列求和优化

[Tjoi2016&Heoi2016]求和 Time Limit: 40 Sec  Memory Limit: 128 MBSubmit: 679  Solved: 534[Submit][Status][Discuss] Description 在2016年,佳媛姐姐刚刚学习了第二类斯特林数,非常开心. 现在他想计算这样一个函数的值: S(i, j)表示第二类斯特林数,递推公式为: S(i, j) = j ∗ S(i − 1, j) + S(i − 1, j − 1), 1 <= j &l

【BZOJ4555】求和(第二类斯特林数,组合数学,NTT)

[BZOJ4555]求和(第二类斯特林数,组合数学,NTT) 题面 BZOJ 题解 推推柿子 \[\sum_{i=0}^n\sum_{j=0}^iS(i,j)·j!·2^j\] \[=\sum_{i=0}^n\sum_{j=0}^nS(i,j)·j!·2^j\] \[=\sum_{i=0}^n\sum_{j=0}^nj!·2^j(\frac{1}{j!}\sum_{k=0}^j(-1)^k·C_j^k·(j-k)^i)\] \[=\sum_{j=0}^n2^j\sum_{k=0}^j(-1)^k

【cf961G】G. Partitions(组合意义+第二类斯特林数)

传送门 题意: 给出\(n\)个元素,每个元素有价值\(w_i\).现在要对这\(n\)个元素进行划分,共划分为\(k\)组.每一组的价值为\(|S|\sum_{i=0}^{|S|}w_i\). 最后询问所有划分的总价值. 思路: 直接枚举划分不好计算,考虑单独计算每一个元素的贡献,那么就有式子: \[ \sum_{i=1}^nw_i\sum_{j=1}^{n-k+1}{n-1\choose j-1}\begin{Bmatrix} n - j \\ k - 1 \end{Bmatrix}j \]