[CF] E. Camels

Posted 2021-03-24 baseai

CF 2000 的dp题目还是有点难qwq

题意：

一行有(n)个空位，每个空位可以填([1,4])的整数，要求：

1.有(t)个位置满足 (ai?1<ai>ai+1(1<i<n))

2.有(t?1)个位置满足 (ai?1>ai<ai+1)

的方案总数

题解：

设 (f[i][j][k][0/1]) 表示 (i)位置 当前数是 (j) 已经有 (k) 个满足条件一的位置 上升/下降趋势

转移：

[f[i][j][k][0]=sum f[i-1][l][k][0] + f[i-1][l][k][1] (1<=l<j)]

[f[i][j][k][1]=sum f[i-1][l][k][1] + f[i-1][l][k-1][0] (j<l<=n)]

当 (k = 0)时说明还没有顶峰，第二个转移的后面这种情况就是0

Code

#include<algorithm>
#include<iostream>
#include<cstdio>
#include<cmath>
using namespace std;
inline int read() {
    int x=0,f=1; char ch=getchar();
    while(ch<'0' || ch>'9') { if(ch=='-') f=-1; ch=getchar(); }
    while(ch>='0'&&ch<='9') { x=(x<<3)+(x<<1)+(ch^48); ch=getchar(); }
    return x * f;
}
const int N = 27;
int n,t;
int f[N][N][N][2];  //f[i][j][k][0/1] 表示 i 位置，j高度，k个峰顶，上升/下降 
int main()
{
    n = read(), t = read();
    f[2][4][0][0] = 3;
    f[2][3][0][0] = 2;
    f[2][2][0][0] = 1;
    for(int i=3;i<=n;++i)
        for(int k=0;k<=t;++k)
            for(int j=1;j<=4;++j)
                for(int l=1;l<=4;++l) {
                    if(l < j) f[i][j][k][0] += f[i-1][l][k][0] + f[i-1][l][k][1];   //增加了一个低谷，不改变k
                    if(l > j) f[i][j][k][1] += f[i-1][l][k][1] + (k>0 ? f[i-1][l][k-1][0] : 0);
//                  printf("i = %d %d
",i,f[i][j][k][1]);
                }
    int ans = 0;
    for(int i=1;i<=4;++i)
        ans += f[n][i][t][1];
    printf("%d",ans);
    return 0;
}