https://www.zybuluo.com/ysner/note/1327570
题面
称一个(1,2,...,n)的排列(p_1,p_2...,p_n)是(Magic)的,当且仅当(2leq ileq n)时,(p_i>p_{i/2})。
计算(1,2,...n)的排列中有多少是(Magic)的,答案可能很大,只需输出模(p)以后的值。
- (nleq10^6,pleq10^9)
解析
看到题目中的条件,如果你手打过二叉堆模板,就会想起小根堆。
其实就是小根堆。
设(f[i])表示长度为(i)的排列(小根堆)的答案。
考虑增量转移。
设小根堆根结点左子树大小为(ls)。
那么要从剩下的(i-1)个点中选(ls)个点组成左子树,有(C_{i-1}^{ls})种方法。
左右两个子树是规模更小的子问题,递归下去就行了。
那么(f[i]=C_{i-1}^{ls}f[ls]f[i-1-ls])
现在问题是怎么快速求(ls)。
在题目中,小根堆的构建方式是从上往下,一层层地从左往右排数。
那么先搞出大小为(i)的堆排满的层数(为(log_2(i+1))),剩下的几个数,随便算算即可。
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#define ll long long
#define re register
#define il inline
#define fp(i,a,b) for(re int i=a;i<=b;++i)
#define fq(i,a,b) for(re int i=a;i>=b;--i)
using namespace std;
const int N=1e6+100;
int T,n,k,jc[N],inv[N],f[N],mod,lim,lg[N];
il ll gi()
{
re ll x=0,t=1;
re char ch=getchar();
while(ch!='-'&&(ch<'0'||ch>'9')) ch=getchar();
if(ch=='-') t=-1,ch=getchar();
while(ch>='0'&&ch<='9') x=x*10+ch-48,ch=getchar();
return x*t;
}
il ll ksm(re ll S,re ll n)
{
re ll T=S;S=1;
while(n)
{
if(n&1) S=S*T%mod;
T=T*T%mod;
n>>=1;
}
return S;
}
il ll C(re ll x,re ll y)
{
if(x<0||y<0||x<y) return 0;
return 1ll*jc[x]*inv[y]%mod*inv[x-y]%mod;
}
il ll dfs(re int x)
{
if(~f[x]) return f[x];
re int ls=x+1-(1<<lg[x+1]);
if(ls<(1<<lg[x+1]-1)) ls+=(1<<lg[x+1]-1)-1;else ls=(1<<lg[x+1])-1;
return f[x]=C(x-1,ls)*dfs(ls)%mod*dfs(x-1-ls)%mod;
}
int main()
{
memset(f,-1,sizeof(f));f[0]=f[1]=1;
n=gi();mod=gi();
lim=min(n-1,mod-1);
jc[0]=1;fp(i,1,lim) jc[i]=1ll*jc[i-1]*i%mod;
inv[lim]=ksm(jc[lim],mod-2);
inv[0]=inv[1]=1;fq(i,lim-1,2) inv[i]=1ll*inv[i+1]*(i+1)%mod;
fp(i,2,n+1) lg[i]=lg[i>>1]+1;
printf("%lld
",dfs(n));
return 0;
}