积木

题目描述
小A正在搭积木。有N个位置可以让小A使用,初始高度都为0。小A每次搭积木的时候,都会选定一个拥有相同高度的区间[A..B],然后将位置[A+1..B-1]上的所有积木的高度加一。不幸的是,小A把积木搭好之后没多久,小A调皮的弟弟就将其中若干个位置上的积木弄倒了。小A想知道他原来的积木是如何摆放的,所以他求助于你,请你告诉他原来有多少种可能的摆法。

输入
第一行为一个正整数N,表示小A有N个位置。
第二行有N个由空格分隔的整数Hi,表示第i个位置的积木高度。-1表示这个位置上的积木已经被弄倒了。
输出
唯一的一行,输出包括可能的摆法mod 1,000,000,007的结果。
样例输入
3
-1 2 -1

3
-1 -1 -1

6
-1 -1 -1 2 -1 -1
样例输出
0

2

3

提示
【数据范围】
对于50%的数据 1<=N<=1000-1<=Hi<=1000
对于80%的数据 1<=N<=10000
对于100%的数据 1<=N<=20000-1<=Hi<=10000

显然相邻两列的高度差不会超过1,于是我们可以设计出dp方程,记dp[i,j]表示到第i列,高度为j的方案书,dp[i,j]=dp[i-1,j]+dp[i-1,j-1]+dp[i-1,j+1],如果有高度,那就只能从那个高度转移。开上滚动。

#include<iostream>
#include<cstdio>
#define MOD 1000000007
using namespace std;
int dp[2][10010],n,h[20010];
int M(int x){
    if (x<=n/2) return x-1;
    else return n-x;
}
int main(){
    scanf("%d",&n);
    scanf("%d",&h[1]);
    if (h[1]!=0&&h[1]!=-1){
        printf("0");
        return 0;
    }
    dp[1][0]=1;
    for (int i=2;i<=n;i++){
        scanf("%d",&h[i]);
        int i1=i%2,i2=(i-1)%2;
        for (int j=0;j<=M(i-2);j++) dp[i1][j]=0;
        if (h[i]==-1){
            dp[i1][0]=(dp[i2][0]+dp[i2][1])%MOD;
            for (int j=1;j<=M(i);j++){
                dp[i1][j]=((dp[i2][j]+dp[i2][j-1])%MOD+dp[i2][j+1])%MOD;
            }
        }else{
            if (h[i]==0) dp[i1][0]=(dp[i2][0]+dp[i2][1])%MOD;
            else dp[i1][h[i]]=((dp[i2][h[i]]+dp[i2][h[i]-1])%MOD+dp[i2][h[i]+1])%MOD;
        }
    }
    printf("%d",dp[n%2][0]);
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值