연속된 인덱스들의 합이 K와 같으면 되기에
합이 K보다 크면 마지막인덱스를 뒤로 보내고
합이 K보다 작다면 시작인덱스를 뒤로 보내는 식으로 풀이했다.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
public class 연속된부분수열의합 {
public static void main(String[] args) {
int[] sequence = {1,1,1,2,3,4,5};
int k = 7;
solution(sequence, k);
}
static int[] solution(int[] sequence, int k) {
int[] answer = new int[2];
int startIndex=0; //시작 Index값을 담는다
int endIndex=0; //마지막 Index값을 담는다.
int sum=sequence[startIndex]; //Index사이 값들의 합, list첫 값으로 초기화
int min=Integer.MAX_VALUE; //제일 작은 값을 넣어줘야하기에 Integer최댓값으로 초기화
//마지막 Index가 요청값 list 길이를 넘어버리면 반복문을 종료한다.
while(endIndex<sequence.length) {
//합계가 K와 같고, 두 Index사이 거리가 최솟값보다 작다면 answer에 넣어준다.
if(sum==k && (endIndex-startIndex)<min) {
answer[0] = startIndex;
answer[1] = endIndex;
min = endIndex-startIndex;
}
//합계가 K보다 크다면 시작Index를 앞으로 땡기고 합계해서 빼준다.
if(sum>k) {
sum-=sequence[startIndex];
startIndex++;
}
else { //합계가 K보다 작다면 마지막Index를 앞으로 땡기고 합에 더해준다.
endIndex++;
if(endIndex!=sequence.length) {
sum+=sequence[endIndex];
}
}
}
return answer;
}
}