mooni
[JavaScript] 연속된 부분 수열의 합 본문
Q. 문제
비내림차순으로 정렬된 수열이 주어질 때, 다음 조건을 만족하는 부분 수열을 찾으려고 합니다.
- 기존 수열에서 임의의 두 인덱스의 원소와 그 사이의 원소를 모두 포함하는 부분 수열이어야 합니다.
- 부분 수열의 합은 k입니다.
- 합이 k인 부분 수열이 여러 개인 경우 길이가 짧은 수열을 찾습니다.
- 길이가 짧은 수열이 여러 개인 경우 앞쪽(시작 인덱스가 작은)에 나오는 수열을 찾습니다.
수열을 나타내는 정수 배열 sequence와 부분 수열의 합을 나타내는 정수 k가 매개변수로 주어질 때, 위 조건을 만족하는 부분 수열의 시작 인덱스와 마지막 인덱스를 배열에 담아 return 하는 solution 함수를 완성해주세요. 이때 수열의 인덱스는 0부터 시작합니다.
* 문제풀이흐름
1. answer는 정답 부분 수열의 시작과 끝의 index를 저장할 배열, prefix는 누적합을 저장할 배열, minL은 정답인 부분 배열의 길이
2. prefix에 부분 수열의 합을 모두 저장, O(n)의 시간 복잡도
3. 부분 수열의 시작과 끝을 가리키는 투 포인터 선언
4. 현재 구간 합을 계산하고 k와 동일하면 수열의 길이를 계산해 이전 답보다 짧다면 정답으로 갱신
5. k보다 작다면 right 증가, 크다면 left 증가
A. 풀이
function solution(sequence, k) {
var answer = []
//누적합을 저장할 배열
//prefix[i] === sequence[0] + sequencce[1] + ... + sequence[i-1]
let prefix = [0]
let minL = Infinity
//prefix에 부분 수열의 합을 모두 저장
sequence.forEach((num, i)=>{
prefix.push(num + prefix[i])
})
//부분 수열의 시작과 끝을 가리키는 투 포인터
let left = 0
let right = 0
while(left <= right) {
//현재 구간의 합
let sum = prefix[right] - prefix[left]
//현재 구간의 합이 k일 경우
if(sum === k) {
//현재 수열의 길이
let nowL = right - 1 - left
//여태까지의 찾아낸 부분 수열의 길이보다 작다면 갱신 후 정답 index 저장
if(minL > nowL) {
minL = nowL
answer = [left, right-1]
}
}
//sum이 k보다 작으면 끝 인덱스 증가, k보다 크면 시작 인덱스 증가
if(sum < k){
right++
} else left++
}
return answer;
}
'Coding Test' 카테고리의 다른 글
[JavaScript] 과제 진행하기 (0) | 2025.05.26 |
---|---|
[JavaScript] 두 원 사이의 정수 쌍 (0) | 2025.05.19 |
[JavaScript] 요격 시스템 (0) | 2025.05.11 |
[JavaScript] PCCP 기출문제 2번 - 충돌위험 찾기 (0) | 2025.04.05 |
[JavaScript] PCCP 기출문제 2번 - 퍼즐 게임 챌린지 (0) | 2025.04.02 |