관리 메뉴

TEAM EDA

[엘리스] 가로수 본문

EDA Study/알고리즘

[엘리스] 가로수

김현우 2020. 3. 29. 00:08

가로수

직선으로 되어있는 도로의 한 편에 가로수가 임의의 간격으로 심어져 있습니다. KOI 시에서는 가로수들이 모두 같은 간격이 되도록 가로수를 추가로 심는 사업을 추진하고 있습니다. KOI 시에서는 예산 문제로 가능한 한 가장 적은 수의 나무를 심고 싶습니다. 편의상 가로수의 위치는 기준점으로부터 떨어져 있는 거리로 표현되며, 가로수의 위치는 모두 양의 정수입니다. 예를 들어, 가로수가 (1, 3, 7, 13)의 위치에 있다면 (5, 9, 11)의 위치에 가로수를 더 심으면 모든 가로수들의 간격이 같게 됩니다. 또한, 가로수가 (2, 6, 12, 18)에 있다면 (4, 8, 10, 14, 16)에 가로수를 더 심어야 합니다. 심어져 있는 가로수의 위치가 주어질 때, 모든 가로수가 같은 간격이 되도록 새로 심어야 하는 가로수의 최소수를 구하는 프로그램을 작성하세요. 단, 추가되는 나무는 기존의 나무들 사이에만 심을 수 있습니다.

입력

첫째 줄에는 이미 심어져 있는 가로수의 수를 나타내는 하나의 정수N이 주어진다(3≤N≤100,000). 둘째 줄부터 N개의 줄에는 각 줄마다 심어져 있는 가로수의 위치가 양의 정수로 주어지며, 가로수의 위치를 나타내는 정수는 100,000,000 이하이다. 가로수의 위치를 나타내는 정수는 모두 다르다.

출력

모든 가로수가 같은 간격이 되도록 새로 심어야 하는 가로수의 최소수를 첫 번째 줄에 출력합니다.

입력 예시

4 1 3 7 13 

출력 예시

3

풀이

  • 최대공약수를 활용한 풀이
  • 간격의 최대공약수를 구하고 가득 채웠을 때의 거리에서 있는 길이를 빼줌
    • (max(myInput) - min(myInput)) / GCD - len(myInput) + 1
    • 1을 더해주는 이유는 끝의 자리가 거리에서 계산이 되지 않았기 때문
def gcd(x, y): 
    if x % y == 0: 
        return y
    else:
        return gcd(y, x % y)

def howManyTree(n, myInput) :
    '''
    모든 가로수가 같은 간격이 되도록 새로 심어야 하는 가로수의 최소수를 리턴하는 함수를 구현하세요.
    '''
    diff = []
    myInput = sorted(myInput)
    for i in range(n-1):
        diff.append(myInput[i+1] - myInput[i])

    GCD = gcd(diff[0], diff[1])
    diff = diff[2:]
    for i in range(len(diff)):
        GCD = gcd(GCD, diff[i])

    cnt =  (max(myInput) - min(myInput)) / GCD - len(myInput) + 1
    return int(cnt)

def main():
    '''
    수정하시면 안됩니다.
    '''
    n = int(input())
    myInput = []
    for _ in range(n) :
        myInput.append(int(input()))

    print(howManyTree(n, myInput))
if __name__ == "__main__":
    main()

 

출처 : 엘리스 아카데미, https://academy.elice.io/learn

'EDA Study > 알고리즘' 카테고리의 다른 글

[엘리스] 최대 공통 부분 수열  (0) 2020.03.29
[엘리스] N-Queen  (0) 2020.03.29
[알고리즘] 최대공약수 구하기  (0) 2020.03.28
[알고리즘] Quick sort (퀵 정렬)  (0) 2020.03.28
[엘리스] 점토 놀이  (0) 2020.03.28