LeetCode——1931. 用三种不同颜色为网格涂色[Painting a Grid With Three Different Colors][困难]——分析及代码[Java]
一、题目
给你两个整数 m 和 n 。构造一个 m x n 的网格,其中每个单元格最开始是白色。请你用 红、绿、蓝 三种颜色为每个单元格涂色。所有单元格都需要被涂色。
涂色方案需要满足:不存在相邻两个单元格颜色相同的情况 。返回网格涂色的方法数。因为答案可能非常大, 返回 对 10^9 + 7 取余 的结果。
示例 1:
输入:m = 1, n = 1
输出:3
示例 2:
输入:m = 1, n = 2
输出:6
示例 3:
输入:m = 5, n = 5
输出:580986
提示:
- 1 <= m <= 5
- 1 <= n <= 1000
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/painting-a-grid-with-three-different-colors
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二、分析及代码
1. 动态规划
(1)思路
先预处理得到单行的所有有效涂色方案,再进一步计算得到每种单行方案对应的有效邻行方案。在此基础上,结合动态规划方法,逐行求解各种涂色状态对应的方案总数,最后统计得到总方案数。
(2)代码
class Solution {
static int mod = (int)1e9 + 7;
public int colorTheGrid(int m, int n) {
//获得单行所有涂色方案
Map<Integer, List<Integer>> line = new HashMap<>();//用0、1、2表示各个网格的颜色,key为方案对应的数值,value为方案对应的数组
int range = (int)Math.pow(3, m);//对单行的m个网格,可能的取值范围为[0, 3^m)
for (int i = 0; i < range; i++) {
//遍历可能的数值
//获得数值对应的数组
List<Integer> list = new ArrayList<Integer>();
int num = i;