URAL1513. Lemon Tale(dp)

时间:2023-03-08 19:46:44

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