洛谷传送门
BZOJ传送门
题目描述
称一个 1 , 2 , . . . , N 1,2,...,N 1,2,...,N的排列 P 1 , P 2 . . . , P n P_1,P_2...,P_n P1,P2...,Pn是Magic的,当且仅当 2 ≤ i ≤ N 2\le i\le N 2≤i≤N时, P i > P i / 2 P_i>P_{i/2} Pi>Pi/2. 计算 1 1 1, 2 2 2,… N N N的排列中有多少是Magic的,答案可能很大,只能输出模P以后的值
输入输出格式
输入格式:
输入文件的第一行包含两个整数 n n n和 p p p,含义如上所述。
输出格式:
输出文件中仅包含一个整数,表示计算 1 , 2 , ⋯ , N 1,2,⋯, N 1,2,⋯,N的排列中, Magic排列的个数模 p p p 的值。
输入输出样例
输入样例#1:
20 23
输出样例#1:
16
说明
100%的数据中, 1 ≤ N ≤ 1 0 6 , P ≤ 1 0 9 1 ≤N ≤ 10^6, P≤ 10^9 1≤N≤106,P≤109, p p p是一个质数。
解题分析
还是先建出树来, 发现一个节点最多有两个儿子, 那么直接有 d p [ i ] = d p [ l s o n ] ∗ d p [ r s o n ] ∗ ( s i z [ i ] − 1 s i z [ l s o n ] ) dp[i]=dp[lson]*dp[rson]*\binom{siz[i]-1}{siz[lson]} dp[i]=dp[lson]∗dp[rson]∗(siz[lson]siz[i]−1)。 于是一个树 D P DP DP就 A A 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]);
}