백준 1541 수 정렬하기 - 잃어버린 괄호 (Python / 파이썬)

링딩딩 코딩딩·2023년 5월 28일
0

Algorithm

목록 보기
1/5
post-thumbnail

문제 설명
https://www.acmicpc.net/problem/1541

1541번: 잃어버린 괄호

첫째 줄에 식이 주어진다. 식은 ‘0’~‘9’, ‘+’, 그리고 ‘-’만으로 이루어져 있고, 가장 처음과 마지막 문자는 숫자이다. 그리고 연속해서 두 개 이상의 연산자가 나타나지 않고, 5자리보다

문제

세준이는 양수와 +, -, 그리고 괄호를 가지고 식을 만들었다. 그리고 나서 세준이는 괄호를 모두 지웠다.

그리고 나서 세준이는 괄호를 적절히 쳐서 이 식의 값을 최소로 만들려고 한다.

괄호를 적절히 쳐서 이 식의 값을 최소로 만드는 프로그램을 작성하시오.

입력

첫째 줄에 식이 주어진다. 식은 ‘0’~‘9’, ‘+’, 그리고 ‘-’만으로 이루어져 있고, 가장 처음과 마지막 문자는 숫자이다. 그리고 연속해서 두 개 이상의 연산자가 나타나지 않고, 5자리보다 많이 연속되는 숫자는 없다. 수는 0으로 시작할 수 있다. 입력으로 주어지는 식의 길이는 50보다 작거나 같다.

접근 방법

식의 값이 가장 최소가 되려면 음수값에 대해서 가장 최댓값이 되어야한다. 왜냐하면 조건 중 "연산은 오로지 +와 -로 이루어져있기 때문에 -값을 제외하고 전부 괄호로 묶어 부분집합을 만들어주면 된다.

로직은 생각보다 간단하고 나만그렇게 생각한것이 아닌것 같아 아래의 블로그 코드를 참고하여 문제를 solve하였다. 코드 리뷰를 하던 중 어쩌면 Python의 이점을 활용하여 eval()함수를 사용하면 어떨까 싶었다.

https://pacific-ocean.tistory.com/228

data = x.split('-')
num = []
for i in data:
    cnt = 0
    s = i.split('+')
    for j in s:
        cnt += int(j)
    num.append(cnt)
n = num[0]
for i in range(1, len(num)):
    n -= num[i]
print(n)

또다른 접근

어쨋든, +연산에 대해 for문을 돌기때문에 수식이 길어질수록 비효율적이라고 생각이 들었다. 그래서 eval 함수를 추가하였더니 속도측면에서 개선이 되었다!!

data = input().split('-')
num = []
for i in data:
    cnt = 0
    try: 
        cnt = eval(i)
    except:
        s = i.split('+')
        
        for j in s:
            cnt += int(j)
    num.append(cnt)
    
n = num[0]
for i in range(1, len(num)):
    n -= num[i]
print(n)

try문이 추가되었음에도 44m/s --> 40m/s 로 개선되었다. 하지만 "eval함수가 과연 무조건 빠를까?"라고하면 아니였다.

a = [55,-40,30]

start_time = time.time()

sum(a)

end_time = time.time()
print("time : ", end_time - start_time)

start_time = time.time()

eval("55-40+30")

end_time = time.time()
print("time : ", end_time - start_time)

# time :  6.103515625e-05
# time :  7.414817810058594e-05

다음과 같을 때 eval로 수식계산을 했을 때 더 오래걸린 것을 확인할 수 있다. 하지만 또 수식이 너무 짧아서 그런것이 아닐까 싶은 생각에 실험을 해보았는데 굉장히 신기한 결과가 나왔다.

a = [55,-40,30,20]

start_time = time.time()

sum(a)

end_time = time.time()
print("time : ", end_time - start_time)

start_time = time.time()

eval("55-40+30+20")

end_time = time.time()
print("time : ", end_time - start_time)

# time :  0.000164031982421875
# time :  0.00010967254638671875

즉, 수식이 길어질수록 eval함수를 사용하는 것이 시간효율적이라는 것이다. 문제를 잘 읽고 상세한 설정에 따라 코드 구현 전략을 잘 세우면 시간 복잡도를 이렇게 함수추가로써도 해결할 수 있다.

결론 : eval함수를 사용하여 수식을 해결하는 것이 시간효율적이다.


번외

왜 try문을 사용하여 eval을 사용하는가? --> "0009"같은 숫자는 eval함수가 올바르게 받지 못한다. 따라서 가능하다면 eval쓰고 안되면 원래대로 풀도록 유도하였다. 여기서 또 드는 의문은 "그럼 0009같은 숫자가 있음에도 내 알고리즘은 기존의 알고리즘보다 빠르게 작동할까?" 정답은 No다.

예제를 "00009-00009"로 했을 때의 결과는 0.0005357 --> 0.0005929이다. 그렇다면 수식이 긴 case를 제외하고 어쩌면 0으로 먼저 시작하는 숫자의 case가 매우 적었을 가능성도 존재한다. 따라서 데이터의 구성을 생각하며 코드를 구현해야한다.

0개의 댓글