Programming/Python
[이코테] 그리디 알고리즘 - 곱하기 혹은 더하기
한우콩
2021. 7. 6. 18:03
각 자리가 숫자(0부터 9)로만 이루어진 문자열 S가 주어졌을 때, 왼쪽부터 오른쪽으로 하나씩 모든 숫자를 확인하며 숫자 사이에 'x' 혹은 '+' 연산자를 넣어 결과적으로 만들어질 수 있는 가장 큰 수를 구하는 프로그램을 작성하세요.
단, +보다 x를 먼저 계산하는 일반적인 방식과는 달리, 모든 연산은 왼쪽에서부터 순서대로 이루어진다고 가정합니다.
예를 들어 02984라는 문자열이 주어지면, 만들어질 수 있는 가장 큰 수는 ((((0 + 2) x 9) x 8) x 4) = 576입니다.
또한 만들어질 수 있는 가장 큰 수는 항상 20억 이하의 정수가 되도록 입력이 주어집니다.
📌 입력 조건
- 첫째 줄에 여러 개의 숫자로 구성된 하나의 문자열 S가 주어집니다. (1 <= S <= 20)
📌 출력 조건
- 첫째 줄에 만들어질 수 있는 가장 큰 수를 출력합니다.
📌 입력 예시
02984
567
📌 출력 예시
576
210
📌 풀이
0과 1은 더했을 때 더 큰수가 나온다. 문자열의 첫번째 문자로 0 또는 1이 올 수 있으므로 미리 result변수에 할당하고
result와 각 문자를 받는 num변수가 1보다 작거나 같으면 result에 덧셈을 하고, 아니라면 곱셈을 한다.
s = input()
result = int(s[0])
for i in range(1, len(s)):
num = int(s[i])
if num <= 1 or result <= 1:
result += num
else:
result *= num
print(result)