본문 바로가기

Java/프로그래머스

[JAVA] 프로그래머스 - 마지막 두 원소

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

 

프로그래머스

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

programmers.co.kr

 

- 정수 리스트 num_list가 주어질 때, 마지막 원소가 그 전 원소보다 크면 마지막 원소에서 그 전 원소를 뺀 값을 마지막 원소가 그전 원소보다 크지 않다면 마지막 원소를 두 배한 값을 추가하여 return 하도록 solution 함수를 완성 

 

 

 

 

정답 

 

import java.util.Arrays;

class Solution {
    public int[] solution(int[] num_list) {
       int answer[] = Arrays.copyOf(num_list,num_list.length+1);
        
       int last = 0;
       
        if(num_list[num_list.length-1]>num_list[num_list.length-2])
        {
            last = num_list[num_list.length-1] - num_list[num_list.length-2];
        }
        else
        {
            last = num_list[num_list.length-1]*2;
        }
        
        answer[answer.length-1] = last;
        
        return answer;
    }
}