Codeforces 285E

最新推荐文章于 2019-12-10 23:06:45 发布

After0514

最新推荐文章于 2019-12-10 23:06:45 发布

阅读量384

点赞数

分类专栏： dp

本文链接：https://blog.csdn.net/wen_xp/article/details/53467181

版权

dp 专栏收录该内容

46 篇文章 0 订阅

订阅专栏

d p [i] [j] [k] 前 i 个 数 字 有 j 个 好 位 置 处 于 第 k 种 情 况 ⎧ ⎩ ⎨ ⎪ ⎪ ⎪ ⎪ ⎪ ⎪ ⎪ ⎪ ⎪ ⎪ ⎪ ⎪ ⎪ ⎪ ⎪ ⎪ 1. a [i] = i 2. a [i - 1] = i, a [i] = i - 1 3. a [i - 1] \neq i, a [i] = i - 1 4. a [i - 1] = i, a [i] \neq i - 1 5. a [i - 1] \neq i, a [i] \neq i - 1

$\begin{equation*} dp[i][j][k]\\ 前i个数字有j个好位置处于第k种情况\\ \left\{ \begin{array}{} 1.a[i]=i\\ \hline 2.a[i-1]=i,a[i]=i-1\\ \hline 3.a[i-1]\neq i,a[i]= i-1\\ \hline 4.a[i-1]=i,a[i]\neq i-1\\ \hline 5.a[i-1]\neq i,a[i]\neq i-1\\ \hline \end{array} \right. \end{equation*}$
可能顺序和代码的不一致，我也忘了

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int N=1e3+10;
typedef __int64 LL;
const int mod=1e9+7;
LL dp[N][N][5];
int xa,ya,za;
void fun(int x1,int y1,int z1,int co=1){
    if(x1<0||y1<0||z1<0)return ;
    dp[xa][ya][za]+=dp[x1][y1][z1]*co%mod;dp[xa][ya][za]%=mod;
}
void pushup(int i,int j,int k){
    xa=i,ya=j,za=k;
    if(k==0){
        for(int k1=0;k1<5;k1++){
            fun(i-1,j,k1);
        }
    }
    else if(k==1){
        fun(i-1,j-1,1);
        fun(i-1,j,2);
        fun(i-1,j-1,3);
        fun(i-1,j,4);
    }
    else if(k==2){
        for(int k1=0;k1<5;k1++){
            fun(i-2,j-2,k1);
        }
    }
    else if(k==3){
        fun(i-1,j,0,i-2-j);
        fun(i-1,j+1,0,j+1);

        fun(i-1,j,1,i-2-j);
        fun(i-1,j+1,1,j+1-1);

        fun(i-1,j,2,i-2-(j-1));
        fun(i-1,j+1,2,j+1-2);

        fun(i-1,j,3,i-2-j-1);
        fun(i-1,j+1,3,j+1);

        fun(i-1,j,4,i-2-(j-1)-1);
        fun(i-1,j+1,4,j+1-1);
    }
    else {
        fun(i-1,j,1);
        fun(i-1,j,2);
        fun(i-1,j-1,3);
        fun(i-1,j-1,4);
    }
}
int main(){
    #ifdef DouBi
    freopen("in.cpp","r",stdin);
    #endif // DouBi
    int n,k;
    while(scanf("%d%d",&n,&k)!=EOF){
        memset(dp,0,sizeof(dp));
        dp[1][0][0]=1;dp[0][0][0]=1;
        for(int i=2;i<=n;i++){
            for(int j=0;j<=n;j++){
                for(int k1=0;k1<5;k1++){
                    pushup(i,j,k1);
                   // printf("%d %d %d %d\n",i,j,k1,dp[i][j][k1]);
                }
            }
        }
        int sum=0;
        for(int k1=0;k1<5;k1++){
            sum+=dp[n][k][k1];sum%=mod;
        }
        printf("%d\n",sum);
    }
    return 0;
}