题:
思:
如果房子数为0则直接返回-1
先数出已有街区machi ,如果已有街区大于target,直接返回-1
M=target-machi代表还需要的街区数量
如果M = 0,则答案唯一,直接计算即可 —> 不对,如果M = 0 答案也不一定唯一
e.g houses[1,2,0,1,1] target=3 M=0 ,此时0可以被刷成1也可以被刷成2
否则题目化简为,在剩余的所有没上色的房子houses里,划分M个街区的最小消耗
开始时,当前已有街区cur为0
从houses[0]里选择最小cost,cur = 1
每次从houses[i]里选择最小的cost,如果颜色发生了变化,则cur+1。 houses[i]最多有M-cur种选择,所以要分别对这些选择进行讨论
所以是深搜的思路
结果
AC: 9/60. 超出时间限制
代码如下:
public static int minCost(int[] houses, int[][] cost, int m, int n, int target) {
if (m == 0)
return -1;
// 没有上色的房子的数量
int house = 0;
// 已有街区
int cur = 0;
// 上一个颜色
int lastColor = 0;
int allCost = 0;
int[] minCost = new int[1];
minCost[0] = Integer.MAX_VALUE;
for (int i = 0; i < m; i++) {
if (houses[i] != 0)
house++;
if (cur == 0 && houses[i] != 0) {
cur = 1;
lastColor = houses[i];
} else if (houses[i] != 0 && houses[i] != lastColor) {
cur++;
lastColor = houses[i];
}
}
// 还需要的街区数量M
int M = target - cur;
// 特别注意,如果新刷的房子和前面那个房子连起来都是同一种颜色,那么M不减1,如果不一样,则+1
// eg. houses[0,1,0,1,0,0,0],target = 3 则初始M = 2
// 如果刷成了[1,1,0,1,0,0,0],刷了一个1以后,M应该变为1,但它和已有街区衔接,因此M+1,仍然为2
dfs(houses, cost, M, allCost, 0, minCost);
return minCost[0] == Integer.MAX_VALUE ? -1 : minCost[0];
}
// cur代表当前该刷的房子
public static void dfs(int[] houses, int[][] cost, int M, int allCost, int cur, int[] minCost) {
// dfs每次选择最小的消耗(因为dfs要讨论所有的消耗情况,因此没必要每次选择最小的消耗)
// 递归结束条件,全部房子上完色
if (cur == houses.length && M == 0) {
minCost[0] = Math.min(minCost[0], allCost);
return;
} else if (cur == houses.length || M < 0) {
return;
}
if (houses[cur] != 0) {
dfs(houses, cost, M, allCost, cur + 1, minCost);
} else {
for (int i = 0; i < cost[cur].length; i++) {
// 分别递归选择第i+1种颜色的消耗情况
houses[cur] = i + 1;
int Q = 1;
// 如果当前房子与前面的颜色相同或者后面的颜色不为0且和这个房子颜色相同,则Q=0
if ((cur - 1 >= 0 && houses[cur - 1] == houses[cur]) || (cur + 1 < houses.length && houses[cur + 1] == houses[cur]))
Q = 0;
dfs(houses, cost, M - Q, allCost + cost[cur][i], cur + 1, minCost);
houses[cur] = 0;
}
}
}
原因分析:dfs时间复杂度太大,算是暴力解法,不知道该如何剪枝
但如果这个题用dp做,又完全没思路,于是看答案。
答案是三维dp,嗯,看懂思想了,细节边界很难写,直接贴答案吧
https://leetcode-cn.com/problems/paint-house-iii/solution/fen-shua-fang-zi-iii-by-leetcode-solutio-powb/
还是太菜了呀!一个小时写了个不能剪枝的回溯dfs,直接时间爆炸了,哎!