백준은 javascript 를 따로 지원하고 있지 않아서 node.js로 풀어주었다.
사실 그래서 백준보다는 프로그래머스를 많이 풀었는데 BFS 문제를 더 연습해보고 싶어 백준문제를 풀게 되었다.
백준에서는 node.js 를 사용하면 되고 input 값만 잘 받아주면 문제 없이 문제를 풀 수 있었다 !
유기농 배추 (실2)
1 초 | 512 MB | 170400 | 68405 | 45827 | 37.997% |
문제
차세대 영농인 한나는 강원도 고랭지에서 유기농 배추를 재배하기로 하였다. 농약을 쓰지 않고 배추를 재배하려면 배추를 해충으로부터 보호하는 것이 중요하기 때문에, 한나는 해충 방지에 효과적인 배추흰지렁이를 구입하기로 결심한다. 이 지렁이는 배추근처에 서식하며 해충을 잡아 먹음으로써 배추를 보호한다. 특히, 어떤 배추에 배추흰지렁이가 한 마리라도 살고 있으면 이 지렁이는 인접한 다른 배추로 이동할 수 있어, 그 배추들 역시 해충으로부터 보호받을 수 있다. 한 배추의 상하좌우 네 방향에 다른 배추가 위치한 경우에 서로 인접해있는 것이다.
한나가 배추를 재배하는 땅은 고르지 못해서 배추를 군데군데 심어 놓았다. 배추들이 모여있는 곳에는 배추흰지렁이가 한 마리만 있으면 되므로 서로 인접해있는 배추들이 몇 군데에 퍼져있는지 조사하면 총 몇 마리의 지렁이가 필요한지 알 수 있다. 예를 들어 배추밭이 아래와 같이 구성되어 있으면 최소 5마리의 배추흰지렁이가 필요하다. 0은 배추가 심어져 있지 않은 땅이고, 1은 배추가 심어져 있는 땅을 나타낸다.
1 | 1 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
0 | 1 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
0 | 0 | 0 | 0 | 1 | 0 | 0 | 0 | 0 | 0 |
0 | 0 | 0 | 0 | 1 | 0 | 0 | 0 | 0 | 0 |
0 | 0 | 1 | 1 | 0 | 0 | 0 | 1 | 1 | 1 |
0 | 0 | 0 | 0 | 1 | 0 | 0 | 1 | 1 | 1 |
입력
입력의 첫 줄에는 테스트 케이스의 개수 T가 주어진다. 그 다음 줄부터 각각의 테스트 케이스에 대해 첫째 줄에는 배추를 심은 배추밭의 가로길이 M(1 ≤ M ≤ 50)과 세로길이 N(1 ≤ N ≤ 50), 그리고 배추가 심어져 있는 위치의 개수 K(1 ≤ K ≤ 2500)이 주어진다. 그 다음 K줄에는 배추의 위치 X(0 ≤ X ≤ M-1), Y(0 ≤ Y ≤ N-1)가 주어진다. 두 배추의 위치가 같은 경우는 없다.
출력
각 테스트 케이스에 대해 필요한 최소의 배추흰지렁이 마리 수를 출력한다.
예제 입력 1
2
10 8 17
0 0
1 0
1 1
4 2
4 3
4 5
2 4
3 4
7 4
8 4
9 4
7 5
8 5
9 5
7 6
8 6
9 6
10 10 1
5 5
예제 출력 1
5
1
예제 입력 2
1
5 3 6
0 2
1 2
2 2
3 2
4 2
4 0
예제 출력 2
2
나의 풀이
const fs = require('fs');
const input = fs.readFileSync('/dev/stdin').toString().trim().split('\n');
const T = input[0]; //테스트 케이스가 몇개 인지
const testCases = input.slice(1); // 첫째줄빼고 나머지를 새로운 배열로 선언
let dy = [1,-1,0,0]; //상하좌우 좌표 이동을 위함
let dx = [0,0,1,-1];
let M,N,K,visited,maps; // bfs 함수에 사용하기 위해 전역으로 선언함
for(let t =0; t<T ; t++){
[M ,N ,K] = testCases[0].split(' ').map(Number); //첫번째줄은 가로 길이 ,세로길이 , 배추의 위치 수
maps = Array.from({length:N}, ()=>Array(M).fill(0)); //M*N 크기의 0인 배열 선언
visited = Array.from({length: N},() => Array(M).fill(false)); // M*N크기의 false 배열 선언
let cabbage = []; // 배추의 위치를 담을 배열
for (let i = 0; i < K; i++) {
const [x, y] = testCases[i+1].split(' ').map(Number); // 테스트케이스 1번째 줄은 M,N,K를 나타내므로 i+1번째 부터 접근함
maps[y][x] = 1; // 좌표 반대로 저장 -> 이차원 배열을 생각해볼때 가로가 y좌표이고 세로가 x좌표를 의미하기 때문
cabbage.push([y, x]); //배추의 위치를 담음
}
let answer = 0; //지렁이 갯수
for(let i =0; i< cabbage.length; i++){ //배추의 수만큼 for문 반복
const [y, x] = cabbage[i];
if(!visited[y][x] && maps[y][x]===1){ //방문하지 않은 배추 위치일 때
visited[y][x]= true; // 방문 표시를 한 후
bfs(y,x); // bfs 실행 -> 인접한 배추를 확인함 없을 땐 종료된 후 answer -> 지렁이 한마리 추가
answer++; // -> bfs 가 종료된 것은 인접한 배추를 다 확인했다는 뜻
}
}
console.log(answer);
if(t!==T){ // 테스트 케이스가 1개 이상일 때 다음 케이스를 위해 하지만 마지막 케이스일때는 자를필요 없음
testCases.splice(0,K+1); // 배열을 잘라서 없애줌
}
}
function bfs(y, x){ // 현재 위치로 bfs 실행
const queue = [[y,x]]; //현재 위치 큐에 삽입
while(queue.length){ //큐가 빌때까지 실행 , 즉 인접한 배추가 없을때 까지 실행
let [cy,cx] = queue.shift(); // 현재 위치를 꺼냄
// console.log(cy,cx);
for(let i =0; i<4; i++){ // 위치 이동
let ny = cy + dy[i]; // dy = [1,-1,0,0];
let nx = cx + dx[i]; // dx = [0,0,1,-1];
if(ny >= 0 && ny < N && nx >=0 && nx < M && !visited[ny][nx]) { //만약 상하좌우로 이동시킨 좌표가 유효한 값이고 방문하지 않았을때 방문처리해줌
visited[ny][nx]=true;
if(maps[ny][nx]===1){ //배추가 있을 때
queue.push([ny,nx]); //queue 에 push 해줌
}
}
}
}
}
bfs(0,0);
정말!! 하나의 예제에 테스트가 2개가 있어서 input을 받는데 좀 애먹었던 것 같다.
문제에 대한 설명은 각 줄마다 주석으로 남겨놓았다.
좌표랑 배열이랑 반대로 생각해야해서 헷갈려서 근처 다이소에서 공책과 볼펜을 구매해 직접 손으로 써가면서 이해했다.
그래도 4시간의 노력 끝에 문제를 풀 수 있어서 정말 뿌듯했다!!
'코테 준비 > 백준' 카테고리의 다른 글
[백준] 벽 부수고 이동하기 (js) ( 2% 시간초과 해결 ) (2) | 2024.11.29 |
---|---|
[백준] 숨바꼭질 2 (js) (2) | 2024.11.12 |
[백준] 9935번 문자열 폭발 (js) (1) | 2024.11.09 |
[백준] 토마토 (javascript , node.js) 시간초과 해결 (1) | 2023.12.09 |