力扣每日一题5-18

class Solution {

public int colorTheGrid(int m, int n) {

// 每一列可能的状态总数 每个单元有3可能

int totalState = 1;

for (int i = 0; i < m; ++i) totalState *= 3;

// pre[k] 代表前一轮dp 状态为k 的方案总数

int [] pre = new int [totalState];

// 初始化合法填色 的 方案数为1

for (int state = 0; state < totalState; ++state) {

if (checkCol(state, m)) pre[state] = 1;

}

for (int i = 1; i < n; ++i) {

int [] cur = new int [totalState];

// 枚举可能的转移

for (int preState = 0; preState < totalState; ++preState) {

if (pre[preState] == 0) continue;

for (int curState = 0; curState < totalState; ++curState) {

// 行和列的填色都满足题设

if (checkCol(curState, m) && checkRow(preState, curState, m)) {

cur[curState] = addWithMode(cur[curState], pre[preState]);

}

}

}

pre = cur;

}

int planNum = 0;

for (int state = 0; state < totalState; ++state) {

planNum = addWithMode(planNum, pre[state]);

}

return planNum;

}

private boolean checkCol(int state, int m) {

int prevCeilColor = -1;

while (m-- > 0) {

if (state % 3 == prevCeilColor) {

return false;

}

prevCeilColor = state % 3;

state /= 3;

}

return true;

}

private boolean checkRow(int pre, int cur, int m) {

while (m-- > 0) {

if (pre % 3 == cur % 3) return false;

pre /= 3;

cur /= 3;

}

return true;

}

private int addWithMode(int a, int b) {

int mode = (int) 1E9 + 7;

return (a + b) % mode;

}

}

相关推荐
runafterhit几秒前
算法基础 典型题 数学(基础)
算法
superlls18 分钟前
(场景题)Java 导出 Excel 的两种方式
java·开发语言·后端·excel
harmful_sheep26 分钟前
springMvc
java
一只小透明啊啊啊啊28 分钟前
Java电商项目中的概念: 高并发、分布式、高可用、微服务、海量数据处理
java·分布式·微服务
三维小码29 分钟前
相机外参初始估计
算法·计算机视觉
宁清明2 小时前
【小宁的学习日记2 C语言】C语言判断
c语言·学习·算法
2401_841495643 小时前
【数据结构】基于Prim算法的最小生成树
java·数据结构·c++·python·算法·最小生成树·prim
木井巳4 小时前
[Java数据结构和算法] HashMap 和 HashSet
java·数据结构·1024程序员节
祈祷苍天赐我java之术5 小时前
解析常见的限流算法
java·数据结构·算法
摇滚侠5 小时前
IDEA 启动前端项目 IDEA 切换分支
java·ide·intellij-idea