[programmers] 조건에 맞게 수열 변환하기 2 - Java

정수 배열 arr의 각 원소에 대해 특정 작업을 반복한 결과인 배열 arr(x)에서 arr(x) = arr(x + 1)인 x가 항상 존재하는 가장 작은 x를 반환하는 solution 함수를 구현해야 합니다. arr의 길이는 1 이상 1,000,000 이하이며, arr의 원소 값은 1 이상 100 이하입니다.

Jan 14, 2024

문제 설명

정수 배열 arr가 주어집니다. arr의 각 원소에 대해 값이 50보다 크거나 같은 짝수라면 2로 나누고, 50보다 작은 홀수라면 2를 곱하고 다시 1을 더합니다.

이러한 작업을 x번 반복한 결과인 배열을 arr(x)라고 표현했을 때, arr(x) = arr(x + 1)인 x가 항상 존재합니다. 이러한 x 중 가장 작은 값을 return 하는 solution 함수를 완성해 주세요.

단, 두 배열에 대한 "="는 두 배열의 크기가 서로 같으며, 같은 인덱스의 원소가 각각 서로 같음을 의미합니다.

제한사항

1 ≤ arr의 길이 ≤ 1,000,000

1 ≤ arr의 원소의 값 ≤ 100

입출력 예

arr	result
[1, 2, 3, 100, 99, 98]	5

입출력 예 설명

입출력 예 #1

위 작업을 반복하면 다음과 같이 arr가 변합니다.

반복 횟수	arr
0	[1, 2, 3, 100, 99, 98]
1	[3, 2, 7, 50, 99, 49]
2	[7, 2, 15, 25, 99, 99]
3	[15, 2, 31, 51, 99, 99]
4	[31, 2, 63, 51, 99, 99]
5	[63, 2, 63, 51, 99, 99]
6	[63, 2, 63, 51, 99, 99]

이후로 arr가 변하지 않으며, arr(5) = arr(6)이므로 5를 return 합니다.

solution.java


import java.util.Arrays;

class Solution {
    public int solution(int[] arr) {
        int[] answer = Arrays.copyOf(arr, arr.length);
        int count = 0;

        while (true) {
            
            for(int i=0; i<arr.length; i++){
                if(arr[i]>=50 && arr[i]%2==0){
                    arr[i] /= 2;
                }else if(arr[i]<50 && arr[i]%2!=0){
                    arr[i] = arr[i] * 2 + 1;
                }
            }
            
            if (Arrays.equals(arr, answer)) {
                break;
            }
            answer = Arrays.copyOf(arr, arr.length);
            
            count++;
        }
        return count;
    }
}

핵심 키워드

두 배열을 비교하기 위해 == 연산자를 사용하면 배열의 요소가 아닌 heap 영역에 존재하는 배열의 값을 참조한다.

따라서 두 배열의 값을 비교하기 위해서는 Arrays.equals() 와 같은 메소드를 사용할 수 있다.

Arrays.copyOf() 메소드를 사용해서 특정 배열의 원하는 길이만큼 새로운 배열로 복사를 할 수 있다.

결론!

해당 문제를 풀면서 배열을 복사하는데 좀 더 효율적인 방법이 없을까 고민했고, Arrays.copyOf() 메소드의 활용 방법을 익힐 수 있었다.

[programmers] 조건에 맞게 수열 변환하기 2 - Java

Jan 14, 2024

import java.util.Arrays; class Solution { public int solution(int[] arr) { int[] answer = Arrays.copyOf(arr, arr.length); int count = 0; while (true) { for(int i=0; i<arr.length; i++){ if(arr[i]>=50 && arr[i]%2==0){ arr[i] /= 2; }else if(arr[i]<50 && arr[i]%2!=0){ arr[i] = arr[i] * 2 + 1; } } if (Arrays.equals(arr, answer)) { break; } answer = Arrays.copyOf(arr, arr.length); count++; } return count; } }