[프로그래머스/Lv. 0] 다음에 올 숫자

내 코드 (틀림)

class Solution {
    public int solution(int[] common) {
        int answer = 0;
        
        for(int i=1 ; i < common.length -1 ; i++) {
            // 등차수열인 경우
            int diff1 = common[i] - common[i-1];
            int diff2 = common[i+1] - common[i];
            
            // 등비수열인 경우
            int div1 = common[i] / common[i-1];
            int div2 = common[i+1] / common[i];
            
            if(diff1 == diff2) {
                answer = common[common.length - 1] + diff1;
            }
            else if(div1 == div2) {
                answer = common[common.length - 1] * div1;
            }   
        }
        
        return answer;
    }
}

왜지.... 뭘 놓친걸까..

힌트 보다 보니까 0일 때가 있어서 에러가 난다고 하는데...

자꾸 해결이 안 되길래 결국 다른 분꺼 보고 했다..😥

 

정답 코드

import java.util.*;

class Solution {
    public int solution(int[] common) {
        int answer = 0;
        
        if((common[1] - common[0]) == (common[2] - common[1])) {
            answer = common[common.length-1] + (common[1] - common[0]);
        } else {
            answer = common[common.length-1] * (common[1] / common[0]);
        }
        return answer;
    }
}

---

(참고)

[프로그래머스] 다음에 올 숫자 자바(JAVA)