[C++] 프로그래머스 조건에 맞게 수열 변환하기 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

입출력 예

arr	result
[1, 2, 3, 100, 99, 98]	[2, 2, 6, 50, 99, 49]

풀이

이전 문제에서 arr(x) = arr(x + 1)인 가장 작은 x값을 구하는 조건이 추가된 문제입니다.

#include <string>
#include <vector>

using namespace std;

bool arr_equal(vector<int> a, vector<int> b){
    
    for(int i = 0; i < a.size(); i++)
    {
        if(a[i] != b[i])
            return false;
    }
    
    return true;
}

int solution(vector<int> arr) {
    int answer = 1;
    vector<int> pre_arr = arr;
    
    while(true)
    {
        vector<int> after_arr;
        
        for(int i = 0; i < pre_arr.size(); i++)
        {
            if((pre_arr[i] >= 50) && (pre_arr[i] % 2 == 0))
            {
                after_arr.push_back(pre_arr[i] / 2);
            }
            else if((pre_arr[i] < 50) && (pre_arr[i] % 2 != 0))
            {
                after_arr.push_back(pre_arr[i] * 2 + 1);
            }
            else
            {
                after_arr.push_back(pre_arr[i]);
            }
        }
        
        if(arr_equal(pre_arr, after_arr))
        {
            answer--;
            break;
        }
        else
        {
            answer++;
            pre_arr = after_arr;
        }
    }
    
    return answer;
}

결과