[백준 9465 파이썬/python] 스티커

스티커

시간 제한	메모리 제한	제출	정답	맞힌 사람	정답 비율
1 초	256 MB	61016	28481	19806	46.710%

문제

상근이의 여동생 상냥이는 문방구에서 스티커 2n개를 구매했다. 스티커는 그림 (a)와 같이 2행 n열로 배치되어 있다. 상냥이는 스티커를 이용해 책상을 꾸미려고 한다.

상냥이가 구매한 스티커의 품질은 매우 좋지 않다. 스티커 한 장을 떼면, 그 스티커와 변을 공유하는 스티커는 모두 찢어져서 사용할 수 없게 된다. 즉, 뗀 스티커의 왼쪽, 오른쪽, 위, 아래에 있는 스티커는 사용할 수 없게 된다.

모든 스티커를 붙일 수 없게된 상냥이는 각 스티커에 점수를 매기고, 점수의 합이 최대가 되게 스티커를 떼어내려고 한다. 먼저, 그림 (b)와 같이 각 스티커에 점수를 매겼다. 상냥이가 뗄 수 있는 스티커의 점수의 최댓값을 구하는 프로그램을 작성하시오. 즉, 2n개의 스티커 중에서 점수의 합이 최대가 되면서 서로 변을 공유 하지 않는 스티커 집합을 구해야 한다.

위의 그림의 경우에 점수가 50, 50, 100, 60인 스티커를 고르면, 점수는 260이 되고 이 것이 최대 점수이다. 가장 높은 점수를 가지는 두 스티커 (100과 70)은 변을 공유하기 때문에, 동시에 뗄 수 없다.

입력

첫째 줄에 테스트 케이스의 개수 T가 주어진다. 각 테스트 케이스의 첫째 줄에는 n (1 ≤ n ≤ 100,000)이 주어진다. 다음 두 줄에는 n개의 정수가 주어지며, 각 정수는 그 위치에 해당하는 스티커의 점수이다. 연속하는 두 정수 사이에는 빈 칸이 하나 있다. 점수는 0보다 크거나 같고, 100보다 작거나 같은 정수이다.

출력

각 테스트 케이스 마다, 2n개의 스티커 중에서 두 변을 공유하지 않는 스티커 점수의 최댓값을 출력한다.

예제 입력 1

2
5
50 10 100 20 40
30 50 70 10 60
7
10 30 10 50 100 20 40
20 40 30 50 60 20 80

예제 출력 1

260
290

풀이

t = int(input())
for _ in range(t):
    n = int(input())
    line_0 = [0] + list(map(int, input().split()))
    line_1 = [0] + list(map(int, input().split()))
    a = list(zip(line_0, line_1))
    d = [[0] * 3 for _ in range(100001)]  # 0: 선택 0, 1: 1개 선택(위), 2: 1개 선택(아래)
    for i in range(1, n + 1):
        d[i][0] += max(d[i - 1])
        d[i][1] += max(d[i - 1][0], d[i - 1][2]) + a[i][0]
        d[i][2] += max(d[i - 1][0], d[i - 1][1]) + a[i][1]
    ans = max(d[n])
    print(ans)

1. 첫 번째 줄에서는 t라는 변수에 정수형 값을 입력받습니다. 이 값은 반복문을 실행하는 횟수를 결정합니다.

2. for문을 통해 t번 반복합니다. 이때, 반복문 안에서는 n이라는 변수에 정수형 값을 입력받습니다.

3. 다음으로 line_0과 line_1이라는 두 개의 리스트를 입력받습니다. 이때, 리스트의 첫 번째 원소로 0을 추가해주고, 입력받은 값들을 int 형으로 변환하여 리스트에 추가합니다.

4. zip 함수를 이용하여 line_0과 line_1을 합친 a 리스트를 만들어줍니다.

5. 2차원 리스트 d를 만들어줍니다. 이때, 리스트의 크기는 (n+1) x 3입니다. 0번 인덱스는 사용하지 않고, 1번부터 n번까지 이용합니다. 또한, 3개의 열을 가진 2차원 리스트입니다.

6. for문을 이용하여 1부터 n까지의 인덱스에 대해 다음 과정을 반복합니다.

7. 현재 인덱스에서 선택하지 않은 경우의 최대값을 d[i][0]에 저장합니다. 이전 인덱스에서의 3개 값 중 가장 큰 값을 현재 값으로 이용합니다.

8. 현재 인덱스에서 윗쪽에서 선택한 경우의 최대값을 d[i][1]에 저장합니다. 이전 인덱스에서 선택하지 않은 경우(d[i-1][0])와, 이전 인덱스에서 아랫쪽에서 선택한 경우(d[i-1][2]) 중에서 더 큰 값을 현재 값(a[i][0])과 더해줍니다.

9. 현재 인덱스에서 아랫쪽에서 선택한 경우의 최대값을 d[i][2]에 저장합니다. 이전 인덱스에서 선택하지 않은 경우(d[i-1][0])와, 이전 인덱스에서 윗쪽에서 선택한 경우(d[i-1][1]) 중에서 더 큰 값을 현재 값(a[i][1])과 더해줍니다.

10. n번째 인덱스에서 가능한 3가지 경우의 최대값을 찾아서 ans에 저장합니다.

11. ans를 출력합니다.

Uploaded by N2T