[Luogu P2606] [BZOJ 2111] [ZJOI2010]排列计数

洛谷传送门

BZOJ传送门

题目描述

称一个$1,2,...,N$的排列$P_1,P_2...,P_n$是Magic的，当且仅当$2\le i\le N$时，$P_i>P_{i/2}$. 计算$1$，$2$，…$N$的排列中有多少是Magic的，答案可能很大，只能输出模P以后的值

输入输出格式

输入格式：

输入文件的第一行包含两个整数 $n$和 $p$，含义如上所述。

输出格式：

输出文件中仅包含一个整数，表示计算$1,2,\cdots ,N$的排列中， Magic排列的个数模 $p$ 的值。

输入输出样例

输入样例#1：

20 23 

输出样例#1：

16

说明

100%的数据中，$1\le N\le 10^6,P\le 10^9$，$p$是一个质数。

解题分析

还是先建出树来， 发现一个节点最多有两个儿子， 那么直接有$dp[i]=dp[lson]\ast dp[rson]\ast \left(\genfrac{}{}{0ex}{}{siz[i]-1}{siz[lson]}\right)$。 于是一个树$DP$就$A$了。

代码如下：

#include <cstdio>
#include <cstring>
#include <cmath>
#include <cctype>
#include <cstdlib>
#include <vector>
#include <algorithm>
#define R register
#define IN inline
#define W while
#define gc getchar()
#define ll long long
#define MX 1005000
int n, MOD, cnt;
std::vector <int> to[MX];
ll fac[MX], dp[MX], inv[MX];
int head[MX], siz[MX], deg[MX];
IN ll fpow(ll base, R int tim)
{
	ll ret = 1;
	W (tim)
	{
		if (tim & 1) ret = ret * base % MOD;
		base = base * base % MOD, tim >>= 1;
	}
	return ret;
}
void DFS(R int now)
{
	siz[now] = 1;
	for (R int i = to[now].size() - 1; ~i; --i)
	DFS(to[now][i]), siz[now] += siz[to[now][i]];
}
IN ll C(R int n, R int m)
{
	if (n < m) return 0;
	return fac[n] * inv[m] % MOD * inv[n - m] % MOD;
}
IN ll get(R int n, R int m)
{
	if (n < MOD && m < MOD) return C(n, m);
	return C(n % MOD, m % MOD) * get(n / MOD, m / MOD) % MOD;
}
int main(void)
{
	scanf("%d%d", &n, &MOD); fac[0] = inv[0] = 1;
	int bd = std::min(n, MOD);
	for (R int i = 2; i <= n; ++i) to[i >> 1].push_back(i);
	for (R int i = 1; i <= bd; ++i) fac[i] = fac[i - 1] * i % MOD;
	inv[0] = inv[1] = 1;
	for (R int i = 2; i <= bd; ++i) inv[i] = inv[MOD % i] * (MOD - MOD / i) % MOD;
	for (R int i = 2; i <= bd; ++i) inv[i] = inv[i - 1] * inv[i] % MOD;
	DFS(1);
	for (R int i = n; i; --i)
	{
		if (!to[i].size()) dp[i] = 1;
		else if (to[i].size() == 1) dp[i] = dp[to[i][0]];
		else dp[i] = dp[to[i][0]] * dp[to[i][1]] % MOD * get(siz[i] - 1, siz[to[i][0]]) % MOD;
	}
	printf("%lld", dp[1]);
}