1931. 用三種不同顏色為網格涂色
mod_value = 10**9 + 7
class Solution:def colorTheGrid(self, m: int, n: int) -> int:# 1、預處理所有合法的單行涂色方案# 存儲 3^i,用于快速計算顏色編碼的每一位(類似位運算)# [3^0, 3^1, 3^2, ..., 3^(m-1)]pow3 = [3**i for i in range(m)]# 存儲所有合法的單行涂色方案(即同一行內相鄰格子顏色不同)valid = []for color in range(3**m):for i in range(1, m):# color // pow3[i] % 3表示顏色編碼 color 的第 i 位(從右往左數,最低位是 i=0)if color // pow3[i] % 3 == color //pow3[i-1] % 3:break # 相鄰顏色相同,非法else:valid.append(color)
?
?