석유시추

Lee1231234·2024년 4월 15일

코딩테스트

목록 보기
75/95

석유가 묻힌 땅과 석유 덩어리를 나타내는 2차원 정수 배열 land가 매개변수로 주어집니다. 이때 시추관 하나를 설치해 뽑을 수 있는 가장 많은 석유량을 return 하도록 solution 함수를 완성해 주세요.

제한사항
1 ≤ land의 길이 = 땅의 세로길이 = n ≤ 500
1 ≤ land[i]의 길이 = 땅의 가로길이 = m ≤ 500
land[i][j]는 i+1행 j+1열 땅의 정보를 나타냅니다.
land[i][j]는 0 또는 1입니다.
land[i][j]가 0이면 빈 땅을, 1이면 석유가 있는 땅을 의미합니다.
정확성 테스트 케이스 제한사항
1 ≤ land의 길이 = 땅의 세로길이 = n ≤ 100
1 ≤ land[i]의 길이 = 땅의 가로길이 = m ≤ 100
효율성 테스트 케이스 제한사항
주어진 조건 외 추가 제한사항 없습니다.

처음한 생각

석유의 값은 BFS로 구할수있다. 그런데 y값을 통해서 가장많이 뽑히는 구간을 찾아야한다.
구한 값을 또 다시 순회한다면 효율성 테스트에서 걸린다. SET을 사용해서 구한후 바로 값을 추가해준다면 다시 순회할 이유가 없다.
이 문제는 BFS와 SET의 사용방법을 하는지 물어보는 문제같다.

코드

import java.util.*;
class Solution {
    public int solution(int[][] land) {
        int x = land.length;
        int y = land[0].length;
        int[] oil = new int[y];
        int answer = 0;
        
        boolean[][] visit = new boolean[x][y];
        for(int i=0;i<x;i++){
            for(int j=0;j<y;j++){
                if(!visit[i][j]&&land[i][j]==1){
                    bfs(i,j,oil,visit,land);
                    
                }
            }
        }
        for(int i=0;i<y;i++){
            answer =Math.max(answer,oil[i]);
        }
        
        return answer;
    }
    void bfs(int x, int y,int[] oil,boolean[][] visit,int[][] land){
        int[] dx = {1,-1,0,0};
        int[] dy = {0,0,1,-1};
        int count = 1;
        visit[x][y] = true;
        Queue<int[]> queue= new LinkedList<>();
        queue.add(new int[]{x,y});
        HashSet<Integer> set = new HashSet<>();
        while(!queue.isEmpty()){
            int[] tmp = queue.poll();
            set.add(tmp[1]);                      
            for(int i=0;i<4;i++){
                int nx = tmp[0] + dx[i];
                int ny = tmp[1] + dy[i];
            
                if(check(nx,ny,land)) continue;
                
                if(!visit[nx][ny]){
                    queue.add(new int[]{nx,ny});
                    count++;
                    visit[nx][ny]= true;
                }
            }
        }
        for(int i:set){
            oil[i] += count;
        }
    }
    boolean check(int x,int y,int[][] land){
        int n = land.length;
        int m = land[0].length;
        if(x<0||y<0||x>=n||y>=m||land[x][y]==0) return true;
        return false;
    }
}//리스트를 통한 모으기
profile
not null

0개의 댓글