격자판 최대합 - 제가 너무 어렵게 생각하는걸까요?

 public static void main(String[] args) {
        Main main = new Main();
        Scanner scanner = new Scanner(System.in);
        int num = scanner.nextInt();
        int[][] answers = new int[num][num];
        for (int row = 0; row < num; row++) {
            for (int col = 0; col < num; col++) {
                answers[row][col] = scanner.nextInt();
            }
        }

        System.out.println(solution(answers));
    }

    private static int solution(int[][] nums) {
        int size = nums.length;
        int[] colSum = new int[size];
        int[] rowSum = new int[size];
        int leftDiagonalSum = 0;
        int rightDiagonalSum = 0;

        for (int row = 0; row < size; row++) {
            for (int col = 0; col < size; col++) {
                rowSum[row] += nums[row][col];
                colSum[col] += nums[row][col];

                if (row == col) {
                    leftDiagonalSum += nums[row][col];
                }
                if (row + col == 4) {
                    rightDiagonalSum += nums[row][col];
                }
            }
        }

        int colMax = Arrays.stream(colSum).max().getAsInt();
        int rowMax = Arrays.stream(rowSum).max().getAsInt();
        return Stream.of(colMax, rowMax, rightDiagonalSum, leftDiagonalSum)
                        .max(Comparator.comparing(Integer::intValue))
                        .get();
    }
}

전체를 돌면서 row, col 의 좌표값으로 대각선값을 구하게 했는데요.
선생님 강의를 보면 더 단순하게 생각할 수 있다고 느낍니다.
단순 코드차이로 보면 될까요?
아니면 제가 앞으로 문제를 풀면서 더 단순하게 생각해야할까요?