【BZOJ1925】地精部落(SDOI2010)-DP+滚动数组

本文介绍了一种利用动态规划与滚动数组技术解决特定数学问题的方法——求解长度为n的波动排列数量。文章详细阐述了算法原理,包括避免使用逆元以减少计算复杂度、使用组合数进行状态转移等,并提供了完整的代码实现。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

测试地址:地精部落
做法:本题需要用到DP+滚动数组。
注意到题目问的就是长为 n n 的波动排列数目,但是由于p不一定是质数,所以我们要避免使用逆元,又因为空间只有64MB,我们需要严格控制空间的使用。
考虑如何计算长为 i i 的波动排列数目,我们可以枚举i的位置 j j ,因为i一定是最大的元素,所以它的两边一定是谷,因此我们就要求长为 j1 j − 1 ij i − j 的第一位或最后一位是谷的方案数,然后乘上一个 Cj1i1 C i − 1 j − 1 累加入答案即可。显然组合数可以通过 Cji=Cj1i1+Cji1 C i j = C i − 1 j − 1 + C i − 1 j 这个递推式滚动计算。
问题是我们怎么计算上面那些带限制的方案数呢?注意到一个性质:波动排列的第一位是峰或谷的方案数是一样的。为什么呢?因为显然每一个第一位是峰的方案,都可以通过所有元素 i i 变为ni+1唯一对应一个第一位是谷的方案。所以我们定义 f(i) f ( i ) 为第一位是谷的方案,那么我们可以得到状态转移方程:
2f(i)=ij=1Cj1i1f(j1)f(ij) 2 f ( i ) = ∑ j = 1 i C i − 1 j − 1 f ( j − 1 ) f ( i − j )
注意到 f(i) f ( i ) 有个 2 2 的系数,然而上面我们已经说了要尽量避免逆元的计算,所以根据定义,我们只计算j是偶数的情况,即:
f(i)=ij=1[j]Cj1i1f(j1)f(ij) f ( i ) = ∑ j = 1 i [ j 是 偶 数 ] C i − 1 j − 1 f ( j − 1 ) f ( i − j )
特殊地, f(0)=f(1)=1 f ( 0 ) = f ( 1 ) = 1 ,那么这道题的答案就是 2f(n) 2 f ( n ) ,这样我们就得到了一个 O(n2) O ( n 2 ) 的做法,可以通过此题。
以下是本人代码:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int n;
ll p,f[5010]={0},C[2][5010]={0};

int main()
{
    scanf("%d%lld",&n,&p);

    f[0]=f[1]=1;
    C[0][0]=1;
    int past=0,now=1;
    for(int i=2;i<=n;i++)
    {
        C[now][0]=1;
        for(int j=1;j<=i-1;j++)
            C[now][j]=(C[past][j-1]+C[past][j])%p;
        for(int j=2;j<=i;j+=2)
            f[i]=(f[i]+C[now][j-1]*f[j-1]%p*f[i-j])%p;
        swap(now,past);
    }
    printf("%lld",(f[n]<<1)%p);

    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值