알고리즘

[JS] 알고리즘 기초 정복 5-3번_연속 부분수열 1

햄❤️ 2021. 7. 20. 11:28
반응형

문제 5-3번

N개의 수로 이루어진 수열이 주어집니다. 이 수열에서 연속부분수열의 합이 특정숫자 M이 되는 경우가 몇 번 있는지 구하는 프로그램을 작성하세요. 만약 N=8, M=6이고 수열이 다음과 같다면 1 2 1 3 1 1 1 2 합이 6이 되는 연속부분수열은 {2, 1, 3}, {1, 3, 1, 1}, {3, 1, 1, 1}로 총 3가지입니다.

 

🍄 입력설명

첫째 줄에 N(1≤N≤100,000), M(1≤M≤100,000,000)이 주어진다. 수열의 원소값은 1,000을 넘지 않는 자연수이다.

 

🍄 출력설명

첫째 줄에 경우의 수를 출력한다.

 

🍄 입력예제 1

8 6

1 2 1 3 1 1 1 2

 

🍄 출력예제 1

3

 

 


정답 코드

투포인터 알고리즘

function solution(m, array) {
	let answer = 0;
	let lt = 0, sum = 0;
	for (let rt = 0; rt < arr.length; rt++) {
		sum += arr[rt];
		if (sum === m) answer++;
		while (sum >= m) {
		//sum이 m보다 크거나 같으면 while문 반복, sum이 m보다 작아지면 멈춤
			sum -= arr[lt++]; //sum에서 lt값을 빼주고, lt를 증가시킴
			if (sum === m) answer++;
		}
	}
	return answer;
}

let arr = [1, 2, 1, 3, 1, 1, 1, 2];
console.log(solution(6, arr));
//3출력

 

728x90
반응형