[코딩테스트 Lv.0] 조건에 맞게 수열 변환하기 1

문제 설명

정수 배열 arr가 주어집니다. arr의 각 원소에 대해 값이 50보다 크거나 같은 짝수라면 2로 나누고, 50보다 작은 홀수라면 2를 곱합니다. 그 결과인 정수 배열을 return 하는 solution 함수를 완성해 주세요.

제한사항

● 1 ≤ arr의 길이 ≤ 1,000,000
    ○ 1 ≤ arr의 원소의 값 ≤ 100

입출력 예

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

입출력 설명

입출력 예 #1
1, 3은 50 미만의 홀수 이므로 2를 곱하고, 100, 98은 50 이상의 짝수이므로 2로 나눕니다.
나머지 값들은 변경 조건에 해당하지 않으므로 바꾸지 않습니다.
따라서 [2, 2, 6, 50, 99, 49]를 return 합니다.

---

내 풀이

Java

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

C

#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>

// arr_len은 배열 arr의 길이입니다.
int* solution(int arr[], size_t arr_len) {
    // return 값은 malloc 등 동적 할당을 사용해주세요. 할당 길이는 상황에 맞게 변경해주세요.
    int* answer = (int*)malloc(sizeof(int) * arr_len);
    
    answer = arr;
    int i = 0;
    for(i = 0; i < arr_len; i++){
        if(answer[i] >= 50 && answer[i] % 2 == 0){
            answer[i] /= 2;
        }else if(answer[i] < 50 && answer[i] % 2 == 1){
            answer[i] *= 2;
        }
    }
    
    
    return answer;
}

Python

def solution(arr):
    answer = []
    
    for i in arr:
        if i >= 50 and i % 2 == 0:
            answer.append(i / 2)
        elif i < 50 and i % 2 == 1:
            answer.append(i * 2)
        else:
            answer.append(i)
    
    return answer

---

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

프로그래머스