[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
[programmers] 조건에 맞게 수열 변환하기 2 - Java

문제 설명

정수 배열 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() 메소드의 활용 방법을 익힐 수 있었다.
 
Share article

👨🏻‍💻DriedPollack's Blog