[소프티어] 나무공격 (Java)

Lv. 2

https://softeer.ai/practice/9657

첫번째 코드

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

public class Main {

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int m = sc.nextInt();

        int[][] grid = new int[n][m];

        for(int i=0; i<n; i++){
            for(int j=0; j<m; j++){
                grid[i][j] = sc.nextInt();
            }
        }

        int L1 = sc.nextInt() - 1;
        int R1 = sc.nextInt() - 1;
        
        attack(grid, L1, R1, m);
        
        int L2 = sc.nextInt() - 1;
        int R2 = sc.nextInt() - 1;

        sc.close();
        
        attack(grid, L2, R2, m);

        System.out.println(countRemaining(grid, n, m));
        
    }

    public static void attack(int[][] grid, int l, int r, int m) {
        for(int i=l; i<=r; i++) {
            for(int j=0; j<m; j++) {
                if(grid[i][j] == 1){
                    grid[i][j] = 0;
                    break;
                }
            }
        }
    }

    public static int countRemaining(int[][] grid, int n, int m) {
        int cnt = 0;
        for(int i=0; i<n; i++){
            for(int j=0; j<m; j++){
                if(grid[i][j] == 1) {
                    cnt++;
                }
            }
        }
        return cnt;
    }

    
}

시간단축이나 메모리 절약 측면도 고려해봐야겠다.

---

두번째 코드

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

public class Main {

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

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

        int[][] grid = new int[n][m];

        for (int i = 0; i < n; i++) {
            st = new StringTokenizer(br.readLine());
            for (int j = 0; j < m; j++) {
                grid[i][j] = Integer.parseInt(st.nextToken());
            }
        }
        st = new StringTokenizer(br.readLine());
        // 첫번째 공격
        int L1 = Integer.parseInt(st.nextToken()) - 1;
        int R1 = Integer.parseInt(st.nextToken()) - 1;

        attack(grid, L1, R1, m);

        st = new StringTokenizer(br.readLine());
        // 두번째 공격
        int L2 = Integer.parseInt(st.nextToken()) - 1;
        int R2 = Integer.parseInt(st.nextToken()) - 1;
        attack(grid, L2, R2, m);

        System.out.println(countRemaining(grid, n, m));


    }


    private static int countRemaining(int[][] grid, int n, int m) {
        int sum = 0;
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                if (grid[i][j] == 1) {
                    sum++;
                }
            }
        }
        return sum;
    }

    private static void attack(int[][] grid, int L1, int R1, int m) {
        for (int i = L1; i <= R1; i++) {
            for (int j = 0; j < m; j++) {
                if (grid[i][j] == 1) {
                    grid[i][j] = 0;
                    break;
                }
            }
        }
    }

}

첫번째 코드에서 Scanner 사용부분만 BufferedReader,StringTokenizer 조합으로 변경해보았다. 실행시간이나 메모리가 많이 줄어드는 효과가 컸다.

BufferedReader는 내부적으로 8KB 버퍼(기본값)를 사용하여 한꺼번에 데이터를 읽어온다.
반면, Scanner는 한 글자씩 읽어서 처리하기 때문에 입출력 속도가 느려진다.
특히 반복문에서 Scanner로 여러 번 읽는 경우 차이가 더 심해진다고 한다.

대신 BufferedReader는 의 readLine()은 반환타입이 String 으로 고정되어 있다.
꼭 사용용도에 맞는 형변환을 해주어야 한다.

예외처리도 필요하다.try/catch 나 throws IOException 처리를 해주어야 한다.

• 클래스 import 추가
  import java.io.IOException;

• main 클래스 옆에 throws IOException 처리
  public static void main(String[] args) throws IOException {}

🧚 같은 로직이라도 사용하는 클래스와 메서드에 따라 성능 차이가 크므로, 최적의 방법을 선택하는 것이 중요하군