실버V. 백준 1059 좋은구간

https://www.acmicpc.net/problem/1059

1059번: 좋은 구간

 

처음으로 파이썬으로 풀어본 문제.

 

1. 수학식을 구해야 한다.

result = (contain_number - left) * (right - contain_number + 1) + right - contain_number

 

문제를 해결 한 뒤 여러 블로그를 뒤져본 결과 저와 비슷하게 푼 블로그가 나와있어 공유합니다.

https://blog.naver.com/PostView.nhn?blogId=repeater1384&logNo=222278285158 

[파이썬]백준 1059번: 좋은 구간

 

2. 예외처리

문제에서 포함숫자가 배열 숫자보다 작을 수도 있다는 점을 알아야 합니다.

그렇기에 [0] 인자를 추가하여 정렬해 예외에 대한 처리도 해줘야합니다.

 

import sys

def getGoodSection(numbers, contain_number):
    result = 0

    if contain_number in numbers:
        print(result)
        return

    numbers = [0] + numbers
    numbers.sort()

    for idx in range(0, len(numbers) - 1):
        left = numbers[idx]
        right = numbers[idx + 1]
        if contain_number in range(left, right):
            break

    left = left + 1
    right = right - 1
    result = (contain_number - left) * (right - contain_number + 1) + right - contain_number

    print(result)

if __name__ == '__main__':
    L = int(sys.stdin.readline())
    numbers = list(map(int, sys.stdin.readline().split()))
    contain_number = int(sys.stdin.readline())
    getGoodSection(numbers, contain_number)