백준 2851번 슈퍼마리오

문제

슈퍼 마리오 앞에 10개의 버섯이 일렬로 놓여져 있다. 이 버섯을 먹으면 점수를 받는다.

슈퍼 마리오는 버섯을 처음부터 나온 순서대로 집으려고 한다. 하지만, 모든 버섯을 집을 필요는 없고 중간에 중단할 수 있다. 중간에 버섯을 먹는 것을 중단했다면, 그 이후에 나온 버섯은 모두 먹을 수 없다. 따라서 첫 버섯을 먹지 않았다면, 그 이후 버섯도 모두 먹을 수 없다.

마리오는 받은 점수의 합을 최대한 100에 가깝게 만들려고 한다.

버섯의 점수가 주어졌을 때, 마리오가 받는 점수를 출력하는 프로그램을 작성하시오.

입력

총 10개의 줄에 각각의 버섯의 점수가 주어진다. 이 값은 100보다 작거나 같은 양의 정수이다. 버섯이 나온 순서대로 점수가 주어진다.

출력

첫째 줄에 마리오가 받는 점수를 출력한다. 만약 100에 가까운 수가 2개라면 (예: 98, 102) 마리오는 큰 값을 선택한다.

예제 입력 1 복사

10
20
30
40
50
60
70
80
90
100

예제 출력 1 복사

100

예제 입력 2 복사

1
2
3
5
8
13
21
34
55
89

예제 출력 2 복사

87

 

정수를 한줄에 하나를 입력받을 때는 1차원 배열로 받는아래 코드가 더 유용하다!

import sys
# 2차원 배열생성
a=[]
for i in range(n):
	a.append(list(map(int,sys.stdin.readline().split())))
# 1차원 배열생성
for i in range(n):
	n.append(int(sys.stdin.readline().strip()))

정답 코드1:

import sys
a=[]
sum=0

for i in range(10):
        a.append(list(map(int,sys.stdin.readline().split())))
for i in range(10):
        sum+=a[i][0]
        if sum>=100 and abs(100-(sum-a[i][0]))<abs(100-sum):
            print(sum-a[i][0])
            break
        elif sum>=100 and  abs(100-(sum-a[i][0]))>=abs(100-sum):
            print(sum)
            break
        else:
            continue
if sum<100:
    print(sum)

 

개선한 정답 코드2:

import sys
n=[]
s=0
for i in range(10):
    n.append(int(sys.stdin.readline().strip()))

for j in range(10):
    s+=n[j]
    if s>=100 and abs(100-s)<=abs(100-(s-n[j])):
        print(s)
        break
    elif s>=100 and abs(100-s)>abs(100-(s-n[j])):
        print(s-n[j])
        break
    else:
        continue
if s<100:
    print(s)