[74일차][Lv2][프로그래머스][누적합][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의 부분 수열로 만들 수 있는 값입니다.

 

입출력

sequence	k	return
[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]

 

코드 및 설명

def solution(sequence, k):
    len_sequence = len(sequence)
    sorted_sequence = sorted(sequence)
    check_arr = []
    check_break = False    
       
    for idx1 in range(len_sequence):
        if not check_break:
            for idx2 in range(len_sequence-idx1):               
                if sum(sorted_sequence[idx2:idx1+idx2+1]) == k:
                    check_arr.append([idx2, idx1+idx2])
                    check_break = True
        else:
            break          
            
    return check_arr[0]

 

이 문제는 누적합 문제이다(누적합으로 안되서 투포인트 알고리즘으로...). 정해진 배열에서 1자리수부터 정해진 배열의 자리수까지 더해가며 k의 값과 같은 수를 찾고 그 수의 자리 값을 배열로 return하면 되는 것이다. 위에 코드를 짰으나... 속도가 안나온다. sum혹은 이중 for문 문제인 것 같은데... 흠........ 일단 좀 더 생각을 해봐야겠다...