ultra_dev
연속부분수열 본문
4. 연속 부분수열
설명
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
📌 lt rt 개념, lt 하나만 빼는게 아니고 while문으로 작아질때까지 빼기
import java.util.*;
import java.util.Scanner;
class Main {
public int solution(int n, int m, int[] arr) {
//left(lt) right(rt) 이용.. lt부터 rt까지 더한다 생각
int answer = 0, sum = 0, lt = 0;
//rt는 바로 for문에서
for (int rt = 0; rt < n; rt++) {
sum += arr[rt];
if (sum == m) answer++;//rt값을 더하면 m이랑 값 같은지 바로 확인. 증가하고 더하고 확인하고 루틴
while (sum >= m) { //그리고 이제 m이랑 같거나 커지면 rt는 멈추고 m보다 작아질 때까지 lt 하나씩 제거해야지
//빼고 그 다음에 lt ++
sum -= arr[lt++]; //1,1,1,1,5이고 m이 6이면 맨앞에꺼 하나 제거한다고 m보다 작아지는게 아니니 while문으로
if (sum == m) {
answer++; //그리고 이제 또 빼면서 m이랑 같아질 경우도 answer++ 해줘야지 다른종류의 연속부분수열이니
}
}
}
return answer;
}
public static void main(String[] args) {
Main T = new Main();
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
int m = sc.nextInt();
int[] arr = new int[n];
for (int i = 0; i < n; i++) {
arr[i] = sc.nextInt();
}
System.out.print(T.solution(n, m, arr));
}
}
'알고리즘' 카테고리의 다른 글
최대 길이 연속 부분수열 (0) | 2023.02.03 |
---|---|
연속된 자연수의 합(two pointers + 수학) (0) | 2023.02.03 |
최대매출(Sliding window) (0) | 2023.02.03 |
공통 원소 구하기(two pointers) (0) | 2023.01.29 |
두 배열 합치기(two pointers) (2) | 2023.01.29 |
Comments