题意:
牛牛喜欢整数序列,他认为一个序列美丽的定义是
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
思路:
对于这一题目,我们最先需要知道的是,这是个动态规划的题,对于dp数组,我们最先考虑到第一维可以是第几个数,第二维是当前数为多少。又因为题目中还有两个条件,一是没有三个连续的递减的数,二是每个数都小于等于之前的数的平均值。所以我们可以考虑第三维是当前数位于递减序列中的第几个,四维是总和是多少。对此我们可以得出dp[i][j][l][k]表示到了第i个数值是j位于递减序列中的第l个总和为k的数列有多少,这样我们再通过分类讨论与动态规划得出转移方程,最后得出答案。
分类讨论及其转移方程的思路分析:
对于最开始时,如果第一个数就为-1,我们可以从0循环到40,令dp[1][i][1][i]=1,因为-1可以变为任意一个数,所以第一项有41种可能的数字,总和也有41种可能,都为0到40。如果第一项不是-1,就直接dp[1][a[1]][1][a[1]]=1即可。然后从数组的第二项开始循环,如果当前数字为-1,当前数字可以为0到40一共41种可能,又因为当前数字的前一个数字我们不知道是多少,所以也可以理解为0到40一共41种可能,最后我们需要遍历总和,因为要保证每个数都小于等于之前的数的平均值,所以需要遍历j*(i-1)到1600-j,如果当前数字大于上一个数字,就有dp[i][j][1][k+j]=(dp[i][j][1][k+j]+dp[i-1][l][1][k]+dp[i-1][l][2][k])%mod,否则就有dp[i][j][2][k+j]=(dp[i][j][2][k+j]+dp[i-1][l][1][k])%mod。如果当前数字不为-1,我们可以根据为-1时的讨论,省去对当前数字的讨论,同时遍历总和时为a[i]*(i-1)到1600-a[i],这是如果当前数字大于上一个数字,就有dp[i][a[i]][1][k+a[i]]=(dp[i][a[i]][1][k+a[i]]+dp[i-1][l][1][k]+dp[i-1][l][2][k])%mod;,否则就有dp[i][a[i]][2][k+a[i]]=(dp[i][a[i]][2][k+a[i]]+dp[i-1][l][1][k])%mod。上代码再分析一下:
//这里为对第一个数字是否为-1进行讨论
if(a[1]==-1)
{
for(ll i=0;i<=40;i++)
dp[1][i][1][i]=1;
}
else
dp[1][a[1]][1][a[1]]=1;
//这里为在循环中,如果当前数字为-1进行的操作
if(a[i]==-1)
{
for(ll j=0;j<=40;j++) //这里循环的是当前数字,-1可以随便变数,所以为0到40的循环
{
for(ll l=0;l<=40;l++) //这里为当前数字的前一个数字进行0到40的循环
{
for(ll k=j*(i-1);k<=1600-j;k++) //这里是对总和的遍历
{
if(j>=l) //如果为当前数大于等于前一个数,则当前数为递减序列的第一个,上一个数可以为递减序列的第一个或是第二个。
dp[i][j][1][k+j]=