BJ模拟 生日礼物（斯特林数+NTT）

最新推荐文章于 2021-08-22 10:36:09 发布
最新推荐文章于 2021-08-22 10:36:09 发布
阅读量466 收藏 1
点赞数
分类专栏： ntt、fft 斯特林数 文章标签： nttfft 斯特林数
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_35649707/article/details/65445628
版权

Description

     今天是Jane的生日。Alice和Bob都有一些糖果，于是这两个人就去买一些白色的盒子去包装这些糖果作为Jane的生日礼物。他们将随机地把这些盒子分成两堆，一堆给Alice，一堆给Bob（每堆至少有一个盒子）。

      我们知道Alice有 N1 个不同的糖果，Bob有 N2 个相同的糖果（由于Bob很懒，所以他直接买了相同的糖果），然后Alice买的糖果和Bob买的糖果是完全不一样的。他们两人各自把糖果放进盒子里，并且任意一个盒子都不会是空的。Alice和Bob各自有各自包装糖果的规则：

      Alice很有个性，她会根据盒子的数量来决定包装糖果的方法。记Alice拥有的盒子数量为 k 。若 k 是个质数，则她会随机地包装糖果（每个糖果都有可能进入任意一个盒子里）。若 k 不是质数，她会往其中的 k−1 个盒子里分别随机放入1个糖果，将剩下的糖果全部放入剩下那个盒子里。

     Bob有M种颜料（这M种颜料中没有白色），他将随机地给盒子上色，使得每个盒子都有上色且颜色不一样。然后Bob会随机地包装糖果（每个糖果都可能进入任意一个盒子里）。

     问题来了，Jane会收到多少种礼物？

Input

     第一行为一个整数T，表示测试数据组数。

      对于每组测试数据，第一行为四个整数 N1,N2,M,Q ，分别表示Alice的糖果数目、Bob的糖果数目、Bob的颜色数目、询问个数。

      接下来Q行，每行一个整数N，表示Alice和Bob购买的盒子的总数 (2≤N≤M) 。

Output

      对于每组测试数据，输出Q行，每行一个整数表示Jane可能收到的礼物的种数。由于答案可能非常大，请将答案对786433取模后输出。

Sample Input

1

4 3 3 2

2

3

Sample Output

3

27

HINT

【样例解释】

Alice的糖果为 A,B,C,D，Bob的糖果为 a,a,a ，Bob的颜色为 C1, C2, C3。盒子初始的颜色为W。

 

对于n = 2，Jane可以收到3种礼物：

W{A,B,C,D},C1{a,a,a}

W{A,B,C,D},C2{a,a,a}

W{A,B,C,D},C3{a,a,a}

 

对于n = 3，Jane可以收到27种礼物：

W{A},W{B,C,D},C1{a,a,a}

W{A},W{B,C,D},C2{a,a,a}

W{A},W{B,C,D},C3{a,a,a}

W{B},W{A,C,D},C1{a,a,a}

W{B},W{A,C,D},C2{a,a,a}

W{B},W{A,C,D},C3{a,a,a}

W{C},W{A,B,D},C1{a,a,a}

W{C},W{A,B,D},C2{a,a,a}

W{C},W{A,B,D},C3{a,a,a}

W{D},W{A,B,C},C1{a,a,a}

W{D},W{A,B,C},C2{a,a,a}

W{D},W{A,B,C},C3{a,a,a}

W{A,B},W{C,D},C1{a,a,a}

W{A,B},W{C,D},C2{a,a,a}

W{A,B},W{C,D},C3{a,a,a}

W{A,C},W{B,D},C1{a,a,a}

W{A,C},W{B,D},C2{a,a,a}

W{A,C},W{B,D},C3{a,a,a}

W{A,D},W{B,C},C1{a,a,a}

W{A,D},W{B,C},C2{a,a,a}

W{A,D},W{B,C},C3{a,a,a}

W{A,B,C,D},C1{a},C2{a,a}

W{A,B,C,D},C1{a},C3{a,a}

W{A,B,C,D},C2{a},C1{a,a}

W{A,B,C,D},C2{a},C3{a,a}

W{A,B,C,D},C3{a},C1{a,a}

W{A,B,C,D},C3{a},C2{a,a}

【数据范围与约定】

      对于10%的数据： 1≤N1,N2,M,Q≤10

      对于50%的数据： 1≤N1,N2,M,Q≤1000

      对于所有数据： 1≤N1,N2,M,Q≤105,T≤5




题解：斯特林数+NTT

暴力（40）：

对于Alice是质数时相当于把n个本质不同的球放入m个本质相同的盒子中为第一类斯特林数

是合数时相当于从n个球中挑出n-m+1个球放入一个球中的方案数采用隔板法可知为组合数。

对于Bob可以看成把n个相同球放入m个盒子的方案数乘将M个颜色分配给m个盒子的方案数

对于前一部分可用插板法（n-1个空位用m-1个板子隔开）方案数为C[n-1][m-1].

暴力枚举即可。

40分代码：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
using namespace std;
typedef unsigned long long ll;
const int Maxn=3e3+50;
const ll Mod=786433;
bool IsNotPrime[Maxn];
int Prime[Maxn],PrimeCnt;
ll p[Maxn][Maxn],c[Maxn][Maxn];
int n,n1,n2,m,T,Q;
inline int read()
{
    char ch=getchar();
    int i=0,f=1;
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){i=(i<<3)+(i<<1)+ch-'0';ch=getchar();}
    return i*f;
}
inline void pre()
{
    IsNotPrime[1]=1;
    for(int i=2;i<Maxn;i++)
    {
        if(!IsNotPrime[i])
        {
            Prime[++PrimeCnt]=i;
        }
        for(int j=1;j<=PrimeCnt&&i*Prime[j]<Maxn;j++)
        {
            IsNotPrime[i*Prime[j]]=1;
            if(i%Prime[j]==0)break;
        }
    }
    for(int i=0;i<Maxn;i++)
    {
        c[i][0]=1;
    }
    for(int i=1;i<Maxn;i++)
        for(int j=1;j<Maxn;j++)
        {
            c[i][j]=(c[i-1][j-1]%Mod+c[i-1][j]%Mod)%Mod;
        }
}
inline void Striling()
{
    p[0][0]=1;
    for(int i=1;i<Maxn;i++)
        for(int j=1;j<Maxn;j++)
        p[i][j]=(p[i-1][j-1]%Mod+1LL*p[i-1][j]%Mod*j)%Mod;
}
inline ll calc1(int n1,int i)
{
    if(i==n1) return 1;
    if(!IsNotPrime[i])
    {
        return p[n1][i]%Mod;
    }
    else return c[n1][i-1]%Mod;
}
inline ll calc2(int n2,int i)
{
    return (c[n2-1][i-1]%Mod*c[m][i]%Mod)%Mod;
}
int main()
{
    pre();
    Striling();
    T=read();
    while(T--)
    {
        n1=read(),n2=read(),m=read(),Q=read();
        while(Q--)
        {
            ll ans=0;
            n=read();
            for(int i=1;i<n;i++)
            {
                if(i>n1||n-i>n2)continue;
               (ans+=(calc1(n1,i)%Mod*calc2(n2,n-i)%Mod)%Mod)%=Mod;
            }
            cout<<ans%Mod<<endl;
        }
    }
}


正解（100）：

 

第二类斯特林数的大数求法：

F[i]=1i!∑k=0i(−1)kCki∗(i−k)n


然后转化一下：

F[i]=∑k=0i(−1)k(i−k)nk!(i−k)!

 

F[i]=∑k=0i(−1)kk!×(i−k)n(i−k)!

 

正解代码：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
const int Mod=786433;
const int G=13;
const int Maxn=1e6+50;


inline int read()
{
	char ch=getchar();
	int i=0,f=1;
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){i=(i<<3)+(i<<1)+ch-'0';ch=getchar();}
	return i*f;
}


ll A[Maxn],B[Maxn],C[Maxn],Alice[Maxn],Bob[Maxn],Ans[Maxn],fac[Maxn],inv[Maxn];
int n1,n2,m,T,Q;
bool IsNotPrime[Maxn];
int Prime[Maxn],Primecnt;
inline ll ksm(ll a,ll b)
{
	ll tmp=1;
	while(b)
	{
		if(b&1)tmp=(tmp*a)%Mod;
		a=(a*a)%Mod;
		b>>=1;
	}
	return tmp;
}
struct NTT
{
	ll pos[Maxn],k;
	inline void Init(int len)
	{
		for(k=1;k<len+2000;k<<=1);//数据会卡k<<1
		for(ll i=1;i<k;i++)
		pos[i]=(i&1)?((pos[i>>1]>>1)^(k>>1)):(pos[i>>1]>>1);
	}
	inline void ntt(ll *a)
	{
		for(int i=1;i<k;i++)
			if(i<pos[i])swap(a[i],a[pos[i]]);
		for(int m1=1,m2;m1<k;m1<<=1)
		{
			m2=m1<<1;
			ll wn=ksm(G,(Mod-1)/m2);
			for(int i=0;i<k;i+=m2)
			{
				ll w=1;
				for(int j=0;j<m1;j++)
				{
					ll &A1=a[i+j],&B1=a[i+j+m1],t=1LL*w*B1%Mod;
					B1=(A1-t+Mod)%Mod;
					A1=(A1+t)%Mod;
					w=(w*wn)%Mod;
				}
			}
		}
	}
	inline void muitiply(ll *a,ll *b,ll *c)
	{
		
		ntt(a),ntt(b);
		for(ll i=0;i<k;i++)c[i]=(a[i]*b[i])%Mod;
		reverse(c+1,c+k);
		ntt(c);
		ll Inv=ksm(k,Mod-2);
		for(ll i=0;i<k;i++)c[i]=(1LL*Inv*c[i])%Mod;
	}
}ntt;
inline void Pre()
{
	IsNotPrime[1]=1;
	for(int i=2;i<Maxn;i++)
	{
		if(!IsNotPrime[i])
		{
			Prime[++Primecnt]=i;
		}
		for(int j=1;j<=Primecnt&&i*Prime[j]<Maxn;j++)
		{
			IsNotPrime[i*Prime[j]]=1;
			if(i%Prime[j]==0)break;
		}
	}
	fac[0]=inv[0]=1;
	for(int i=1;i<Maxn;i++)fac[i]=fac[i-1]*i%Mod;
	for(int i=1;i<Maxn;i++)inv[i]=inv[i-1]*ksm(i,Mod-2)%Mod;
}
int main()
{
	//freopen("lx.in","r",stdin);
	Pre();
	scanf("%d",&T);
	while(T--)
	{
		memset(A,0,sizeof(A));
		memset(B,0,sizeof(B));
		scanf("%d%d%d%d",&n1,&n2,&m,&Q);
		ntt.Init(n1+n2);
		A[0]=1;
		for(int i=1;i<=n1;i++)A[i]=A[i-1]*i%Mod;
		for(int i=0;i<=n1;i++)A[i]=B[i]=ksm(A[i],Mod-2);
		for(int i=1;i<=n1;i+=2)A[i]=A[i]*(Mod-1)%Mod;
		for(int i=0;i<=n1;i++)B[i]=B[i]*ksm(i,n1)%Mod;
		ntt.muitiply(A,B,C);
		memset(Alice,0,sizeof(Alice));
		memset(Bob,0,sizeof(Bob));
		memset(Ans,0,sizeof(Ans));
		for(int i=1;i<=n1;i++)
		{
			if(!IsNotPrime[i])Alice[i]=C[i];
			else if(i==n1)Alice[i]=1;
			else Alice[i]=fac[n1]*inv[i-1]%Mod*inv[n1-i+1]%Mod;
		}
		for(int i=1;i<=n2;i++)
			Bob[i]=fac[m]*inv[i]%Mod*inv[m-i]%Mod*fac[n2-1]*inv[i-1]%Mod*inv[n2-i]%Mod;
		ntt.muitiply(Alice,Bob,Ans);
		while(Q--)
		{
			int x;
			scanf("%d",&x);
			printf("%lld\n",Ans[x]);
		}
	}
}





DZYO
关注
  • 0
    点赞
  • 踩
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
david harvey NTT
11-18
We show how to improve the efficiency of the computation of fast Fourier transforms over F_p where p is a word-sized prime. Our main technique is optimisation of the basic arithmetic, in effect ...
【组合数学】第二类斯特林数
BlessingXRY的博客
01-20 2404
一、定义 第二类Stirling数即:，又可记为[与第一类的表示有大小写的区别]。其表示将n个不同的元素分成m个集合的方案数。 二、理解关键词句 1.集合的一个拆分(表示将n个不同的元素拆分成m个集合的方案数) 2.dp[n][m] = m*dp[n-1][m] + dp[n-1][m-1] (dp[n][m]表示n个元素划分为m个集合的方案数) 2.1对于第n个元素，如果前面的n-1
第二类Stirling数（第二类斯特林数）
ZSJZ_liuzian的博客
01-28 1万+
第二类Stirling数（第二类斯特林数） 定义 第二类Stirling数表示把nnn个不同的数划分为mmm个集合的方案数，要求不能为空集，写作S(n,m)S(n,m)S(n,m). 和第一类Stirling数不同，划分集合不必考虑排列次序。 递推式 考虑S(n,m)S(n,m)S(n,m)可以由什么转移得到？ 1、S(n−1,m−1)S(n-1,m-1)S(n−1,m−1)，将n−1n...
第二类斯特林数
fdxgcw的博客
08-22 1155
{n,k}表示将n个两两不同的元素划分成k个非空子集的方案数 注意:不需要排序，因为集合无序 假设已知{n-1,k-1}和{n-1,k} 同样的，有两种情况 1:n单独作为集合，结果为{n-1,k-1} 2:n放入之前的集合，结果为(k){n-1,k} 性质: {n,0}=0 {n,1}=1 {n,n}=1 {n,2}=2^(n-1)-1 {n,n-1}=C(n,2) {n,n-2}=C(n,3)+3C(n,4) 第二类斯特林数有通项公式: {n,m}=(1/m!)*(k从0累加到m)(-1)^k * C(
组合数学 —— 斯特林数（Stirling）
热门推荐
Alex_McAvoy的博客
04-25 2万+
【第一类斯特林数】 1.定理 第一类斯特林数 S1(n,m) 表示的是将 n 个不同元素构成 m 个圆排列的数目。 2.递推式 设人被标上1,2,.....p，则将这 p 个人排成 m个圆有两种情况： 在一个圆圈里只有标号为 p 的人自己，排法有 S1(n-1,m-1) 个。 p 至少和另一个人在一个圆圈里。 这些排法通过把 1,2....n-1 排成 m 个圆再把 n 放在 1,2...
第一类和第二类斯特林数
weixin_30677073的博客
03-22 283
第一类斯特林数 第一类斯特林数定义如下： \(s_1(n,k)\)表示\(n\)个元素组成\(k\)个圆排列的方案数。 其中\(n\)个元素的圆排列定义为\(n\)个元素围成一圈的排列，两个圆排列本质相同当且仅当两个圆排列以任意方式旋转之后相同。 那么我们可以得到第一类斯特林数的递推公式： \[ s_1(n,k)=s_1(n-1,k-1)+(n-1)\cdot s_1(n-1,k) \] 这个很...
NTT.rar_NTT_NTT c++_NTT%20c%2B%2B_nttc++_快速数论变换
09-20
快速数论变换（Fast Number Theoretic Transform, NTT）是一种在数论背景下进行的高效算法，主要用于在有限域上进行大规模的线性运算，如乘法、卷积等。与快速傅里叶变换（FFT）类似，NTT在计算速度上具有显著优势，...
任意模数NTT的代码
08-22
这是任意模数NTT的代码，非常地实用。我的代码比较好理解，希望能够和大家分享。
日本NTT网络中SDN技术应用实践
01-19
讨论了SDN/NFV技术在通信领域的应用情况，重点分析了日本运营商NTT在此领域的探索和最新进展，简要对比了海外市场与国内市场商业应用的情况，初步给出了国内市场发展SDN/NFV的基本建议。
NTTDATA订制就业培养对日软件外包人才需求信息.doc
11-21
NTTDATA 订制就业培养对日软件外包人才需求信息 本文档主要介绍了 NTTDATA 订制就业培养对日软件外包人才需求信息，旨在满足重庆 NTTDATA 对日软件外包人才需求的需求。该培养项目旨在培养日本企业模式规范化治理的...
第二类斯特林数总结
weixin_30855099的博客
04-23 383
最主要的两个式子： 套路1： $$\begin{array}{l}x^k=\sum_{i=0}^x\begin{pmatrix}x\\i\end{pmatrix}\begin{Bmatrix}k\\i\end{Bmatrix}i!\\\end{array}$$ 左边的式子可以看成将k个球放到x个有标号的盒子中，总共的方案数。 右边的式子可以看成先选出i个非空的盒子，将k个球放到这i个盒子中...
两类斯特林数的整理
weixin_33910460的博客
06-21 392
定义 第一类斯特林数（又称轮换数，by zzh），表示将\(n\)个不同的数分成\(k\)个无标号圆排列的方案数，用\(\begin{bmatrix}n\\k\end{bmatrix}\)表示，根据定义有如下递推式： \[\begin{bmatrix}n\\k\end{bmatrix}=\begin{bmatrix}n-1\\k-1\end{bmatrix}+(n-1)\begin{bmatrix...
第二类斯特林数模板（高精度）
weixin_42754727的博客
03-08 753
第二类斯特林数：将n个不同元素分成m个集合（n个不同的小球放进m个相同的盒字的方案数） 将n个元素的集合定义m个等价类的数目记做S(n,m)S(n,m)即第二类Stirling数。 s[n][m]=s[n-1][m-1]+m*s[n-1][m] 考虑第n个元素：新开辟一个集合，或者进入已经开辟的集合 #include &amp;amp;amp;lt;bits/stdc++.h&amp;amp;amp;gt; using namespace st...
FFT/NTT高精度
最新发布
08-22
对于使用FFT（快速傅里叶变换）或NTT（快速数论变换）来实现高精度计算，可以采以下步骤： 1. 将数字表示为多项式形式：将需要进行高精度计算的数字转换为多项式的系数形式。例如，将一个整数表示为多项式的系数。 ...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值