본문 바로가기

Java/프로그래머스

[JAVA] 프로그래머스 - 연속된 수의 합

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

 

프로그래머스

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

programmers.co.kr

 

 

 

 

 

 

정답 

 

import java.util.*;

class Solution {
    public int[] solution(int num, int total) {
        ArrayList<Integer> list = new ArrayList<>();
        
        for(int i = -1000; i <= 1000; i++) {
            
            int sum = 0;
            
            // 연속된 num개의 수를 더한다. 
            for(int j = i; j < i + num; j++)
                sum += j;
            
            
            if(sum == total) {
                for(int j = i; j < i + num; j++)
                    list.add(j);
            }
        }
        
        int[] answer = new int[list.size()];
        for(int i = 0; i < list.size(); i++)
            answer[i] = list.get(i);
        
        return answer;
    }
}

 

 

다른 사람의 풀이

 

class Solution {
        public int[] solution(int num, int total) {
                int[] answer = new int[num];
                int check = num*(num+1) / 2;
                int start = (total - check) / num + 1;
                for (int i = 0; i < answer.length; i++) {
                        answer[i] = start + i ;
                }
                return answer;
        }
}