[Programmers] 도둑질

개발자·2021년 3월 30일
0
post-thumbnail

문제 설명

도둑이 어느 마을을 털 계획을 하고 있습니다. 이 마을의 모든 집들은 아래 그림과 같이 동그랗게 배치되어 있습니다.

각 집들은 서로 인접한 집들과 방범장치가 연결되어 있기 때문에 인접한 두 집을 털면 경보가 울립니다.

각 집에 있는 돈이 담긴 배열 money가 주어질 때, 도둑이 훔칠 수 있는 돈의 최댓값을 return 하도록 solution 함수를 작성하세요.

제한 사항

  • 이 마을에 있는 집은 3개 이상 1,000,000개 이하입니다.
  • money 배열의 각 원소는 0 이상 1,000 이하인 정수입니다.

소스코드

#include <string>
#include <vector>

using namespace std;

int solution(vector<int> money) {
    int answer = 0;
    int dp1[1000010];
    int dp2[1000010];
    
    // 첫 번째 집을 터는 경우
    dp1[0] = dp1[1] = money[0];
    
    for(int i=2;i<=money.size();i++) {
        dp1[i] = max(dp1[i-2]+money[i], dp1[i-1]);
    }
    
    // 첫 번째를 털지 않고, 두 번째 집부터 터는 경우
    dp2[0] = 0;
    dp2[1] = money[1];
    
    for(int i=2;i<money.size();i++) {
        dp2[i] = max(dp2[i-2]+money[i], dp2[i-1]);
    }
    
    // 둘 중 큰 값이 정답
    answer = max(dp1[money.size()-2], dp2[money.size()-1]);
    return answer;
}

profile
log.info("공부 기록 블로9")

0개의 댓글

Powered by GraphCDN, the GraphQL CDN