leetcode：265. 粉刷房子 II

题目来源

• leetcode：265-vip. 粉刷房子 II
• lintcode

题目描述

class Solution {
public:
    int minCostII(vector<vector<int>>& costs){
  
    }
};

题目解析

• leetcode：256. 粉刷房子的扩展题

思路

第i 个房子要选costs[i][ x ] 颜色成本最小的哪一个， 如果这个颜色与i-1 得到的结果颜色一致那么就不能选这个颜色， 继续找成本第二小的颜色。 这样一直算到最后一个。

class Solution {
public:
    int minCostII(vector<vector<int>>& costs){
        if(costs.empty() || costs[0].empty()){
            return 0;
        }
        int N = costs.size(), M = costs[0].size();
        int min1 = -1, min2 = -1; //min1   min2  当前第一小， 第二小。 
        for (int i = 0; i < N; ++i) {
            int preMin1 = min1, preMin2 = min2;  //preMin1 preMin2 前一个第一小，第二小
            min1 = -1, min2 = -1;
            for (int j = 0; j < M; ++j) {  //研究当前行油漆当前颜色的最小花费
                if(preMin1 != j){  //与上一个颜色不一样
                    costs[i][j] += preMin1 == -1 ? 0 : costs[i - 1][preMin1];
                }else{
                    costs[i][j] += preMin2 == -1 ? 0 : costs[i - 1][preMin2];
                }
                
                if(min1 == -1 || costs[i][j] < costs[i][min1]){
                    min2 = min1;
                    min1 = j;
                }else if(min2 == -1 || costs[i][j] < costs[i][min2]){
                    min2 = j;
                }
            }
        }
        
        return costs[N - 1][min1];
    }
};