99클럽 23번째 TIL(파이썬 - 약수의 합, 연속된 부분 수열의 합)

99클럽 23번째 TIL

비기너 문제 - 약수의 합

문제 설명
정수 n을 입력받아 n의 약수를 모두 더한 값을 리턴하는 함수, solution을 완성해주세요.

제한 사항
n은 0 이상 3000이하인 정수입니다.
입출력 예
n return
12 28
5 6
입출력 예 설명
입출력 예 #1
12의 약수는 1, 2, 3, 4, 6, 12입니다. 이를 모두 더하면 28입니다.

입출력 예 #2
5의 약수는 1, 5입니다. 이를 모두 더하면 6입니다.

정답

def solution(n):
    answer = 0
    for i in range(1,n+1):
        if n%i == 0:
            answer += i
    return answer
풀이
n만큼 반복문을 돌면서 나머지가 0인것을 찾아서 더하면 된다.

 

미들러 문제 - 연속된 부분 수열의 합

문제 설명
비내림차순으로 정렬된 수열이 주어질 때, 다음 조건을 만족하는 부분 수열을 찾으려고 합니다.

기존 수열에서 임의의 두 인덱스의 원소와 그 사이의 원소를 모두 포함하는 부분 수열이어야 합니다.
부분 수열의 합은 k입니다.
합이 k인 부분 수열이 여러 개인 경우 길이가 짧은 수열을 찾습니다.
길이가 짧은 수열이 여러 개인 경우 앞쪽(시작 인덱스가 작은)에 나오는 수열을 찾습니다.
수열을 나타내는 정수 배열 sequence와 부분 수열의 합을 나타내는 정수 k가 매개변수로 주어질 때, 위 조건을 만족하는 부분 수열의 시작 인덱스와 마지막 인덱스를 배열에 담아 return 하는 solution 함수를 완성해주세요. 이때 수열의 인덱스는 0부터 시작합니다.

제한사항
5 ≤ sequence의 길이 ≤ 1,000,000
1 ≤ sequence의 원소 ≤ 1,000
sequence는 비내림차순으로 정렬되어 있습니다.
5 ≤ k ≤ 1,000,000,000
k는 항상 sequence의 부분 수열로 만들 수 있는 값입니다.
입출력 예
sequence k result
[1, 2, 3, 4, 5] 7 [2, 3]
[1, 1, 1, 2, 3, 4, 5] 5 [6, 6]
[2, 2, 2, 2, 2] 6 [0, 2]
입출력 예 설명
입출력 예 #1

[1, 2, 3, 4, 5]에서 합이 7인 연속된 부분 수열은 [3, 4]뿐이므로 해당 수열의 시작 인덱스인 2와 마지막 인덱스 3을 배열에 담아 [2, 3]을 반환합니다.

입출력 예 #2

[1, 1, 1, 2, 3, 4, 5]에서 합이 5인 연속된 부분 수열은 [1, 1, 1, 2], [2, 3], [5]가 있습니다. 이 중 [5]의 길이가 제일 짧으므로 해당 수열의 시작 인덱스와 마지막 인덱스를 담은 [6, 6]을 반환합니다.

입출력 예 #3

[2, 2, 2, 2, 2]에서 합이 6인 연속된 부분 수열은 [2, 2, 2]로 3가지 경우가 있는데, 길이가 짧은 수열이 여러 개인 경우 앞쪽에 나온 수열을 찾으므로 [0, 2]를 반환합니다.

정답

def solution(sequence, k):
    start = 0
    end = 0
    current_sum = sequence[0]
    min_length = float('inf')  # 최소 길이를 무한대로 초기화
    answer = []

    while start < len(sequence):
        if current_sum == k:
            if end - start + 1 < min_length:
                min_length = end - start + 1
                answer = [start, end]
            if end < len(sequence) - 1:
                end += 1
                current_sum += sequence[end]
            else:
                current_sum -= sequence[start]
                start += 1
        elif current_sum < k:
            if end < len(sequence) - 1:
                end += 1
                current_sum += sequence[end]
            else:
                break
        else:  # current_sum > k
            current_sum -= sequence[start]
            start += 1

    return answer if min_length != float('inf') else []

풀이
일단 k보다 수열의 합이 클때, 같을 때, 작을 때 세부분으로 나누어서 가정한다.

그리고 같다면 이 수열의 길이를 min_length와 비교하여 더 작다면 교체한다.

만약 k보다 작다면 수열의 길이를 더 늘려준다.

만약 k보다 크다면 시작부분을 빼주고, 시작 부분 인덱스를 증가시켜 준다.