글 읽기 - 1149 RGB 거리 문제입니다(파이썬)

dlwhdgh98 2년 전 0

제가 코드를 짜봤는데요....

분명 테스트 케이스나 다른것들은 되는데 왜 틀렸다고 뜰까요???

혹시 반례가 있을까요??

전 입력을 받으면 1번쨰 집을 기준으로

1) 1번쨰 집이 빨강인 경우 최솟값

2) 1번째 집이 초록인 경우 최솟값

3) 1번째 집이 파랑인 경우 최솟값들을 result라는 리스트에 넣어서

그 result의 min을 이용해서 출력하는 형식으로 구현했는데 왜 채점하면 틀렸다고 뜨는 걸까요??ㅜㅜ 아무리 생각해도 이유를 모르겠네요.....

n = int(input())

cost = []
result = []
for _ in range(n):
    cost.append(list(map(int, input().split())))

for i in range(3):
    sum = cost[0][i] # 첫번째 집의 색깔.
    k = i # 그 인덱스
    for j in range(1, n):
        if k == 0:
            a = min(cost[j][1:]) # 그 전 인덱스가 R이면 현재G와 B중에 최솟값 저장.
            sum += a
            k = cost[j].index(a) # 그 최솟값의 인덱스 저장.
        elif k == 2:
            a = min(cost[j][:2])
            sum += a
            k = cost[j].index(a)
        else:
            a = min(cost[j][0], cost[j][2])
            sum += a
            k = cost[j].index(a)
    result.append(sum)
print(min(result))

djm03178 2년 전 1

첫 번째 집의 색을 정한 후 이후의 전략을 다른 두 색 중 최솟값으로 지정하는 것이 최적이 되는 것은 아닙니다. 아래 예시를 생각해 보세요.

4
1 1 1
1 1 1
2 1 2
9 1 9

정답: 5
이 코드의 출력: 12

dlwhdgh98 2년 전 0

아 index함수를 사용하는것도 잘못됐었네요ㅜㅜ 감사합니다!!

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

1149 RGB 거리 문제입니다(파이썬)

dlwhdgh98 2년 전 0

djm03178 2년 전 1

dlwhdgh98 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말