본문 바로가기

알고리즘 · 코딩

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

문제 링크

https://programmers.co.kr/learn/courses/30/lessons/12913

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr


C++ 풀이

#include <vector>
#include <algorithm>
using namespace std;

int previous_max(vector<int> &previous_line, int index) {
	if (index == 0)
		return *max_element(previous_line.begin() + 1, previous_line.end());
	else if (index == 1)
		return (previous_line[0] > *max_element(previous_line.begin() + 2, previous_line.end())) ? previous_line[0] : *max_element(previous_line.begin() + 2, previous_line.end());
	else if (index == 2)
		return (previous_line[3] > *max_element(previous_line.begin(), previous_line.begin() + 2)) ? previous_line[3] : *max_element(previous_line.begin(), previous_line.begin() + 2);
	else if (index == 3)
		return *max_element(previous_line.begin(), previous_line.end() - 1);
}

int solution(vector<vector<int> > land)
{
	for (int i = 1; i < land.size(); i++) {
		for (int j = 0; j < 4; j++)
			land[i][j] += previous_max(land[i - 1], j); //이전 행에서 같은 열을 제외한 원소들 중 가장 큰 원소를 더함
	}
	return *max_element((land.back()).begin(), (land.back()).end());
}

동적계획법(Dynamic Programming)을 활용하여 풀었다.

 

이전 행에서 같은 열을 제외한 원소들 중 가장 큰 원소를 더해 내려갔다.

 

 

 

반응형