[Lv.0] 마지막 두 원소

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

프로그래머스

 

문제 설명

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

 

제한 사항

 

입출력 예

 

입출력 예 설명

 

아이디어

• 크지 않다는 게 같다는 것도 포함이다.

초기 코드

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

근데 지금 이러면 answer의 값은 경우마다 변동. 따라서 결과를 저장해주고 들어가야 한다. 담을 변수 하나 더 생성 필요.

그리고 그 변수를 마지막에 answer에 담아주면 된다. 리턴하기 전에 num이 answer의  마지막 원소에 추가되게

 

** 배열이 한 번 만들어졌으면, 그 배열의 length라고 하는 게 덜 헷갈리는 모먼트이다.

그래서 num_list의 length와 관련된 식으로 표현하지 말고, 만들어졌으니까

 

나의 코드

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