[이것이 코딩 테스트다] 그리디 - 곱하기 혹은 더하기

YEAh·2021년 3월 2일
0
post-thumbnail

그리디
현재 상황에서 지금 당장 좋은 것만 고르는 방법


✅ 문제

각 자리가 숫자(0부터 9)로만 이루어진 문자열 S가 주어졌을 때, 왼쪽부터 오른쪽으로 하나씩 모든 숫자를 확인하며 숫자 사이에 'x' 혹은 '+' 연산자를 넣어 결과적으로 만들어질 수 있는 가장 큰 수를 구하는 프로그램을 작성하세요. 단, +보다 x를 먼저 계산하는 일반적인 방식과는 달리, 모든 연산은 왼쪽에서부터 순서대로 이루어진다고 가정합니다.

입력 예시1

02984

출력 예시1

576

입력 예시2

567

출력 예시2

210


💻 코드

data = input()

result = 0
flag = 0

# 0과 1이 나온 숫자 다음에만 더하기 연산을 한다. 나머지는 곱하기 연산
for i in data:
    # 앞문자가 0 또는 1인지 아닌지 flag가 담고 있기 때문에 flag 값에 따라 연산 수행
    if flag == 0:
        result += int(i)
    else :
        result *= int(i)
    # 문자가 0 또는 1이면 flag = 0, 아니면 flag = 1
    if int(i) == 0 or int(i) == 1:
        flag = 0
    else:
        flag = 1

print(result)

설계

앞의 숫자가 0 또는 1인지 확인하기 위해 flag 변수를 두어 0과 1로 구분한다. 앞의 숫자가 0 또는 1이면 flag 값을 0으로 하여 더하기 연산을 하고, 0 또는 1이 아니면 flag 값을 1로 하여 곱하기 연산을 수행한다.

➕ 다른 풀이

data = input()

# 첫 번째 문자를 숫자로 변경하여 대입
result = int(data[0])

for i in range(1, len(data)):
	# 두 수 중에서 하나라도 '0' 혹은 '1'인 경우, 곱하기보다는 더하기 수행
    num = int(data[i])
    if num <= 1 or result <= 1:
        result += num
    else : 
        result *= num

print(result)
  • 첫번째 숫자를 result 변수에 저장하고 그 다음부터는 for문을 돌면서 현재 숫자가 1이하의 숫자일 경우 더하기 연산을 수행한다. 첫번째 숫자가 0일 경우 처음부터 현재 숫자가 1이하인지 확인하여 연산을 하면 오류가 발생한다. result <= 1if 절에서 같이 확인하여 오류가 발생하지 않게 하였다.
profile
End up being.

0개의 댓글