790. 多米诺和托米诺平铺

https://leetcode-cn.com/problems/domino-and-tromino-tiling/
有两种形状的瓷砖:一种是 2x1 的多米诺形,另一种是形如 “L” 的托米诺形。两种形状都可以旋转。

XX <- 多米诺

XX <- “L” 托米诺
X
给定 N 的值,有多少种方法可以平铺 2 x N 的面板?返回值 mod 10^9 + 7。

(平铺指的是每个正方形都必须有瓷砖覆盖。两个平铺不同,当且仅当面板上有四个方向上的相邻单元中的两个,使得恰好有一个平铺有一个瓷砖占据两个正方形。)

示例:输入: 3输出: 5解释:
下面列出了五种不同的方法,不同字母代表不同瓷砖:
XYZ XXZ XYY XXY XYY
XYZ YYZ XZZ XYY XXY

我们定义fe[n]表示2*n的格子有多少种铺法,定义fo[n]表示2*n + 1的格子有多少种铺法(也就是在2*n的格子上面含有一个格子),然后推导fe和fo的递推关系。对于fe的最顶层,我们分别有如下图所示的几种铺法,因此,可以得出递推关系为:
790. 多米诺和托米诺平铺_第1张图片

fe[n] = fe[n - 1] + fe[n - 2] + 2fe[n - 3] + 2fo[n - 3];
上图3,4,5,6可以归为一类
fe[n] = fe[n - 1] + fe[n - 2] + 2fo[n-2]

fo[n] = fo[n - 1] + fe[n - 1]

特别地,我们容易得知:fe[1] = 1, fe[2] = 2, fe[3] = 5, fo[1] = 2, fo[2] = 2, fo[3] = 4。因此就可以根据初始条件和递推关系写出基于动态规划的源代码了。

我们在下面写出的源代码的空间复杂度为O(N),时间复杂度为O(N)。但是注意到fe[n]和fo[n]也仅仅只和fe[n-1], fe[n-2], fe[n-3]以及fo[n-1], fo[n-2], fo[n-3]有关,所以还可以进一步将空间复杂度从O(n)优化到O(1)。

更新:其实把fo(n) = fo(n - 1) + fe(n - 1)代入fe(n),可以得到更简单的递推公式:fe[n] = fe[n - 1] + fe[n - 2] + 2fo[n-2],甚至还可以进一步优化为fe[n] = fe[n - 1] + fo[n - 1] + fo[n-2],还可以进一步优化为fe[n] = fo[n] + fo[n - 2]。这样写出来的代码应该就更简洁优雅了。

class Solution:
    def numTilings(self, N: int) -> int:
        a=[0, 1, 2, 5]
        b=[0, 1, 2, 4]
        c = 10**9+7
        if N<4:
            return a[N]
        for i in range(4, N+1):
            a.append((a[i-1]+2*b[i-2]+a[i-2])%c)
            b.append((a[i-1]+b[i-1])%c)
        return a[N]

内存占用最少:

class Solution:
   def numTilings(self, N: int) -> int:
       a=[1, 2, 0]
       b=[1, 2, 0]
       c = 10**9+7
       if N<3:
           return a[N-1]
       for i in range(3, N+1):
           a[2] = (a[1]+2*b[0]+a[0])%c
           b[2] = (a[1]+b[1])%c
           a[0] = a[1]
           a[1] = a[2]
           b[0] = b[1]
           b[1] = b[2]     
       return a[2]

你可能感兴趣的:(LeetCode算法题解析)