[프로그래머스/JAVA] 조건에 맞게 수열 변환하기 1

문제 설명

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

---

제한사항

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

---

입출력 예arrresult

[1, 2, 3, 100, 99, 98]

[2, 2, 6, 50, 99, 49]

 

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

답을 저장할 answer 배열을 arr 길이만큼 생성

조건에 맞춰 answer 배열에 값 저장 후 리턴

 

다른 풀이들을 검색하다가 새로운 배열을 또 생성하지 않고 풀이하는 것을 발견했다.

새로운 배열을 만들지 않고 필요한 조건들만 작성해 더 효율적으로 풀이할 수 있는 것 같다. 

public class H181882 {
	public int[] solution(int[] arr) {
		// 배열의 길이만큼 반복문을 돌려 각 원소에 대한 조건에 따라 연산을 수행
		for (int i = 0; i < arr.length; i++) {
			// 원소가 50이상의 짝수인 경우, 원소를 2로 나눔
			if (arr[i] >= 50 && arr[i] % 2 == 0) {
				arr[i] /= 2;
			} 
			// 원소가 50미만의 홀수인 경우, 원소에 2를 곱함
			else if (arr[i] < 50 && arr[i] %2 == 1) {
				arr[i] *= 2;
			}
			// 그 외의 경우에는 원소를 변경하지 않음
		}
		// 변경된 원소들로 이루어진 배열을 반환!
		return arr;
	}
}

출처

https://aihtnyc-h.tistory.com/m/entry/%EC%A1%B0%EA%B1%B4%EC%97%90-%EB%A7%9E%EA%B2%8C-%EC%88%98%EC%97%B4-%EB%B3%80%ED%99%98%ED%95%98%EA%B8%B01