programmers_연속된 부분 수열의 합_java

2023. 4. 28. 11:06Algorithm/Programmers

728x90

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

 

프로그래머스

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

programmers.co.kr

class Solution {
    public int[] solution(int[] sequence, int k) {
        int[] answer = new int[2];
        
        int head = 0;
        int tail = 0;
        int sum = sequence[0];
        int leng = 987654321;
        
        while(tail < sequence.length){
            if(sum == k){ // 일치 -> tail 이동
                if(leng > tail - head){ // 길이가 짧은 것 우선
                    leng = tail - head;
                    answer[0] = head;
                    answer[1] = tail;                    
                }
                if(tail == sequence.length-1) break; // tail이 인덱스 범위를 초과하는 경우 처리
                sum += sequence[++tail];
            }else if(sum > k){ // sum이 더 큰경우 -> head 이동
                sum -= sequence[head++];
            }else if(sum < k){ // sum이 더 작은경우 -> tail 이동
                if(tail == sequence.length-1) break;
                sum += sequence[++tail];
            }
        }
        
        return answer;
    }
}