leetcode 2320. 统计放置房子的方式数

2320. 统计放置房子的方式数

class Solution {
public:
    const int MOD = 1'000'000'007;
    int countHousePlacements(int n) {
        if(n == 1) return 4;
        vector<long> dpZero(n+1);//dpZero[i]代表在位置 i 上不放置房子
        vector<long> dpOne(n+1);//dpOne[i]代表在位置 i 上放置房子
        dpZero[0] = 1;dpZero[1] = 1;
        dpOne[0] = 1;dpOne[1] = 1;
        for(int i = 2;i <= n;++i){
            dpOne[i] = dpZero[i-1];//如果在 i 上放置房子,位置i-1必须不放置房子
            dpZero[i] = (dpOne[i-1] + dpZero[i-1]) % MOD;//如果在 i 上不放置房子，位置i-1放不放房子都可以
        }
        return (dpOne[n] + dpZero[n]) * (dpOne[n] + dpZero[n])  % MOD;//由于两侧的房屋互相独立，根据乘法原理，答案为相乘而不是相加
    }
};

 灵神题解：其实就相当于斐波那契数列

class Solution {
public:
//若不放房子，那么第 i−1 个地块可放可不放，则有 f[i]=f[i−1]；
//若放房子，那么第 i−1 个地块无法放房子，第 i−2 个地块可放可不放，则有 f[i]=f[i−2]。
//所以有 : f[i]=f[i−1]+f[i−2]
    const int MOD = 1'000'000'007;
    int countHousePlacements(int n) {
        if(n == 1) return 4;
        vector<int> dp(n+1);//dp[i]代表在直到位置 i 单侧的方案数 
        dp[0] = 1;dp[1] = 2;
        for(int i = 2;i <= n;++i){
            dp[i] = (dp[i-1] + dp[i-2]) % MOD;
        }
        return (long)dp[n] * dp[n] % MOD;//由于两侧的房屋互相独立，根据乘法原理，答案为相乘而不是相加
    }
};