티스토리 뷰

728x90

https://www.acmicpc.net/problem/1149

 

1149번: RGB거리

첫째 줄에 집의 수 N(2 ≤ N ≤ 1,000)이 주어진다. 둘째 줄부터 N개의 줄에는 각 집을 빨강, 초록, 파랑으로 칠하는 비용이 1번 집부터 한 줄에 하나씩 주어진다. 집을 칠하는 비용은 1,000보다 작거나

www.acmicpc.net

동적계획법 C++

각 집을 빨강, 초록, 파랑으로 칠할때의 비용을 입력받고 인접한 집은 서로 색이 다르도록 칠한 경우 모든 집을 칠했을 때 드는 최저비용구하기

 

접근방법

DP 문제라서 첫번째까지의 최선, 두번째 까지의 최선,,,,을 구해가면 되겠다 까지는 알겠는데

뭔가 부족한 부분이 있었다. 첫번째 까지의 최선을 저장하는 값이 하나가 아니라 한 집당 세개의 색을 칠했을 경우 모두를 고려해야 했다.

배열엔 이 집에 빨간색을 칠했을때의 최저비용, 초록색을 칠했을 때의 최저비용, 파란색을 칠했을 때의 최저비용을 저장한다.

배열은 다 쓰지않고 두 집 만큼만 저장하면서 갱신해간다. (n-1, n)

n번째 집에 빨간색을 칠한다면 n-1에 초록혹은 파랑을 칠했을 경우의 최솟값을 비교하여 둘 중 작은 값에 n번째 집에 빨간색을 칠하는 비용을 더한다. 그 값을 배열 n번에 갱신하고 n-1에 복사한다. (거의 뭐 나만 알아듣게 쓰는중) R,G,B 모두 똑같이 반복한다.

마지막까지 이 과정을 반복하고 마지막 배열의 RGB중에 가장 작은 값을 출력하면 끝

 

소스코드

#include <iostream>
using namespace std;

int main(){
    int n;
    cin >> n;
    
    int mins[2][3];
    for(int i = 0 ; i < 3 ; i++)
        cin >> mins[0][i];
    for(int i = 1 ; i < n ; i++){
        cin >> mins[1][0] >> mins[1][1] >> mins[1][2];
        mins[1][0] += min(mins[0][1], mins[0][2]);
        mins[1][1] += min(mins[0][0], mins[0][2]);
        mins[1][2] += min(mins[0][0], mins[0][1]);
        for(int j = 0 ; j < 3 ; j++){
            mins[0][j] = mins[1][j];
        }
    }
    cout << min(min(mins[1][0], mins[1][1]), mins[1][2]);
    return 0;
}
728x90
250x250
공지사항
최근에 올라온 글
최근에 달린 댓글
Total
Today
Yesterday
링크
«   2024/03   »
1 2
3 4 5 6 7 8 9
10 11 12 13 14 15 16
17 18 19 20 21 22 23
24 25 26 27 28 29 30
31
글 보관함