프로그래머스 250136 석유 시추 Java

: ) YOUNG·2023년 12월 1일
1

알고리즘

목록 보기
271/417
post-thumbnail

프로그래머스 250136번
https://school.programmers.co.kr/learn/courses/30/lessons/250136

문제



생각하기


  • BFS를 활용해서 푸는 문제이다.

동작

먼저 BFS를 석유 덩어리가 얼마의 크기를 가지고 있는지를 파악하고, land 배열에 저장을 idx 값을 1씩 증가시키며 저장한다.

그리고 이 idx값을 key으로 하고 size를 value로 저장해서 HashMap에 지도에서 idx가 얼마의 석유를 가지고 있는지를 저장해둔다.


        int ans = 0;
        for(int i=0; i<M; i++) {
            Set<Integer> set = new HashSet<>();
            
            for(int j=0; j<N; j++) {
                if(land[j][i] > 0) {
                    set.add(land[j][i]);
                }
            }
            
            int sum = 0;
            for(int num : set) {
                sum += map.get(num);
            }
            ans = Math.max(ans, sum);
        }

그리고 land 배열을 세로로 탐색하도록 반복문을 구현한 뒤
set에서 중복되는 값을 제외하고, 해당 set에 있는 oil를 다시 map에 key값으로 대입해 value를 가져와서 총 합을 구하면 해당 컬럼으로 탐색했을 때 구해지는 석유 매장량에서 최대 값을 구하면 정답을 얻을 수 있다.



결과


코드



import java.util.*;

class Solution {
    public static int[][] land;
    public static boolean[][] isVisited;
    public static int N, M;
    public static final int[] dirX = {-1, 0, 1, 0}; // 상 우 하 좌
    public static final int[] dirY = {0, 1, 0, -1};
    
    public static class Coordinate {
        int x;
        int y;
        
        public Coordinate(int x, int y) {
            this.x = x;
            this.y = y;
        }
    } // End of Coordinate class
    
    public int solution(int[][] land) {        
        input(land);
        
        HashMap<Integer, Integer> map = new HashMap<>();
        int idx = 2;
        for(int i=0; i<N; i++) {
            for(int j=0; j<M; j++) {
                if(isVisited[i][j] || land[i][j] == 0) continue;
                
                int count = BFS(i, j, idx);
                map.put(idx++, count);                
            }
        }
        
        int ans = 0;
        for(int i=0; i<M; i++) {
            Set<Integer> set = new HashSet<>();
            
            for(int j=0; j<N; j++) {
                if(land[j][i] > 0) {
                    set.add(land[j][i]);
                }
            }
            
            int sum = 0;
            for(int num : set) {
                sum += map.get(num);
            }
            ans = Math.max(ans, sum);
        }
        
         
        return ans;
    } // End of solution()
    
    public int BFS(int x, int y, int idx) {
        LinkedList<Coordinate> que = new LinkedList<>();
        int count = 1;
        
        que.offer(new Coordinate(x,y));
        land[x][y] = idx;
        isVisited[x][y] = true;
        
        while(!que.isEmpty()) {
            Coordinate current = que.poll();
            
            for(int i=0; i<4; i++) {
                int nextX = dirX[i] + current.x;
                int nextY = dirY[i] + current.y;
                
                if(!isAbleCheck(nextX, nextY)) continue;
                isVisited[nextX][nextY] = true;
                
                land[nextX][nextY] = idx;
                count++;
                que.offer(new Coordinate(nextX, nextY));
            }
        }
        
        return count;
    } // End of BFS()
    
    public boolean isAbleCheck(int nextX, int nextY) {
        return nextX >= 0 && nextX < N && nextY >= 0 && nextY < M && !isVisited[nextX][nextY] && land[nextX][nextY] == 1;
    } // End of isAbleCheck()
    
    public void input(int[][] land) {
        N = land.length;
        M = land[0].length;
        isVisited = new boolean[N][M];
        
        this.land = land;
    } // End of input();
} //End of Solution class

0개의 댓글