[백준 2738번/JAVA] 행렬 덧셈

문제

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

2738번: 행렬 덧셈

풀이 1

첫번 째는 세 배열을 따로따로 생성하는 방법이다.
N * M 배열 arr1, arr2의 각 원소를 덧셈하여 arr3에 저장

코드

import java.util.*;
import java.io.*;

public class Main {    
    public static void main(String[] args)throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringTokenizer st;

        //N M 입력받기
        st = new StringTokenizer(br.readLine());
        int N = Integer.parseInt(st.nextToken());
        int M = Integer.parseInt(st.nextToken());

        int[][] arr1 = new int[N][M];
        for(int i=0; i<N; i++) {
            st = new StringTokenizer(br.readLine());
            for(int j=0; j<M; j++) {
                arr1[i][j] = Integer.parseInt(st.nextToken());
            }
        }

        int[][] arr2 = new int[N][M];
        for(int i=0; i<N; i++) {
            st = new StringTokenizer(br.readLine());
            for(int j=0; j<M; j++) {
                arr2[i][j] = Integer.parseInt(st.nextToken());
            }
        }

        int[][] arr3 = new int[N][M];
        for(int i=0; i<N; i++) {
            for(int j=0; j<M; j++) {
                arr3[i][j] = arr1[i][j] + arr2[i][j];
            }
        }

        for(int i=0; i<N; i++) {
            for(int j=0; j<M; j++) {
                bw.write(arr3[i][j] + " ");
            }
        }
        bw.close();    
    }    
}

풀이 2

두 번째는 N * M 배열을 하나만 생성하고
각 원소에 새로 입력받은 값을 누적하는 방법이다.

코드

import java.util.*;
import java.io.*;

public class Main {	
	public static void main(String[] args)throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
		StringTokenizer st;
		
		//N M 입력받기
		st = new StringTokenizer(br.readLine());
		int N = Integer.parseInt(st.nextToken());
		int M = Integer.parseInt(st.nextToken());
		
		int[][] arr = new int[N][M];
		for(int i=0; i<N; i++) {
			st = new StringTokenizer(br.readLine());
			for(int j=0; j<M; j++) {
				arr[i][j] = Integer.parseInt(st.nextToken());
			}
		}
		
		for(int i=0; i<N; i++) {
			st = new StringTokenizer(br.readLine());
			for(int j=0; j<M; j++) {
				arr[i][j] += Integer.parseInt(st.nextToken());
			}
		}
		
		
		for(int i=0; i<N; i++) {
			for(int j=0; j<M; j++) {
				bw.write(arr[i][j] + " ");
			}
		}
		bw.close();	
	}	
}

풀이 3

세 번째는 N * M 배열을 하나만 생성하고
각 원소에 새로 입력받은 값을 누적하는 것까지는 풀이 2와 같은데
원소 누적을 삼중 for문으로 구현하는 방법이다.

코드

import java.util.*;
import java.io.*;

public class Main {    
    public static void main(String[] args)throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringTokenizer st;

        st = new StringTokenizer(br.readLine());
        int N = Integer.parseInt(st.nextToken());
        int M = Integer.parseInt(st.nextToken());
        int[][] arr = new int[N][M];

        for (int i = 0; i < 2; i++) {
            for (int j = 0; j < N; j++) {
                st = new StringTokenizer(br.readLine());
                for (int j2 = 0; j2 < M; j2++) {
                    arr[j][j2] += Integer.parseInt(st.nextToken());
                }
            }
        }

        for (int i = 0; i < N; i++) {
            for (int j = 0; j < M; j++) {
                bw.write(arr[i][j] + " ");
            }
            bw.write("\n");
        }
        bw.close();
    }    
}

풀이 4

풀이 3의 BufferedReader을 Scanner로 변경

코드

import java.util.Scanner;

public class Main {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);

		int N = sc.nextInt();
		int M = sc.nextInt();
		int[][] arr = new int[N][M];

		for (int i = 0; i < 2; i++) {
			for (int j = 0; j < N; j++) {
				for (int j2 = 0; j2 < M; j2++) {
					arr[j][j2] += sc.nextInt();
				}
			}
		}

		for (int i = 0; i < N; i++) {
			for (int j = 0; j < M; j++) {
				System.out.print(arr[i][j] + " ");
			}
			System.out.println();
		}
	}
}

 

성능 비교

 

풀이 1 : N*M 배열 3개 생성

풀이 2 : N*M 배열 1개 생성, 이중 for문 두 개로 행렬 원소 덧셈

풀이 3 : N*M 배열 1개 생성, 삼중 for문으로 행렬 원소 덧셈

풀이 4 : 풀이 3의 BufferedReader을 Scanner로 변경

 

• 배열 선언 갯수를 줄여 메모리를 아끼자.
• 이중 for문보다 삼중 for문 성능이 더 안 좋다.
• Scanner는 BufferedReader에 비해 성능이 매우 안 좋다. 메모리와 실행시간 모두 차이가 눈에 띄게 크다.
  삼중 for문에 Scanner 끼얹기