题目描述
小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;
}