[补题][Codeforces478D]Red-Green Towers(DP)

题目链接

http://codeforces.com/problemset/problem/478/D

题意

叠放塔：有红、绿两种色块。从第一层开始，第一层1块，第二层2块，第i层i块。
要求每一层只能用同一种颜色的块。
输入：红块和绿块数目
输出；能叠放出的最高高度h的塔的种数。定义塔某一层的颜色不同，则为不同种。

题解

dp

需要get的点

• dp只需要两维，因为当塔的高度h固定，则给出使用的红块数，则使用的绿块数=h-红块数。
• 初始化：dp[0][0]=1即可 ？

• 滚动数组否则爆内存

  代码

package Exam;

import java.util.Scanner;

public class FirstTest {
    public static void main(String args[]) {
        Scanner in=new Scanner(System.in);
        int r=in.nextInt();
        int g=in.nextInt();
        
        int MAXH=(int) Math.floor(Math.sqrt(2*(r+g))) ;
        int MAXR=(int) 2e5;
        int MOD=(int)1e9+7;
        
        int h=MAXH;
        while(h*(h+1)/2>r+g) {
            --h;
        }
        
        int[][] dp=new int[2][MAXR];
        dp[0][0]=1;
        for(int i=1;i<=h;++i) {
            for(int j=0;j<=r;++j) {
                if(j-i>=0) {
                    dp[i&1][j]=((dp[(i-1)&1][j])%MOD+(dp[(i-1)&1][j-i])%MOD)%MOD;
                }
                else {
                    dp[i&1][j]=(dp[(i-1)&1][j])%MOD;
                }
            }
        }
        
        int ans=0;
        for(int i=Math.max(0,h*(h+1)/2-g);i<=r;++i) {
            ans+=dp[h&1][i];
        }
        System.out.println(ans);
    }
}