约翰的后花园

1 题目链接

约翰的后花园

2 题目描述

描述

约翰想在他家后面的空地上建一个后花园，现在有两种砖，一种3 dm的高度，7 dm的高度。约翰想围成x dm的墙。如果约翰能做到，输出YES，否则输出NO。

X是一个整数，取值范围为 [3, 1000]。

示例

样例 1:
输入 : x = 10
输出 : "YES"
解释：
x = 3 + 7 : 即需要1匹3 dm高度的砖和1匹7 dm 高度的砖。

样例 2:
输入 : x = 5
输出 : "NO"
解释：
不能用高度为3 dm的砖和高度为7 dm的砖砌成 5 dm的墙。

样例 3:
输入 : x = 13
输出 : "YES"
解释：
x = 2 * 3 + 7 : 即需要2匹3 dm高度的砖和1匹7 dm 高度的砖。

3 代码思路

背包问题，有无限个物品可以取，求是否能够满足条件。我觉得也可以理解为爬楼梯问题的变种，即两种步长分别是3和5，求是否能准确上到楼梯重点。爬楼梯是问有多少种方案，这个是在求是否存在一条方案。

4 代码实现

public class Solution {
  public String isBuild(int x) {
    // 记录砌成高度为i的墙可行性
    boolean[] dp = new boolean[x+1];
    dp[0] = true;
    for(int i = 1; i <= x; i++){
      if(i >= 3){
        // 如果能砌程高度为i - 3的墙，那么可以砌成高度为i的墙
        dp[i] = dp[i] || dp[i - 3];
      }
      if(i >= 7){
        // 如果能砌程高度为i - 7的墙，那么可以砌成高度为i的墙
        dp[i] = dp[i] || dp[i - 7];
      }
    }
    if(dp[x] == true){
      return "YES";
    }else{
      return "NO";
    }
  }
}