연속된 부분 수열의 합

Lee1231234·2023년 4월 12일

코딩테스트

목록 보기
40/95

비내림차순으로 정렬된 수열이 주어질 때, 다음 조건을 만족하는 부분 수열을 찾으려고 합니다.
기존 수열에서 임의의 두 인덱스의 원소와 그 사이의 원소를 모두 포함하는 부분 수열이어야 합니다.
부분 수열의 합은 k입니다.
합이 k인 부분 수열이 여러 개인 경우 길이가 짧은 수열을 찾습니다.
길이가 짧은 수열이 여러 개인 경우 앞쪽(시작 인덱스가 작은)에 나오는 수열을 찾습니다.
수열을 나타내는 정수 배열 sequence와 부분 수열의 합을 나타내는 정수 k가 매개변수로 주어질 때, 위 조건을 만족하는 부분 수열의 시작 인덱스와 마지막 인덱스를 배열에 담아 return 하는 solution 함수를 완성해주세요. 이때 수열의 인덱스는 0부터 시작합니다.

비내림차순은 오름차순이라는 말과 같다.
투 포인터 문제로 k값보다 커지면 값을 줄이면서 왼쪽 포인터의 값을 올리고
k값보다 작다면 오른쪽 포인터의 값을 올리고 값을 늘리면된다.
만약 k값과 일치한다면 size를 계산해 수열의 크기가 가장 작은 부분을 고를수 있다.
코드

class Solution {
    public int[] solution(int[] sequence, int k) {
        int[] answer = new int [2];
        int[] point = new int[2];
        int sum= sequence[0];
        int size=-1;
      
       
        while(true){      
            if(sum==k){
                if(size==-1||point[1]-point[0]<size){
                    answer[0]= point[0];
                    answer[1]= point[1];
                    size= point[1]-point[0];
                }
            }
            if(point[0]==sequence.length&&point[1]==sequence.length-1) break;
            
            if(sum<=k&&point[1]<sequence.length-1){                
                point[1]+=1;               
                sum+=sequence[point[1]];
            }else{
                sum-=sequence[point[0]];
                point[0]+=1;
               
            }
           
        }
        return answer;
    }
}//투포인터 비내림차순==오름차순
profile
not null

0개의 댓글