[python3] 이코테 - 곱하기 혹은 더하기 (ch.11 그리디 - 유형별 기출문제)

곱하기 혹은 더하기 (ebook기준 p.315)

- 각 자리가 숫자(0~9)로 이루어진 문자열 S 가 주어짐
- 왼쪽부터 오른쪽으로 하나씩 모든 숫자를 확인하여 숫자 사이에 X 또는 + 연산자를 넣어 결과적으로 가장 큰 수로 만들어내고 싶다.

##### 입력예시
- 02984가 주어지면 (0+2)*9*8*4 로 최대값 576이 출력
- 최대값이 20억 이하의 정수가 되도록 입력을 줄것이다
- 첫쨋줄 입력으로 숫자로 구성된 하나의 문자열 S가 주어짐 (1 <= s길이 <= 20)

 

<내가 푼 코드>

정상적으로 출력이 되긴하지만, 답안코드를 보고 더 간결하게 구현하는 방법을 알아냈다. 내가 생각이 조금 짧았던 탓에 조금 코드가 길어진 것 같다. 한번만 더 생각했다면 같은 구현방법에 길이가 짧은 코드를 구현할 수 있었을텐데.. 역시나 아직 부족한점이 많다. 답안코드를 참고해서 고친 나의 코드를 아래 붙여넣겠다.

s = '00000910'

def solution(s):
    s = list(map(int, s))
    
    first_num = s.pop(0)
    while s:
        num = s.pop(0)
        # 0이나 1 이면 곱하지말고 더해줘야함
        if num <= 1:
            first_num += num
            
        # 그 이상의 숫자라면
        else:
            #처음 연속 0이 두번이상나올수도 있으니
            if first_num == 0 or first_num == 1:
                first_num += num
            else:
                first_num *= num
        
    return first_num

solution(s) 

 

 

<답안 코드와 비교하고 새롭게 고친 코드>

더 간결하기 때문에 시간복잡도 면에서 더 효율이 높은 코드가 될 것이다.

s = '01212'

def solution(s):
    s = list(map(int, s))
    
    first_num = s.pop(0)
    while s:
        num = s.pop(0)
        # first_num과 num 둘중 하나라도 0이거나 1이면 더해줌
        if first_num <= 1 or num <= 1:
            first_num += num
        # 그렇지 않다면 곱해줌
        else:
            first_num *= num

    return first_num

solution(s) 

다양한 케이스를 넣어서 테스트를 해보았는데, 올바른 출력이 가능하다.