动态规划 BZOJ 1925: [Sdoi2010]地精部落

动态规划 BZOJ 1925: [Sdoi2010]地精部落

Time Limit: 10 Sec Memory Limit: 64 MB
Submit: 1563 Solved: 980
[Submit][Status][Discuss]
Description

传说很久以前，大地上居住着一种神秘的生物：地精。 地精喜欢住在连绵不绝的山脉中。具体地说，一座长度为 N 的山脉 H可分 为从左到右的 N 段，每段有一个独一无二的高度 Hi，其中Hi是1到N 之间的正 整数。 如果一段山脉比所有与它相邻的山脉都高，则这段山脉是一个山峰。位于边 缘的山脉只有一段相邻的山脉，其他都有两段（即左边和右边）。 类似地，如果一段山脉比所有它相邻的山脉都低，则这段山脉是一个山谷。 地精们有一个共同的爱好——饮酒，酒馆可以设立在山谷之中。地精的酒馆 不论白天黑夜总是人声鼎沸，地精美酒的香味可以飘到方圆数里的地方。 地精还是一种非常警觉的生物，他们在每座山峰上都可以设立瞭望台，并轮 流担当瞭望工作，以确保在第一时间得知外敌的入侵。 地精们希望这N 段山脉每段都可以修建瞭望台或酒馆的其中之一，只有满足 这个条件的整座山脉才可能有地精居住。 现在你希望知道，长度为N 的可能有地精居住的山脉有多少种。两座山脉A 和B不同当且仅当存在一个 i，使得 Ai≠Bi。由于这个数目可能很大，你只对它 除以P的余数感兴趣。
Input
仅含一行，两个正整数 N, P。

Output
仅含一行，一个非负整数，表示你所求的答案对P取余 之后的结果。

Sample Input
4 7

Sample Output
3

HINT

对于 20%的数据，满足 N≤10；
对于 40%的数据，满足 N≤18；
对于 70%的数据，满足 N≤550；
对于 100%的数据，满足 3≤N≤4200，P≤109

Source
第一轮Day2

先简述题意吧：
求1到n的排列满足奇数项的高度比相邻位置都大, 要么满足偶数项的高度比相邻位置都大.的方案数。。。
数据范围是典型的n^2
所以可以用dp+滚动数组来做
首先我们考虑
对于任意一个1~i的排列，不断向排列中加入i+1,i+2,i+3,i+4……
我们可以算出，需要至少加多少个数，才能使其变成一个合法的排列
例如对于1 5 4 3 2这个排列
需要至少加两个数才能变成一个合法排列
合法的序列可能为1 5 4 6 3 7 2
于是我们定义一个排列与合法排列的距离为它变得合法至少需要添加的数的个数
记为len，则新加入的数可能会使len+1，len-1,或len不变
例如1 3 2 4 5距离为1
使得len+1的状况有
1 3 2 4 5 6（2）
1 3 6 2 4 5（2）
1 6 3 2 4 5（2）
使得len-1的状况有
1 3 2 6 4 5（0）
使得len不变的状况有
6 1 3 2 4 5（1）
1 3 2 4 6 5（1）
我们可以发现，对于一个距离不为0的排列，导致它不合法的原因是连续大于二个数递增或递减，而想要把它变成一个合法的排列，就需要用数把这样的递增或递减割裂开
所以对于一个距离为len的排列，有len个地方需要被割裂，也就是说把下一个数放进去
会生成len个距离为len-1的排列
我们继续考虑什么情况下加入下一个数之后，len保持不变
对于一个排列，如果开始的时候是递增，那么把新数加在最开始的位置不变
如果递减，那么把新数加在第一个和第二个之间的位置不变
对于结尾也是这样。。
所以对于一个排列把下一个数放进去会生成2个距离为len的排列
从而推出把下一个数放进去会生成多少个距离为len+1的排列
我们用dp[i][j]表示1~i的排列中距离为j的方案数
转移方程为
dp[i+1][j]+=2*dp[i][j]
dp[i+1][j-1]+=j*dp[i][j]
dp[i+1][j+1]+=(i-j-2)*dp[i][j]
滚动数组就好了
附一下代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<ctime>
using namespace std;
typedef long long ll;
ll a[5005];
ll b[5005];
int main ()
{
    ll n,mod;
    scanf("%lld%lld",&n,&mod);
    if(n==1) printf("%d",1%mod);
    if(n==2) printf("%d",2%mod);
    if(n==3) printf("%d",4%mod);
    a[1]=4;
    a[2]=2;
    for(int i=3;i<n;i++)
    {
        memset(b,0,sizeof(b));
        for(int j=1;j<=i;j++)
        {
            b[j]+=(ll)(a[j]*2)%mod;
            b[j]%=mod;
            b[j-1]+=(ll)(a[j]*((j-1)%mod))%mod;
            b[j-1]%=mod;
            b[j+1]+=(ll)(a[j]*((i-j)%mod))%mod;
            b[j+1]%=mod;
        }
        for(int j=1;j<=i;j++)
        {
            a[j]=b[j];
        }
    }
    printf("%lld",a[1]);
    return 0;
}