[프로그래머스][C++] 땅따먹기

출처 프로그래머스

https://programmers.co.kr/

프로그래머스

문제

너무 복잡하게 생각해서 오래 걸렸던 문제이다 

생각보다 코드는 간단하게 나왔다.

 

#include <iostream>
#include <vector>
#define NUM_COLUMN 4
using namespace std;
int find_max(vector<int> v, int except) {
    int max = 0;
    for (int idx = 0; idx < v.size(); idx++) {
        if (max < v[idx] && idx != except)
            max = v[idx];
    }
    return max;
}
int solution(vector<vector<int> > land)
{
    int answer = 0;
    vector<vector<int>>::iterator viter = land.begin(), next;
    for (next = viter + 1; next != land.end(); next++) {
        for (int idx = 0; idx < NUM_COLUMN; idx++) {
            (*next)[idx] += find_max((*viter), idx);
        }
        viter = next;
    }
    return find_max((*viter), -1);
}