题目
给你两个长度为 n 下标从 0 开始的整数数组 cost 和 time ,分别表示给 n 堵不同的墙刷油漆需要的开销和时间。你有两名油漆匠:
一位需要 付费 的油漆匠,刷第 i 堵墙需要花费 time[i] 单位的时间,开销为 cost[i] 单位的钱。
一位 免费 的油漆匠,刷 任意 一堵墙的时间为 1 单位,开销为 0 。但是必须在付费油漆匠 工作 时,免费油漆匠才会工作。
请你返回刷完 n 堵墙最少开销为多少。
示例 1:
输入:cost = [1,2,3,2], time = [1,2,3,2]
输出:3
解释:下标为 0 和 1 的墙由付费油漆匠来刷,需要 3 单位时间。同时,免费油漆匠刷下标为 2 和 3 的墙,需要 2 单位时间,开销为 0 。总开销为 1 + 2 = 3 。
示例 2:
输入:cost = [2,3,4,2], time = [1,1,1,1]
输出:4
解释:下标为 0 和 3 的墙由付费油漆匠来刷,需要 2 单位时间。同时,免费油漆匠刷下标为 1 和 2 的墙,需要 2 单位时间,开销为 0 。总开销为 2 + 2 = 4 。
提示:
1 <= cost.length <= 500
cost.length == time.length
1 <= cost[i] <= 106
1 <= time[i] <= 500
思路
使用动态规划,创建一个f数组记录从0面墙开始刷到现在i面墙的最小费用,创建一个g数组记录从i面墙开始后的最小费用情况,先遍历每一个墙,然后在这里面遍历每一面墙的可能,分为支付和免费的情况,后更新f为g,返回f从n面墙开始到最后的最小值,就是所有墙面刷完的最小费用
代码
class Solution {
public:
int paintWalls(vector<int>& cost, vector<int>& time) {
int n = cost.size();
vector<int> f(n*2+1,INT_MAX/2);
f[n] = 0;
for(int i = 0;i < n;i++){
vector<int>g(n*2+1,INT_MAX/2);
for(int j = 0;j <= n*2;j++){
//付费
g[min(j+time[i],n*2)] = min(g[min(j+time[i],n*2)],f[j]+cost[i]);
//免费
if(j>0){
g[j-1] = min(g[j-1],f[j]);
}
}
//move函数会返回g对象的移动引用
f = move(g);
}
//f从n面墙开始位置到结束位置的最小值
return *min_element(f.begin()+n,f.end());
}
};
总结
- move函数会返回g对象的移动引用
- *min_element(f.begin()+n,f.end())f从n面墙开始位置到结束位置的最小值
- 注意f和g的初始值