URAL1513. Lemon Tale（dp）

1513

这题好久之前就看过了，悲催的是当时看题解都没看懂，今天又看了看so easy...

n个B里不能出现超过连续k个L的情况 一维递推就可以 两种情况 1、dp[i] += dp[i-1] 在i-1的串后面直接加一个B 2、dp[i]+=dp[i-2]+dp[i-3]+...+dp[i-k-1] 这部分的意思是在串后面补连续1个L 2个L 3个L。。K个L的情况 那补一个L的情况要取决于dp[i-1]里面最后一位为B的个数 正好为dp[i-2]，依次可做。

 import java.text.*;
 import java.io.*;
 import java.util.*;
 import java.math.*;
 import java.applet.*;
 public class Lemon {
     public static void main(String args[])
     {
         Scanner cin = new Scanner(System.in);
         BigInteger dp[] = new BigInteger[10010];
         BigInteger s[] = new BigInteger[10010];
         int i,j,n,k;
         n = cin.nextInt();
         k = cin.nextInt();
         for(i = 0; i <= n ; i++)
             dp[i] = s[i] = BigInteger.valueOf(0);
         dp[0] = s[0] = BigInteger.valueOf(1);
         if(k==0)
         dp[1] = BigInteger.valueOf(1);
         else
         dp[1] = BigInteger.valueOf(2);
         s[1] = s[0].add(dp[1]);
         for(i = 2; i <= n ; i++)
         {
             dp[i] = dp[i].add(dp[i-1]);
             BigInteger ss ;
             if(i-k-2>=0)
             ss = s[i-2].subtract(s[i-k-2]);
             else
             ss = s[i-2];
             if(k>=i) dp[i] = dp[i].add(BigInteger.valueOf(1));
             dp[i] = dp[i].add(ss);
             s[i] = s[i-1].add(dp[i]);
 
         }
         System.out.println(dp[n]);
     }
 
 }