class Solution {
public int[][] solution(int[][] arr) {
int[][] answer = {};
return answer;
}
}
import java.util.*;
class Solution {
public int[][] solution(int[][] arr) {
int[][] answer1 = new int[arr.length][arr.length];
int[][] answer2 = new int[arr[0].length][arr[0].length];
// 행 > 열
if (arr.length > arr[0].length) {
for(int i = 0; i < arr.length; i++) {
answer1[i] = Arrays.copyOf(arr[i], arr.length);
}
return answer1;
// 행 < 열
} else if (arr[0].length > arr.length) {
for (int i = 0; i < arr.length; i++) {
answer2[i] = Arrays.copyOf(arr[i], arr[0].length);
}
return answer2;
}
// 행 = 열
return arr;
}
}
행과 열이 동일한 길이를 가지도록, 처음부터 배열을 그렇게 생성해준다
Arrays.copyOf : 배열 복사
class Solution {
public int[][] solution(int[][] arr) {
int max = Math.max(arr.length, arr[0].length);
int[][] answer = new int[max][max];
for (int i = 0; i < arr.length; i++) {
for (int j = 0; j < arr[i].length; j++) {
answer[i][j] = arr[i][j];
}
}
return answer;
}
}