[프로그래머스] 레벨0 연속된 수의 합

https://school.programmers.co.kr/learn/courses/30/lessons/120923?language=java 

프로그래머스

나는 이 문제를 케이스를 나눠서 풀었다.

일단 total이 num으로 나누어 떨어지는 경우에는 무조건 가운데 수를 알 수 있기 때문에 리턴할 배열의 처음과 끝 수도 쉽게 구할 수가 있었다. 나누어 떨어지지 않을 때에는 num에 대한 홀짝 판별을 한 뒤에 홀수일 경우에는 위의 케이스와 동일하게 처리하고, 짝수일 경우에는 중간값을 모르기 때문에 내림을 사용해서 리턴값을 구할 수 있었다.

---

문제 및 입출력 예제

---

코드

import java.util.ArrayList;

class Solution
{
    public int[] solution(int num, int total)
    {
        ArrayList<Integer> list = new ArrayList<Integer>();
        int start = -1;
        int end = -1;
        
        if(total%num==0)
        {
            start = (total/num) - (num/2);
            end = (total/num) + (num/2);
        }
        else
        {
            if(num%2==1)
            {
                start = (total/num) - (num/2);
                end = (total/num) + (num/2);
            }
            else
            {
                start = (int)(Math.floor(total/num)) - (num/2) + 1;
                end = (int)(Math.floor(total/num)+1) + (num/2) - 1;
            }
        }
        
        for(int i=start; i<=end; i++)
        {
            list.add(i);
        }
        
        int[] answer = new int[num];
        
        for(int i=0; i<list.size(); i++)
        {
            answer[i] = list.get(i);
        }
        
        return answer;
    }
}