조건에 맞게 수열 반환하기 2

문제 설명

정수 배열 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

나의 코드

import java.util.Arrays;

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

느낀 점

문제를 풀면서 에러를 몇 번 마주했는데 먼저, 두 배열을 비교할 때 여러 가지 방법이 있었다. 나는 Arrays.equals() 메소드를 사용하여 이를 해결했다. 두 번째는 while문을 돌면서 변한 값이 arr에 복사를 시키지 않았다. 따라서 while문을 돌면서 조건에 맞을 때까지 반복하기 전 Arrays.copyOf()메소드를 사용하여 이 전 배열을 복사해주고 시작해야 제대로 된 비교가 가능했다. 마지막으로 그냥 count로 return하면 정답보다 1이 더 크게 나오는데 이는 문제에서 arr(x) = arr(x + 1)를 비교하고 있기 때문에 while문을 돌면서 추가된 count값은 정답+1의 값이라 1를 빼주고 return시켜 해결하였다.