-
Notifications
You must be signed in to change notification settings - Fork 122
/
Copy path1473. Paint House III.cpp
86 lines (70 loc) · 2.88 KB
/
1473. Paint House III.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
// 1473.✅ Paint House III
/*
There is a row of m houses in a small city, each house must be painted with one of the n colors (labeled from 1 to n), some houses that have been painted last summer should not be painted again.
A neighborhood is a maximal group of continuous houses that are painted with the same color.
For example: houses = [1,2,2,3,3,2,1,1] contains 5 neighborhoods [{1}, {2,2}, {3,3}, {2}, {1,1}].
Given an array houses, an m x n matrix cost and an integer target where:
houses[i]: is the color of the house i, and 0 if the house is not painted yet.
cost[i][j]: is the cost of paint the house i with the color j + 1.
Return the minimum cost of painting all the remaining houses in such a way that there are exactly target neighborhoods. If it is not possible, return -1.
Example 1:
Input: houses = [0,0,0,0,0], cost = [[1,10],[10,1],[10,1],[1,10],[5,1]], m = 5, n = 2, target = 3
Output: 9
Explanation: Paint houses of this way [1,2,2,1,1]
This array contains target = 3 neighborhoods, [{1}, {2,2}, {1,1}].
Cost of paint all houses (1 + 1 + 1 + 1 + 5) = 9.
Example 2:
Input: houses = [0,2,1,2,0], cost = [[1,10],[10,1],[10,1],[1,10],[5,1]], m = 5, n = 2, target = 3
Output: 11
Explanation: Some houses are already painted, Paint the houses of this way [2,2,1,2,2]
This array contains target = 3 neighborhoods, [{2,2}, {1}, {2,2}].
Cost of paint the first and last house (10 + 1) = 11.
Example 3:
Input: houses = [3,1,2,3], cost = [[1,1,1],[1,1,1],[1,1,1],[1,1,1]], m = 4, n = 3, target = 3
Output: -1
Explanation: Houses are already painted with a total of 4 neighborhoods [{3},{1},{2},{3}] different of target = 3.
Constraints:
m == houses.length == cost.length
n == cost[i].length
1 <= m <= 100
1 <= n <= 20
1 <= target <= m
0 <= houses[i] <= n
1 <= cost[i][j] <= 104
*/
class Solution
{
public:
int dp[100][21][101];
int helper(int i, int l, int t, vector<int> &houses, vector<vector<int>> &cost, int m, int n, int target)
{
if (t > target)
return INT_MAX / 2;
if (i == m)
return (t == target) ? 0 : INT_MAX / 2;
if (dp[i][l][t] != -1)
return dp[i][l][t];
else
{
int ans = INT_MAX / 2;
if (houses[i] == 0)
{
for (int j = 0; j < n; ++j)
{
ans = min(ans, cost[i][j] + helper(i + 1, j + 1, (l == j + 1) ? t : t + 1, houses, cost, m, n, target));
}
}
else
{
ans = min(ans, helper(i + 1, houses[i], (houses[i] == l) ? t : t + 1, houses, cost, m, n, target));
}
return dp[i][l][t] = ans;
}
}
int minCost(vector<int> &houses, vector<vector<int>> &cost, int m, int n, int target)
{
memset(dp, -1, sizeof dp);
int ans = helper(0, 0, 0, houses, cost, m, n, target);
return ans == INT_MAX / 2 ? -1 : ans;
}
};