[프로그래머스] Lv. 0 마지막 두 원소 JAVA

마지막 두 원소

 

문제 설명

 

요구사항

• 마지막 정수가 직전 정수보다 크면 마지막 정수 - 직전 정수 추가
• 마지막 정수가 직전 정수보다 작으면 마지막 정수 * 2

 

테스트

package lv0;

public class 마지막_두_원소 {
	public int[] solution(int[] num_list) {
        int[] answer = new int[num_list.length + 1];
        
        for(int i = 0; i < num_list.length; i++) {
        	answer[i] = num_list[i];
        	
        	if(i == num_list.length - 1) {	// 마지막일 경우
        		if(answer[i - 1] < num_list[i]) { // 마지막 원소가 직전 원소보다 클 경우
        			answer[i + 1] = num_list[i] - num_list[i - 1]; // 마지막 원소 - 직전원소
        		} else { // 마지막 원소가 직전 원소보다 작을 경우
        			answer[i + 1] = num_list[i] * 2; // 마지막 원소 * 2
        		}
        	}
        }
        
        return answer;
    }
}

 

프로그래머스

class Solution {
    public int[] solution(int[] num_list) {
        int[] answer = new int[num_list.length + 1];
        
        for(int i = 0; i < num_list.length; i++) {
        	answer[i] = num_list[i];
        	
        	if(i == num_list.length - 1) {
        		if(answer[i - 1] < num_list[i]) {
        			answer[i + 1] = num_list[i] - num_list[i - 1];
        		} else {
        			answer[i + 1] = num_list[i] * 2;
        		}
        	}
        }
        
        return answer;
    }
}

 

결과