题目链接
题目描述
一条街道上共有 n * 2 个 地块 ,街道的两侧各有 n 个地块。每一边的地块都按从 1 到 n 编号。每个地块上都可以放置一所房子。
现要求街道同一侧不能存在两所房子相邻的情况,请你计算并返回放置房屋的方式数目。由于答案可能很大,需要对 10^9 + 7
取余后再返回。
注意,如果一所房子放置在这条街某一侧上的第 i 个地块,不影响在另一侧的第 i 个地块放置房子。
示例 1:
输入:n = 1
输出:4
解释:
可能的放置方式:
1. 所有地块都不放置房子。
2. 一所房子放在街道的某一侧。
3. 一所房子放在街道的另一侧。
4. 放置两所房子,街道两侧各放置一所。
示例 2:
输入:n = 2
输出:9
解释:如上图所示,共有 9 种可能的放置方式。
提示:
1 <= n <= 10^4
解题思路
动态规划法
首先单独考虑一侧的房子,那么这题和打家劫舍就是一样的了。
dp[i]
表示前i个土地的放置方案数。
- 在考虑第i+1块土地时,如果第i块土地有房子,那么第i+1块土地不能放房子,即
dp[i + 1] = dp[i]
- 如果第i块土地没有房子,那么第i-1块可放可不放,即
dp[i + 1] = dp[i - 1]
- 因此
dp[i] = dp[i - 1] + dp[i - 2]
AC代码
class Solution {
public int countHousePlacements(int n) {
if (n == 0) {
return 1;
}
if (n == 1) {
return 4;
}
int[] dp = new int[n + 1];
dp[0] = 1;
dp[1] = 2;
for (int i = 2; i <= n; i++) {
dp[i] = (dp[i - 2] + dp[i - 1]) % 1000000007;
}
return (int) ((long) dp[n] * dp[n] % 1000000007);
}
}