백준 1541 잃어버린 괄호

1541번: 잃어버린 괄호 (acmicpc.net)

1541번: 잃어버린 괄호

nums = input().split('-')
temp = 0
for i in range(len(nums)):
    if '+' in nums[i]:
        s = map(int, nums[i].split('+'))
        for j in s:
            temp = temp + j
        nums[i] = temp 
    else:
        nums[i] = int(nums[i])
    temp = 0
result = nums[0]
for i in range(1, len(nums)):
    result = result - nums[i]
print(result)

문제 유형이 그리디알고리즘/ 파싱 이었던 문제인데...

키포인트는 얼마나 효율적인! 방법을 선택해내느냐 아는것이었고

그를 위해서는 '-'를 기준으로 split한다는 아이디어가 중요했다...

exp = input().split("-")

ans = 0

for i in exp[0].split("+"):
    ans += int(i)

for i in exp[1:]:
    for j in i.split("+"):
        ans -= int(j)

print(ans)

이건 훨씬 짧은 코드이다.

나는 일일히 숫자들을 죄다 배열에 넣은 다음 마지막에 다 빼주는식으로 했는데

첫 번째 항에 대해서만 따로 더한뒤에 나머지는 빼주면 더 간결

 

그리고 for i in exp[0].split('+')

이렇게하면 훨씬 코드를 축약할 수 있다는 사실..

배열을 1번째 인덱스부터 하고싶으면 for i in exp[1:]: 로 쓸 수 있다는 것두 알아두자.