프로그래머스 - 다음에 올 숫자

문제 설명

등차수열 혹은 등비수열 common 이 매개변수로 주어질 떄, 마지막 원소 다음으로 올 숫자를 return 하도록 solution 함수를 완성하라.

 

입출력 예

common	result
[1,2,3,4]	5
[2,4,8]	16

 

입출력 예 설명

입출력 예 1

[1,2,3,4] 는 공차가 1인 등차수열이므로 다음에 올 수는 5이다.

 

입출력 예 2

[2,4,8] 은 공비가 2인 등비수열이므로 다음에 올 수는 16이다.

 

---

문제 해결 과정

 

주어진 배열이 등비수열인지 등차 수열인지 판단을 먼저 해야한다.

그러기 위해서는 등비수열은 각 수마다 규칙적인 수로 곱해져있는 수 이며, 등차 수열은 각 수마다 규칙적으로 더해져있는 수열을 의미한다는 것을 알고 있어야 한다.

 

isGeometric 는 주어진 수열이 등비 수열인지 아닌지 알려주는 메소드이다.

arr[i+1] / a[i] 가 배열 끝까지 r 값(공비)이 일정하면 true 를 return 하여 등비수열임을 알려 준다.

 

isArithmetic 는 주어진 수열이 등차 수열인지 아닌지 알려주는 메소드이다.

arr[i+1] - a[i] 가 배열 끝까지 d 값(공차)이 일정하면 true 를 return 하여 등차수열임을 알려 준다.

 

 

위 메소드를 통해서 주어진 배열이 등비인지 등차수열인지 알아낸 다음 배열의 맨 끝 숫자 (common[common.length-1])에 공차를 더하거나 공비를 곱하여 다음에 올 숫자를 수할 수 있다.

 

class Solution {

    public static boolean isGeometric (int [] arr){
        boolean result = false;
        int r = arr[1] / arr[0];

        for (int i = 1; i < arr.length-1; i++) {
            if(arr[i+1] / arr[i] == r){
                result = true;
            }
            else{
                result = false;
            }
        }

        return result;
    }

    public static boolean isArithmetic (int[]arr){
        boolean result = false;
        int d = arr[1] - arr[0];

        for (int i = 1; i < arr.length-1; i++) {
            if(arr[i+1] - arr[i] == d){
                result = true;
            }
            else{
                result = false;
            }
        }

        return result;

    }

    public int solution(int[] common) {
        int answer = 0;

        if(isArithmetic(common)){
            int d = common[1] - common[0];
            answer = common[common.length-1]+d;
        }
        else if(isGeometric(common)){
            int r = common[1] / common[0];
            answer = common[common.length-1] * r;
        }

        return answer;
    }
}