牛客NC21313美丽序列

题目链接

美丽序列 (nowcoder.com)

题目描述

牛牛喜欢整数序列，他认为一个序列美丽的定义是：
1：每个数都在0到40之间
2：每个数都小于等于之前的数的平均值
具体地说：for each i, 1 <= i < N,  A[i] <= (A[0] + A[1] + ... + A[i-1]) / i.
3：没有三个连续的递减的数
现在给你一个序列，每个元素是-1到40，你可以将序列中的-1修改成任意的数，求你可以得到多少个美丽序列，答案对1e9+7取模。

输入描述

第一行输入一个整数n(1<=n<=40)

第二行输入n个整数

输出描述

输出一个整数

例1

输入

2

3 -1

输出

4

例2

输入

3

5 3 -1

输出

2

例3

输入

3

-1 0 40

输出

0

例4

输入

11

-1 40 -1 -1 -1 10 -1 -1 -1 21 -1

输出

579347890

备注

子任务1：n<=10 子任务2：n<=20 子任务3：无限制

思路

DP解法：每增一个数，就遍历前数对比大小，再依据大小和前数对于前前数的增减性对自身选择的状态转移方程选择做判断

AC代码

#include<bits/stdc++.h>
using namespace std;
#define inf 1000000007//取模数
int n,a[41],dp[41][41][2][1601];//dp[i][j][k][s]:遍历到i时选择j做第i位,k判断升降(0时表示相对前一个数递增或相等,1时相对前一个数递减),s为前i个数之和.
int main(){                     //dp[i][j][k][s]的值的意义就是讨论到前i个数，且第i个数取j,且j的状态为k,前i个数总和为s是的方案数
    //输入数据
    cin>>n;
    for(int i=1;i<=n;i++)cin>>a[i];//存储n个数的值
    //dp初始化
    if(a[1]==-1)for(int i=0;i<=40;i++)dp[1][i][0][i]=1;//可变则全赋值为1
    else dp[1][a[1]][0][a[1]]=1;//不可变赋值自身为1
    //推导(状态转移)
    for(int i=2;i<=n;i++){//i为当前处理位
        //a[i]为'-1'可以变动
        if(a[i]==-1){
            for(int j=0;j<=40;j++){//j为当前处理数
                for(int k=0;k<=40;k++){//枚举前数k来和j作比较
                    for(int s=j*(i-1);s<=40*(i-1);s++){//根据“条件2.每个数都小于等于前数的平均数”可以得出i-1位和的范围在[j*(i-1),40*(i-1)]之间(ps:与第4行的s不同，此s指前i-1个数据的和,都用s避免看乱导致思路卡壳)
                        //递增时：即第i个数j大于等于第i-1个数k时
                        if(j>=k){//dp[i][j][0][s+j]表示此时结果存递增状态
                            dp[i][j][0][s+j]=(dp[i][j][0][s+j]+dp[i-1][k][0][s])%inf;//前增：dp[i-1][k][0][s]表示第i-1位且值为处于递增状态
                            dp[i][j][0][s+j]=(dp[i][j][0][s+j]+dp[i-1][k][1][s])%inf;//前减：dp[i-1][k][1][s]表示第i-1位且值为处于递减状态
                        }//因为第i位处于递增状态,所以第i-1位处于递增递减状态都需转移状态
                        //递减时：即第i个数j小于第i-1个数k时
                        else{//dp[i][j][1][s+j]表示此时结果存递减状态
                            dp[i][j][1][s+j]=(dp[i][j][1][s+j]+dp[i-1][k][0][s])%inf;//理解同21行注释
                        }//因为第i位处于递减状态,所以只能转移i-1位的递增状态
                    }
                }
            }
        }
        //a[i]不为'-1'不可以变动,此情况就少了第i位数j的遍历
        else{
            for(int k=0;k<=40;k++){
                for(int s=(i-1)*a[i];s<=40*(i-1);s++){
                    if(a[i]>=k){//有所不同的是上述的j改成了已知的a[i]
                        dp[i][a[i]][0][s+a[i]]=(dp[i][a[i]][0][s+a[i]]+dp[i-1][k][0][s])%inf;//前增后增
                        dp[i][a[i]][0][s+a[i]]=(dp[i][a[i]][0][s+a[i]]+dp[i-1][k][1][s])%inf;//前减后增
                    }
                    else{
                        dp[i][a[i]][1][s+a[i]]=(dp[i][a[i]][1][s+a[i]]+dp[i-1][k][0][s])%inf;//后减前必增
                    }
                }
            }
        }
    }
    //统计方案（结果的表示）
    int sum=0;
    for(int j=0;j<=40;j++){//根据第5行注释：将所有第n位的状态加起来即可，即第n位所有取值为j,k的增减态,前n位数和为s的所有情况
        for(int s=j*n;s<=1600;s++){//j*n(换1也对):最后一个取j时,因为第j个数小于等于前面所有数的平均数,所以可以从j*n开始索引.
            sum=(sum+dp[n][j][0][s])%inf;
            sum=(sum+dp[n][j][1][s])%inf;
        }
    }
    cout<<sum;//输出结果
}