[Java][프로그래머스] 연속 부분 수열 합의 개수

import java.util.*;

class Solution {
    public int solution(int[] elements) {
        int answer = 0;
        
        HashSet<Integer> set = new HashSet<>();
   
        for(int i=1; i<=elements.length; i++){
            for(int j=0; j<elements.length; j++){
                int sum = 0;
                for(int k=0; k<i; k++){
                    if(j+k > elements.length-1){
                        sum+= elements[j+k-elements.length];
                    }else sum += elements[j+k];
                }
                set.add(sum);
            }
        }
        answer = set.size();
        return answer;
    }
}

 

원형 수열이므로 연속된 수만 더하되 배열의 인덱스를 벗어날때의 처리만 잘해주면된다.

만약 길이가 5인 배열에서 길이가 3인 연속 부분 수열을 구할때

(0,1,2) , (1,2,3) , (2,3,4) , (3,4,0) , (4,0,1) 을 더해주면 되는데 이때 (3,4,0) , (4,0,1) 만 잘 처리해주면된다는 말이다.

 

이를 j+k > elements.length-1 이라는 if문을 통해 처리하였다.

 

여기서 i는 길이가 i인 부분수열을 의미하며 j는 수열이 시작하는 위치를 의미한다.

중복제거를 위하여 HashSet을 사용하였다.