[백준] 2738 행렬 덧셈

https://www.acmicpc.net/problem/2738

2738번: 행렬 덧셈

 

방법1

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.List;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer str = new StringTokenizer(br.readLine());

        int n = Integer.parseInt(str.nextToken());
        int m = Integer.parseInt(str.nextToken());

        int[][] arr1 = new int[n][m];
        int[][] arr2 = new int[n][m];

        //첫 번째 배열
        for (int i = 0; i < n; i++) {
            str = new StringTokenizer(br.readLine());
            for (int j = 0; j < m; j++) {
                arr1[i][j] = Integer.parseInt(str.nextToken());
            }
        }
        
        //두 번째 배열
        for (int i = 0; i < n; i++) {
            str = new StringTokenizer(br.readLine());
            for (int j = 0; j < m; j++) {
                arr2[i][j] = Integer.parseInt(str.nextToken());
            }
        }
        
        //두 배열 합해서 출력
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                sb.append((arr1[i][j] + arr2[i][j]) + " ");
            }
            sb.append("\n");
        }

        System.out.println(sb.toString());
    }
}

 

방법2

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer str = new StringTokenizer(br.readLine());

        int n = Integer.parseInt(str.nextToken());
        int m = Integer.parseInt(str.nextToken());

        int[][] arr = new int[n][m];
        for(int k=0; k<2; k++){
            for(int i=0; i<n; i++){
                str = new StringTokenizer(br.readLine());
                for(int j=0; j<m; j++){
                    arr[i][j] += Integer.parseInt(str.nextToken());
                }
            }
        }
        //출력
        for(int i=0; i<n; i++){
            for(int j=0; j<m; j++){
                System.out.print(arr[i][j] + " ");
            }
            System.out.println();
        }
    }
}

 

두 번째 방법은 한 번에 두 배열을 더할 수 있어서 첫 번째 보다 간단해 보이지만,

for문이 삼중으로 들어가서 depth가 들어갈 수록 성능이 안좋을 것이라 생각했다.

 

위가 방법1, 아래가 방법2이다. 방법1이 메모리와 시간이 더 효율적이다.