hiho一下 第240周 出勤记录II

题面 :

Link

题意:

长度为n的只含有O,A,L序列,满足不能有连续三个L,最多一个A的要求的序列数量?

思路:

基础dp,设dp[i][j][k][l]表示长度为i序列;j用0,1,2来区分结尾为O,L,A;k用0,1,2表示连续出现几次L;l用0,1来区分A是否出现过。状态转移方程就很容易写出。貌似只用三维就可以表示,我写复杂呢…

代码

#include
using namespace std;
typedef long long ll;
const int mod = 1e9+7;
typedef long long ll;
ll dp[100010][3][3][2];//dp[i][j][k][l] 表示长度为i,j为O,L,A.k为连续几次L,l为几次A.
ll ans[100010];

void pre(){
   memset(dp,0,sizeof(dp));
   dp[1][0][0][0]=1;dp[1][1][1][0]=1;dp[1][2][0][1]=1;
   ans[1] = 3;
   for(int i=2;i<=100000;i++){
     dp[i][0][0][0]=(dp[i-1][0][0][0]+dp[i-1][1][1][0]+dp[i-1][1][2][0])%mod;
     dp[i][0][0][1]=(dp[i-1][0][0][1]+dp[i-1][1][1][1]+dp[i-1][1][2][1]+dp[i-1][2][0][1])%mod;
     dp[i][1][1][0]=(dp[i-1][0][0][0])%mod;
     dp[i][1][1][1]=(dp[i-1][0][0][1]+dp[i-1][2][0][1])%mod;
     dp[i][1][2][0]=(dp[i-1][1][1][0])%mod;
     dp[i][1][2][1]=(dp[i-1][1][1][1])%mod;
     dp[i][2][0][1]=(dp[i-1][0][0][0]+dp[i-1][1][1][0]+dp[i-1][1][2][0])%mod;
     ans[i]=(dp[i][0][0][0]+dp[i][0][0][1]+dp[i][1][1][0]+dp[i][1][1][1]+dp[i][1][2][0]+dp[i][1][2][1]+dp[i][2][0][1])%mod;
   }
}

int main(){
   int n;
   pre();
   while(~scanf("%d",&n)){
      printf("%d\n",ans[n]);
   }
}

你可能感兴趣的:(动态规划基础dp,每周一题(,hihocode,leetcode))