[4코1파] #98. LV 2. 연속된 부분 수열의 합

gunny·2023년 4월 11일

4코1파 코딩테스트 코딩테스트 python 프로그래머스 연속된 부분 수열의 합 프로그래머스 연속된 부분 수열의 합 python

코딩테스트

목록 보기

98/530

[4코1파] 4명의 안드로이드 개발자와 1명의 파이썬 개발자의 코딩 테스트 서막 : 4코1파

Rule :

하루에 1문제씩 풀기.
한 문제당 30분씩은 고민하기.
왜 그렇게 풀었는지 공유하기.
하루라도 놓친다면 벌금은 1,000원

START :

[3코1파] 2023.01.04~ (98일차)
[4코1파] 2023.01.13~ (88일차)

Today :

2023.04.11 [98일차]

프로그래머스 LV 2
연속된 부분 수열의 합
https://school.programmers.co.kr/learn/courses/30/lessons/178870

문제 설명

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

기존 수열에서 임의의 두 인덱스의 원소와 그 사이의 원소를 모두 포함하는 부분 수열이어야 합니다.
부분 수열의 합은 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의 부분 수열로 만들 수 있는 값입니다.

입출력 예

입출력 예에 대한 설명

입출력 예 #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]를 반환합니다.

문제 풀이 방법

처음에는 주어진 sequence에 대한 origin index를 만들고,
다음에 sequence를 queue로 만들어서,
큐를 돌면서 조건의 합에 맞게끔 start 와 end를 뽑아내려고 했으니..

물론 안됨 ㅋ
투포인터로 푸는 방법 네티처럼 훔쳐옴..
몰겠네..

내 코드

def solution(sequence, k):
    answer = []
    n = len(sequence)
    limit_sum, end = 0,0
    
    for i in range(len(sequence)):
        while limit_sum < k and end < n:
            limit_sum += sequence[end]
            end +=1
            
        if limit_sum == k:
            answer.append([i, end-1, end-1-i])
        
        limit_sum-= sequence[i]
        
    answer = sorted(answer, key=lambda x: x[2])
    
    return answer[0][:2]

증빙