SWEA 1209 Sum
문제 정보
- 문제 출처 : SW Expert Academy
- 문제 번호 : 1209
- 문제 제목 : Sum
- 문제 난이도 : D3
- 풀이 언어 : Java
문제
100X100의 2차원 배열이 주어질 때, 각 행의 합, 각 열의 합, 각 대각선의 합 중 최댓값을 구하는 문제
접근
for 문을 활용하여 2차원 배열을 돌면서 각 행의 합, 각 열의 합, 각 대각선의 합 중 최댓값을 구했다.
풀이
- 각 테스트 케이스 첫 줄에는 테스트 케이스 번호가 주어진다.
- 그 다음 줄에는 2차원 배열의 각 행 값이 주어진다.
- 테스트 케이스는 총 10개이다.
- 100×100 크기의 2차원 배열 arr 을 생성한 후 값을 입력 받는다.
- 변수 max 를 초기화 한다.
- 첫 번째 for 문에 rowSum colSum 로 각 행과 열의 합을 저장할 변수를 초기화 한다.
- 두 번째 for 문은 각 행과 열의 합을 계산한다.
- max 변수에 최대값을 갱신한다.
- leftDiagonal rightDiagonal 로 각 대각선의 합을 저장할 변수를 초기화 한다.
- for 문으로 배열의 각 대각선의 합을 계산한다.
- max 변수에 최대값을 갱신한다.
- max 변수에는 각 행의 합, 각 열의 합, 각 대각선의 합 중 최댓값이 들어있게 된다.
후기
2차원 배열을 돌면서 최댓값을 구하는 문제이다. 2차원 배열을 돌면서 문제를 해결하는 기본적인 문제이므로 잘 기억해두었다가 활용해야겠다.
코드
Java
import java.util.Scanner;
import java.io.FileInputStream;
class Solution
{
public static void main(String args[]) throws Exception
{
Scanner sc = new Scanner(System.in);
for(int test_case = 1; test_case <= 10; test_case++)
{
int caseNum = sc.nextInt();
int[][] arr = new int[100][100];
for (int i = 0; i < arr.length; i++) {
for (int j = 0; j < arr[0].length; j++) {
arr[i][j] = sc.nextInt();
}
}
int max = 0;
// 행과 열의 합 중 최대값 계산
for (int i = 0; i < arr.length; i++) {
int rowSum = 0;
int colSum = 0;
for (int j = 0; j < arr[i].length; j++) {
rowSum += arr[i][j];
colSum += arr[j][i];
}
max = Math.max(max, Math.max(rowSum, colSum));
}
// 왼쪽, 오른쪽 대각선의 합 중 최대값 계산
int leftDiagonal = 0;
int rightDiagonal = 0;
for (int i = 0; i < arr.length; i++) {
leftDiagonal += arr[i][i];
rightDiagonal += arr[i][arr.length - 1 - i];
}
max = Math.max(max, Math.max(leftDiagonal, rightDiagonal));
System.out.println("#" + caseNum + " " + max);
}
}
}