Java/프로그래머스
[프로그래머스/JAVA] 조건에 맞게 수열 변환하기 1
녹녹1
2023. 9. 11. 18:25
문제 설명
정수 배열 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;
}
}출처