一个长为 n n n 的序列,由 H , T H,T H,T 构成,每个位置都可以是 T T T 或者 H H H,那么总共有 2 n 2^n 2n 这样的序列,问这些序列中有多少个序列其中包含至少 k k k 个连续的 H H H?
类似 ZOJ 3747 Attack on Titans 带限制的递推DP ,将至少转化为至多: { \{ {至多 n n n 个 H } − { H\}-\{ H}−{至多 k − 1 k-1 k−1 个 H } H\} H}
令 d p [ i ] [ 0 ] , d p [ i ] [ 1 ] dp[i][0],dp[i][1] dp[i][0],dp[i][1] 表示位置 i i i 放 H H H 或 T T T 对应的序列数,令 s u m = d p [ i − 1 ] [ 0 ] + d p [ i − 1 ] [ 1 ] sum=dp[i-1][0]+dp[i-1][1] sum=dp[i−1][0]+dp[i−1][1], 则对于 { \{ {至多 k k k 个 H } H\} H} 而言:
(1) i ≤ k i\le k i≤k ,尚未达到限制:
d p [ i ] [ 0 ] = s u m dp[i][0]=sum dp[i][0]=sum
(2) i = k + 1 i=k+1 i=k+1,需要减去前一段全是 H H H 的情况:
d p [ i ] [ 0 ] = s u m − 1 dp[i][0]=sum-1 dp[i][0]=sum−1
(3) i > k + 1 i>k+1 i>k+1,需要减去 i − k ∼ i − 1 i-k\sim i-1 i−k∼i−1 全是 H H H 的情况:
d p [ i ] [ 0 ] = s u m − d p [ i − k − 1 ] [ 1 ] dp[i][0]=sum-dp[i-k-1][1] dp[i][0]=sum−dp[i−k−1][1]
T T T 的数量并没有限制,因此 d p [ i ] [ 1 ] = s u m dp[i][1]=sum dp[i][1]=sum
还要用高精度。
代码如下:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#define MAXN 105
using namespace std;
struct BigInt{
int a[100],MOD=10000,len;
BigInt(){
memset(a,0,sizeof(a));len=1;
}
BigInt(int b){
memset(a,0,sizeof(a));len=0;
while(b){
a[len++]=b%MOD;
b/=MOD;
}
}
BigInt(const BigInt &b){
memset(a,0,sizeof(a));
len=b.len;
for(int i=0;i<len;i++)a[i]=b.a[i];
}
BigInt operator=(const BigInt &b){
memset(a,0,sizeof(a));
len=b.len;
for(int i=0;i<len;i++)a[i]=b.a[i];
return *this;
}
BigInt operator+(const BigInt &b){
BigInt r;r.len=max(len,b.len);int up=0;
for(int i=0;i<r.len;i++){
int tmp=(i<len?a[i]:0)+(i<b.len?b.a[i]:0)+up;
r.a[i]=tmp%MOD;up=tmp/MOD;
}
if(up)r.a[r.len++]=up;
return r;
}
BigInt operator-(const BigInt &b){
BigInt r;r.len=max(len,b.len);int down=0;
bool flag=false;BigInt t1,t2;
if(*this<b)flag=true,t1=b,t2=*this;
else t1=*this,t2=b;
for(int i=0;i<t1.len;i++){
int tmp=t1.a[i]-(i<t2.len?t2.a[i]:0)-down;
r.a[i]=(tmp%MOD+MOD)%MOD;
if(tmp<0)down=1;else down=0;
}
while(!r.a[r.len-1]&&r.len>1)r.len--;
if(flag)r.a[r.len-1]=-r.a[r.len-1];
return r;
}
bool operator<(const BigInt &b){
if(len<b.len)return true;
else if(len>b.len)return false;
else{
int i=len-1;
while(i>=0){
if(a[i]!=b.a[i])
return a[i]<b.a[i];
i--;
}
return false;
}
}
void output(){
printf("%d",a[len-1]);
for(int i=len-2;i>=0;i--)
printf("%04d",a[i]);
printf("\n");
}
};
int n,k;
BigInt sum,dp[MAXN][2];
BigInt f(int k){
dp[0][0]=0;dp[0][1]=1;
for(int i=1;i<=n;i++){
sum=dp[i-1][0]+dp[i-1][1];
dp[i][1]=sum;
if(i<=k)dp[i][0]=sum;
else if(i==k+1)dp[i][0]=sum-1;
else dp[i][0]=sum-dp[i-k-1][1];
}
return dp[n][0]+dp[n][1];
}
int main(){
#ifdef WINE
freopen("data.in","r",stdin);
#endif
while(scanf("%d%d",&n,&k)!=EOF){
BigInt res=f(n)-f(k-1);
res.output();
}
return 0;
}