原题没有链接,比赛一直怼,怼不出来,我还是太弱了。
比赛时没有学lucas定理(c(n,m)=c(n%p,m%p)c(n/p,m/p)%p)真无奈。但实际上这一题也没用上。主要是费马小定理,费马小定理真是吼滴狠。
题目大意:在前进一步可以是1米或是2米,且不能3次连续走2米的情况下求在n+m步走2n+m米的的方案数;
分析:可以得知n+m步走2n+m米那么走2米的总次数和走1米的总次数分别就只能是n和m次,不能连续3次走两米那么可以把1米作为分隔符,有m个1米就有m+1个空位,假设2个2米连续的次数为a,单个2米的次数为b
那么n = 2a + b; 在m+1个空位中选a+b个空位,放置两种不同规格的物品,由2a <= n, b = n - 2a, a+b <= m+1可得a <= n/2 a >= n-m-1,可以推导出如下排列组合公式。
可以带入b = n - 2a继续化简,但难以解决分母里的阶乘,如果分母里有阶乘,在阶段计算时可能不整除,无法计算,就算能计算,时间复杂度为o(n),还有枚举a的问题o(n)。总时间复杂度o(n^2)无法接受。
使用lucas定理+费马小定理求大组合数a^(p-1) = 1(mod p); 等价于:
xxx * a^(p-2) (mod p) = xxx * 1/a(mod p)
利用组合数的费马小定理类推,于是可以把位于分母的阶乘挪到分子去。
经过逐步化简最终
注意到有个∑单次计算要乘o(n),不能容忍单次计算时间复杂度为o(n),于是可以提前打表,将1~m+1阶乘、值的p-2次方打表、值的阶乘的p-2次方、打表。快速幂时间复杂度为o(logn)那么最终打表时间复杂度为o(mlog(p-2)),有了表单次计算时间复杂度o(1),∑时间复杂度o(n/2-(n-m-1)),最终时间复杂度o(mlog(p-2)+n/2-(n-m-1))
1sec里过m,n<=1e5的数据是没问题了。
code:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e5 + 3;
const int mod = 1e9 + 7;
ll fval[maxn], valp_2[maxn], fvalp_2[maxn];
int n, m, up;
ll quick_pow(ll x, ll n)
{
ll mul = 1, dalta = x;
while (n)
{
if (n & 1)
mul = (mul * dalta) % mod;
dalta = (dalta * dalta) % mod;
n >>= 1;
}
return mul;
}
void inits()
{
up = m + 1;
fval[0] = valp_2[0] = fvalp_2[0] = 1;
for (int i = 1; i <= up; ++i)
{
fval[i] = fval[i - 1] * i % mod;
valp_2[i] = quick_pow(i, mod - 2);
fvalp_2[i] = fvalp_2[i - 1] * valp_2[i] % mod;
}
}
ll solve(int n, int m)
{
int L = n - m - 1, R = n / 2;
if (L > R)
return 0;
else
{
ll sum = 0;
for (int i = L; i <= R; ++i)
sum = (sum + (fvalp_2[i] * fvalp_2[n - 2 * i] % mod) * fvalp_2[m + 1 - n + i] % mod) % mod;
return sum * fval[m + 1] % mod;
}
}
int main()
{
scanf("%d %d", &n, &m);
inits();
printf("%lld\n", solve(n, m));
return 0;
}