5-wnsmir

[wnsmir]

🔗 문제 링크

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

✔️ 소요된 시간

50min

✨ 수도 코드

각 열의 최적 점수를 구하는 과정에서 이전 열의 최적 점수 계산이 반복된다는 사실을 인지한 시점부터 동적계획법을 사용해야겠다고 생각.
-> i-1과 i-2 열에서 다른 행의 값만을 선택해도 충분히 최적의 결과를 보장할 수 있음을 이용.

1. 첫 번째 행의 i번째 스티커를 선택하는 경우:
   dp[0][i] = max(dp[1][i-1], dp[0][i-2], dp[1][i-2]) + stickers[0][i]

2. 두 번째 행의 i번째 스티커를 선택하는 경우:
   dp[1][i] = max(dp[0][i-1], dp[0][i-2], dp[1][i-2]) + stickers[1][i]

import sys
input = sys.stdin.read

def max_sticker_score(n, stickers):
    # n이 1일 때는 간단히 큰 값을 반환
    if n == 1:
        return max(stickers[0][0], stickers[1][0])
    
    # DP 배열 초기화
    dp = [[0] * n for _ in range(2)]
    
    # 초기값 설정
    dp[0][0] = stickers[0][0]
    dp[1][0] = stickers[1][0]
    
    # n이 2 이상인 경우 첫 두 열을 초기화
    if n > 1:
        dp[0][1] = dp[1][0] + stickers[0][1]
        dp[1][1] = dp[0][0] + stickers[1][1]
    
    # DP 배열 채우기
    for i in range(2, n):
        dp[0][i] = max(dp[1][i-1], dp[1][i-2]) + stickers[0][i] # i-2까지 처리 (점화식)
        dp[1][i] = max(dp[0][i-1], dp[0][i-2]) + stickers[1][i]
    
    # 최댓값 반환
    return max(dp[0][n-1], dp[1][n-1])

# 입력 처리
data = input().strip().split()
T = int(data[0])  # 테스트 케이스 수
index = 1
results = []

for _ in range(T):
    n = int(data[index])
    stickers = [
        list(map(int, data[index + 1:index + 1 + n])),
        list(map(int, data[index + 1 + n:index + 1 + 2 * n]))
    ]
    results.append(max_sticker_score(n, stickers))
    index += 2 * n + 1

# 결과 출력
for result in results:
    print(result)

[kangrae-jo]

대면 미팅때 상세하게 설명해주신 문제가 나와서 반가웠습니다.

그때 설명 들은 기억대로 잘 구현할 수 있었습니다.

근데 만약에 설명을 듣지 않았더라면 1시간 이상 걸리는 문제였을 것 같네요..
그리고 실전에서 나왔다면 아마 시간안에 풀지 못했을 것이라 생각되네요.
dp 좀 더 공부 열심히해야 할 것 같네요

좋은 문제 감사합니다.

CPP CODE

#include <algorithm>
#include <iostream>
#include <vector>

using namespace std;

int main() {
    int T;
    cin >> T;

    while (T--) {
        int n;
        cin >> n;

        vector<vector<int>> s(2, vector<int>(n));
        for (int i = 0; i < n; i++) 
            cin >> s[0][i];
        for (int i = 0; i < n; i++) 
            cin >> s[1][i];
        
        if (n == 1){
            cout << max(s[0][0],s[1][0]) << endl;
            continue;
        }

        vector<vector<int>> dp(2, vector<int>(n, 0));
        dp[0][0] = s[0][0];
        dp[1][0] = s[1][0];
        dp[0][1] = dp[1][0] + s[0][1];
        dp[1][1] = dp[0][0] + s[1][1];

        for (int i = 2; i < n; i++) {
            dp[0][i] = max(dp[1][i - 1], dp[1][i - 2]) + s[0][i];
            dp[1][i] = max(dp[0][i - 1], dp[0][i - 2]) + s[1][i];
        }
        cout << max(dp[0][n - 1], dp[1][n - 1]) << endl;
    }

    return 0;
}

그리고 코드 넣으실때
처음 백틱 3개 뒤에 python 적으면 코드에 색 표현이 돼서 더 보기 좋을 것 같아요

[kangrae-jo]

이 케이스를 생각못해서 푸는데 좀 오래걸렸네요...

[kangrae-jo]

파이썬은 복잡한 입력을 이렇게 받는군요...!
완전 신기합니다..

[g0rnn]

재밌는 dp 문제 감사합니다!

my answer

#include <iostream>
#include <algorithm>
#include <climits>
using namespace std;

int t;
int board[2][100001];
int dpCache[2][100001];
int max0 = INT_MIN;
int max1 = INT_MIN;

int dp(int n) {
	if (n == 1) {
		return max(board[0][0], board[1][0]);
	}

	dpCache[0][0] = board[0][0];
	dpCache[1][0] = board[1][0];

	dpCache[0][1] = board[0][1] + board[1][0];
	dpCache[1][1] = board[1][1] + board[0][0];

	for (int i = 2; i < n; i++) {
		dpCache[0][i] = board[0][i] + max(dpCache[1][i-1], dpCache[1][i-2]);
		dpCache[1][i] = board[1][i] + max(dpCache[0][i-1], dpCache[0][i-2]);
	}

	return max(dpCache[0][n-1], dpCache[1][n-1]);
}

int main() {
	cin >> t;
	for (int _ = 0; _ < t; _++) {
		int n;
		cin >> n;
		for (int i = 0; i < n; i++) {
			cin >> board[0][i];
		}
		for (int i = 0; i < n; i++) {
			cin >> board[1][i];
		}
		cout << dp(n) << '\n';

	}
	return 0;
}