알고리즘

프로그래머스 연속된 부분 수열의 합 (JAVA)

박카스마시며코딩 2023. 7. 18. 21:59

https://school.programmers.co.kr/learn/courses/30/lessons/178870

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

저는 투 포인트를 이용해 문제를 해결하였습니다.

투 포인트를 통해 그 사이의 합을 구하고 그 값이 k라면 answer과 비교하여 answer를 초기화해주었습니다.

 

class Solution {
    public int[] solution(int[] sequence, int k) {
        int[] answer = null;
        int size = sequence.length;
        int startIndex = 0;
        int endIndex = 0;
        int sum = 0;
        
        while(true){
            if(sum < k){
                if(endIndex >= size){
                    break;
                }
                sum += sequence[endIndex];
                endIndex++;
            }else {
                if(sum == k && (answer == null || endIndex-1 - startIndex < answer[1] - answer[0]) ){
                    answer = new int[]{startIndex, endIndex-1};
                }
                sum -= sequence[startIndex];
                startIndex++;
            }
        }
        return answer;
    }
}