문제 설명
이차원 정수 배열 arr이 매개변수로 주어집니다. arr의 행의 수가 더 많다면 열의 수가 행의 수와 같아지도록 각 행의 끝에 0을 추가하고, 열의 수가 더 많다면 행의 수가 열의 수와 같아지도록 각 열의 끝에 0을 추가한 이차원 배열을 return 하는 solution 함수를 작성해 주세요.
제한사항
- 1 ≤ arr의 길이 ≤ 100
- 1 ≤ arr의 원소의 길이 ≤ 100
- 1 ≤ arr의 원소의 원소 ≤ 1,000
입출력 예
arr | result |
[[572, 22, 37], [287, 726, 384], [85, 137, 292], [487, 13, 876]] | [[572, 22, 37, 0], [287, 726, 384, 0], [85, 137, 292, 0], [487, 13, 876, 0]] |
[[57, 192, 534, 2], [9, 345, 192, 999]] | [[57, 192, 534, 2], [9, 345, 192, 999], [0, 0, 0, 0], [0, 0, 0, 0]] |
[[1, 2], [3, 4]] | [[1, 2], [3, 4]] |
class Solution {
public int[][] solution(int[][] arr) {
int row = arr.length;
int column = arr[0].length;
if (row > column) {
int[][] newArr = new int[row][row];
for (int i = 0; i < row; i++) {
for (int j = 0; j < row; j++) {
if (j < column) {
newArr[i][j] = arr[i][j];
} else {
newArr[i][j] = 0;
}
}
}
return newArr;
} else if (column > row) {
int[][] newArr = new int[column][column];
for (int i = 0; i < column; i++) {
for (int j = 0; j < column; j++) {
if (i < row) {
newArr[i][j] = arr[i][j];
} else {
newArr[i][j] = 0;
}
}
}
return newArr;
} else {
return arr;
}
}
}
다른 사람의 풀이 맨 위에 있는
class Solution {
public int[][] solution(int[][] arr) {
int max = Math.max(arr.length, arr[0].length);
int[][] array = new int[max][max];
for (int i = 0; i < arr.length; i++) {
System.arraycopy(arr[i], 0, array[i], 0, arr[0].length);
}
return array;
}
}
이 코드를 읽자 마자 내 입에서 '미친...'소리가 절로 나왔다.
모르거나 안 써 본 메서드가 하나도 없다.
근데 진짜 이렇게 할 생각을 0.1g도 해 내지 못 했다.
와, 진짜 쉽고, 간단하고, 단순하고, 좋은 코드다...
나는 도대체 뭐가 문젤까?
좀 더 '입체적인' 사고를 못 하는 건가???
'코딩테스트 연습' 카테고리의 다른 글
x만큼 간격이 있는 n개의 숫자 풀이 - Lv. 1 (0) | 2024.04.25 |
---|---|
하샤드 수 풀이 - 프로그래머스 Lv. 1 (0) | 2024.04.25 |
소인수분해 (0) | 2024.04.24 |
문자 개수 세기 (0) | 2024.04.24 |
정수 제곱근 판별 (0) | 2024.04.24 |