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]);
    }

}