inflearn logo
강의

강의

N
챌린지

챌린지

멘토링

멘토링

N
클립

클립

로드맵

로드맵

지식공유

it 취업을 위한 알고리즘 문제풀이 입문 (with C/C++) : 코딩테스트 대비

89. 토마토(BFS 활용)

입력이 더 많아지면 제 방법을 사용할 수 있을지 모르겠습니다.

349

Eunyoung Roh

작성한 질문수 4

1

안녕하세요 선생님, 강의 잘 듣고 있습니다. 

89번 토마토는 선생님과 다른 방법으로 풀었는데, 마지막 입력인 in5에서 아슬아슬하게 0.8초 ~ 0.9초가 나옵니다. 반면 선생님의 소스 코드로 시간을 측정해 보니 제 코드보다 절반 정도 단축된 시간인 0.47초가 나오더라고요.

제공해 주신 채점기는 모두 통과했지만, 입력이 여기서 더 커지면 제 구현으로 채점 사이트를 통과하지 못할 거라는 생각이 듭니다. 한번 소스 코드를 봐 주시면 감사하겠습니다.

#include <iostream>
#include <vector>
#include <queue>

using namespace std;

int grid[1000][1000];

int width, height;
int gridSize;
int ripeTomatoSize;
queue<pair<int, int>> ripeTomatos;
queue<pair<int, int>> affectedTomatos;
pair<int, int> directions[4] = {{-1, 0}, {1, 0}, {0, -1}, {0, 1}};

int BFS(){
	
	int day = 1;
	
	while(!ripeTomatos.empty()){
		
		// 오늘 익어 있는 토마토들을 꺼내 다른 토마토들을 익힘 
		while(!ripeTomatos.empty()){
			
			pair<int, int> cell = ripeTomatos.front();
			ripeTomatos.pop();
			
			for(int i = 0; i < 4; i++){
				
				int nextRow = cell.first + directions[i].first;
				int nextCol = cell.second + directions[i].second;
				
				// 범위를 벗어난 경우 
				if(nextRow == -1 || nextRow == height || nextCol == -1 || nextCol == width)
					continue;
				// 빈 곳이거나 이미 익은 공간인 경우 
				else if(grid[nextRow][nextCol] == -1 || grid[nextRow][nextCol] == 1)
					continue;
				else{
					// 현재 익어 있는 토마토 덕분에 익은 토마토를 affectedTomatos 큐에 집어넣음
					affectedTomatos.push(make_pair(nextRow, nextCol));
					grid[nextRow][nextCol] = 1;
					ripeTomatoSize++;
				}	
			}		
			
			// 모든 토마토가 다 익었으면 탈출
			if(ripeTomatoSize == gridSize)
				return day; 
			
		}
		
		// 오늘 익은 토마토들(affectedTomatos)을 이미 익은 토마토(ripeTomatos)들로 옮김 
		swap(affectedTomatos, ripeTomatos);
		
		day++;
	}
	
	return -1;
} 

int main(int argc, char** argv) {

	cin >> width >> height;
	bool allRipen = true;
	
	for(int i = 0; i < height; i++){
		for(int j = 0; j < width; j++){
			cin >> grid[i][j];
			if(grid[i][j] == 0) 
				allRipen = false;
			else if(grid[i][j] == 1){
				ripeTomatos.push(make_pair(i, j));
				ripeTomatoSize++;
			}
			else if(grid[i][j] == -1){
				gridSize--;
			}
		}
	}
	
	gridSize += width * height;

	if(allRipen == true){
		cout << 0;
	}	
	else cout << BFS();
	
	return 0;
}

C++ 코테 준비 같이 해요!

답변 2

1

김태원

안녕하세요^^

아무래도 입력량이 많기 때문에 입출력에서 오는 시간차이인것 같습니다.

cin과 cout으로 할 때는  아래 코드처럼 ios_base::sync_with_stdio(false); 를 추가하면 cin과 cout이 scanf나 printf보다 빨라집니다. 자세한 사항은 섹션5의 첫번째 영상에서 설명했습니다.

#include<bits/stdc++.h>
using namespace std;
int main(){
	ios_base::sync_with_stdio(false);
	int n, t, tmp;
	cin>>n>>t;

0

Eunyoung Roh

(추가) 선생님 코드의 printf, scanf를 각각 cout와 cin으로 대체하니 저와 비슷한 수행시간이 나옵니다. 혹시 입출력 방식이 수행시간에 영향을 주었다고 생각해도 될까요?

테스트 케이스 질문

0

372

1

병합정렬 시간복잡도 질문

0

461

1

41.연속된 자연수의 합 문제풀이에서 수학적인 원리를 모르고 있습니다.

0

1343

2

질문드립니다.

0

375

1

질문드립니다!

0

429

1

dev 프로그램 질문

0

275

1

문제가 이해가 안되요

0

375

1

4번 나이차이 문제 접근법 질문 드립니다.

0

306

1

source file not compiled

0

1043

3

59번 질문드립니다.

0

371

1

25번 문제 질문

0

348

1

4. 나이차이 문제 질문입니다.

0

371

1

90번 라이언 킹 심바 1번 테스트 케이스

0

469

1

71번 문제 전역 변수 질문 있습니다

0

364

1

75번, 79번 priority_queue관련

1

354

1

75.최대 수입 스케줄

0

399

2

복면산 정답의 수

0

430

1

테스트 케이스에 대해서

0

444

1

수업 내용 질문입니다!

1

231

1

풀어보면 좋은 문제 목록 - 2580 스토쿠 DFS 질문입니다!!

0

821

2

12. 플로이드-와샬(그래프 최단거리) . 27:25초

0

253

1

다른 풀이 방식

0

316

1

크루스칼 vs 프림

0

305

1

숫자 총개수 small 질문있습니다.

0

241

1