BZOJ 1925 地精部落(DP)

一道很经典的DP题。

题意:求n排列中波动排列的种数。

不妨考虑DP，令dp1[i][j],表示1-j的排列中，第一项为i之后递增的波动排列种数。dp2[i][j]表示1-j的排列中，第一项为i之后递减的波动排列种数。

显然有一个性质，dp1[i][j]=dp2[j+1-i][j],将各项用j+1减去即可。

所以我们主要观察dp1数组。

如果第一项放了i，之后的数字是1,2，，，i-1，i+1,i+2,,j。

如果我们把大于i的数减去1，就又变成了j-1的一个排列，那么则有dp1[i][j]=sum(dp2[i][j-1]+dp2[i+1][j-1]+...dp2[j-1][j-1]).

因为dp2[i][j]=dp1[j+1-i][j],所以dp1[i][j]=sum(dp1[1][j-1]+dp1[2][j-1]+....+dp1[j-i][j-1]).

可以用前缀和优化转移。用滚动数组优化空间。时间复杂度O(n^2).

# include <cstdio>

# include <cstring>

# include <cstdlib>

# include <iostream>

# include <vector>

# include <queue>

# include <stack>

# include <map>

# include <bitset>

# include <set>

# include <cmath>

# include <algorithm>

using namespace std;

# define lowbit(x) ((x)&(-x))

# define pi acos(-1.0)

# define eps 1e-

# define MOD

# define INF

# define mem(a,b) memset(a,b,sizeof(a))

# define FOR(i,a,n) for(int i=a; i<=n; ++i)

# define FO(i,a,n) for(int i=a; i<n; ++i)

# define bug puts("H");

# define lch p<<,l,mid

# define rch p<<|,mid+,r

# define mp make_pair

# define pb push_back

typedef pair<int,int> PII;

typedef vector<int> VI;

# pragma comment(linker, "/STACK:1024000000,1024000000")

typedef long long LL;

int Scan() {

    int x=,f=;char ch=getchar();

    while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}

    while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}

    return x*f;

}

const int N=;

//Code begin...

int dp[][N], sum[N];

int main ()

{

    int n, P, flag=;

    scanf("%d%d",&n,&P);

    dp[][]=; sum[]=;

    for (int i=n-; i>=; --i) {

        flag^=; mem(dp[flag],);

        FOR(j,,n-i+) if (n-i-j>=) dp[flag][j]=sum[n-i-j+];

        mem(sum,);

        FOR(j,,n-i+) sum[j]=(sum[j-]+dp[flag][j])%P;

    }

    LL ans=;

    FOR(i,,n) ans=(ans+dp[flag][i])%P;

    ans=ans*%P;

    printf("%lld\n",ans);

    return ;

}