LeetCode-6100. 统计放置房子的方式数

196 阅读1分钟

1. 描述

一条街道上共有 n * 2 个 地块 ,街道的两侧各有 n 个地块。每一边的地块都按从 1 到 n 编号。每个地块上都可以放置一所房子。

现要求街道同一侧不能存在两所房子相邻的情况,请你计算并返回放置房屋的方式数目。由于答案可能很大,需要对 109 + 7 取余后再返回。

注意,如果一所房子放置在这条街某一侧上的第 i 个地块,不影响在另一侧的第 i 个地块放置房子。

2. 示例

示例 1

输入:n = 1
输出:4
解释:
可能的放置方式:
1. 所有地块都不放置房子。
2. 一所房子放在街道的某一侧。
3. 一所房子放在街道的另一侧。
4. 放置两所房子,街道两侧各放置一所。

示例 2

输入:n = 2
输出:9

3. 答案

class Solution {
    public int countHousePlacements(int n) {
        int mod = (int)1e9 + 7;
        int dp0 = 1;
        int dp1 = 1;
        for(int i = 2; i <= n; i++) {
            int temp = dp0;
            dp0 = (dp0 + dp1) % mod;
            dp1 = temp;
        }
        long sum = (dp0 + dp1) % mod;
        return (int)(sum * sum % mod);
    }
}
  • 标签:动态规划、滚动数组
  • 时间复杂度:O(n)
  • 空间复杂度:O(1)

LeetCode:6100. 统计放置房子的方式数