265. 粉刷房子 II

Q:

A:

首先这题可以和粉刷房子这题一样解法，对于i号房子，遍历k种颜色，对于每一种，都去找i-1号房子除该颜色之外的最小花费。但上一题是3种颜色，总复杂度O(N)，这题k种颜色，复杂度O(NK^2)，题目要求O(NK)，则对于i号房子我们保存下当前房子最小的花费以备i+1号房子使用，但因为相邻房子不能涂相同颜色的油漆。假设对于i号房子最小花费是涂x号油漆，则对于i+1号房子来说，一定不能涂x号了。解决办法是我们对于每一个房子，保存最小花费MIN和次最小花费2_MIN。对于下一个房子，涂某种油漆时不能使用MIN，即油漆号等于MIN所对应的油漆号时，此时只能使用2_MIN。除此之外，都使用MIN。

class Solution {
public:
    int minCostII(vector<vector<int>>& costs) {
        if(costs.size()==0 or costs[0].size()==0){
            return 0;
        }
        int pre_min_index=-1,pre_min_cost=0,pre_2min_cost=0;
        for(int i=0;i<costs.size();++i){
            int cur_min_index,cur_min_cost=INT_MAX,cur_2min_cost=INT_MAX;
            for(int j=0;j<costs[0].size();++j){
                if(j!=pre_min_index){
                    costs[i][j]+=pre_min_cost;
                }
                else{
                    costs[i][j]+=pre_2min_cost;
                }
                if(costs[i][j]<cur_min_cost){
                    cur_2min_cost=cur_min_cost;
                    cur_min_cost=costs[i][j];
                    cur_min_index=j;
                }
                else if(costs[i][j]<cur_2min_cost){
                    cur_2min_cost=costs[i][j];
                }
            }
            pre_min_index=cur_min_index;
            pre_min_cost=cur_min_cost;
            pre_2min_cost=cur_2min_cost;
            // cout<<pre_min_index<<" "<<pre_min_cost<<" "<<pre_2min_cost<<endl;
        }
        return pre_min_cost;
    }
};