석유가 묻힌 땅과 석유 덩어리를 나타내는 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;
}
}//리스트를 통한 모으기