비내림차순으로 정렬된 수열이 주어질 때, 다음 조건을 만족하는 부분 수열을 찾으려고 합니다.
기존 수열에서 임의의 두 인덱스의 원소와 그 사이의 원소를 모두 포함하는 부분 수열이어야 합니다.
부분 수열의 합은 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] |
누가봐도 투 포인터를 써야하는 문제...!
완탐이 아니라 dp를 써보려고 matrix 를 설정해서 시도해보았으나 장렬히 실패
왜그런지 모르겠지만 숫자 입력이 이상하게 되더라구..? 저는 한 element만 입력하고 싶은데 한 column이 다 입력되어부렀어요 for loop 두번 써서 그런가 하기에는 그 전에는.. 괜찮았던거 같은데..? ㅎㅎ... 잘은.. 모르겠지만 아무튼 그런가봐요
def solution(sequence, k):
partial_sum = 0
right = 0
result = []
for left in range(len(sequence)):
#left = left pointer
while right < len(sequence) and partial_sum < k:
#right = right pointer
# while right is smaller than the given sequence
# and while the partial sum is smaller than k (since it is sorted)
partial_sum += sequence[right]
#add the current value of right to the partial sum
right += 1
#and move the right pointer to the right
if partial_sum == k:
#if the partial sum reaches the goal
if not result:
# if result is empty:
result = [left, right - 1]
#assign current pointer pos to the result(the indexes)
else:
# if result is not empty:
result = result if result[1] - result[0] <= (right - 1) - left else [left, right - 1]
#compare the result(old index) with current=new index(left, right)
#if new is smaller : assign that to the result
partial_sum -= sequence[left]
return result