有一个值初始为0,接下来n次你可以令其在之前基础上+2或+1或-1。你需要保证,这个值在整个过程中达到的最大值减去达到的最小值不大于k,求方案数,模1,000,000,007。
题解
首先考虑 O(nk2) O ( n k 2 ) 的算法,枚举上下界, dp[i][j] d p [ i ] [ j ] 表示第 i i 步值为的方案,有
dp[i][j]=dp[i−1][j−1]+dp[i−1][j−2]+dp[i−1][j+1],其中状态只能从在上下界中的状态转移而来
d
p
[
i
]
[
j
]
=
d
p
[
i
−
1
]
[
j
−
1
]
+
d
p
[
i
−
1
]
[
j
−
2
]
+
d
p
[
i
−
1
]
[
j
+
1
]
,
其
中
状
态
只
能
从
在
上
下
界
中
的
状
态
转
移
而
来
这样就可以在
O(n2)
O
(
n
2
)
的时间复杂度内求出一组上下界,不妨设起点是
k
k
,所以有组上下界
[l,r]
[
l
,
r
]
,每次统计答案的时候要删掉和上次的交集
[l,r−1]
[
l
,
r
−
1
]
。
不难发现,这是一个定点动区间的问题。然而如果我们把区间定住,这是一个有多个起点的问题,而每个起点都有贡献1,所以只要把初始状态 f[0][k]=1 f [ 0 ] [ k ] = 1 改成 f[0][i]=1,0<=i<=k f [ 0 ] [ i ] = 1 , 0 <= i <= k ,完美!妙啊!
代码
#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
#define maxn 100005
#define MAXN 1000005
#define mod 1000000007
typedef long long LL;
using namespace std;
int n,k;
LL f[5005][5005],ans;
LL work(int l,int r){
LL res=0;
for(int i=1;i<=n;i++){
for(int j=l;j<=r;j++){
f[i][j]=0;
if(j-1>=l && j-1<=r) f[i][j]+=f[i-1][j-1];
if(f[i][j]>=mod) f[i][j]-=mod;
if(j-2>=l && j-2<=r) f[i][j]+=f[i-1][j-2];
if(f[i][j]>=mod) f[i][j]-=mod;
if(j+1>=l && j+1<=r) f[i][j]+=f[i-1][j+1];
if(f[i][j]>=mod) f[i][j]-=mod;
}
}
for(int i=l;i<=r;i++){
res+=f[n][i];
if(res>=mod) res-=mod;
}
return res;
}
int main(){
scanf("%d%d",&n,&k);
for(int i=0;i<=k;i++) f[0][i]=1;
cout<<((work(0,k)-work(0,k-1))%mod+mod)%mod<<endl;
return 0;
}