24-alstjr7437

[alstjr7437]

🔗 문제 링크

토마토

✔️ 소요된 시간

20분

🧐 TMI

인터넷 참고 X
다른 코드 참고 X
원트만에 푸니까 기분이 뿌듯하구만유

어느정도 기본적인 BFS문제는 이제 풀 수 있을듯..

✨ 수도 코드

처음 생각한 부분으로는

위 사진과 같이 동작하게 생각했습니다.

1. tomato에서 1인 부분 찾아서 queue에 넣기
2. BFS 돌리기(2~4)
3. Visited 체크해주기
4. 익게 되는 tomato의 숫자를 옮겨지는 토마토의 숫자에 +1을 하기
   • 이 부분을 안하게 되면 BFS가 돌때마다 숫자가 커지거나 1만 되서 체크하기 힘들어짐
5. 토마토에서 0이 있는지 찾기
   • 가지 못하는 곳이 있으면 -1 출력
6. 토마토에서 최대 값 찾기
   • 최대 값에서 첫날은 빼야하므로 -1

방식으로 풀었습니다!!!!

from collections import deque
import sys

input = sys.stdin.readline

dx = [0,0,-1,1]
dy = [1,-1,0,0]

n, m = map(int, input().split())
tomato = []
visited = [[0] * n for _ in range(m)]

for _ in range(m):
    tomato.append(list(map(int, input().split())))

queue = deque()

# 1. 1인 부분 queue에 넣기
for y in range(m):
    for x in range(n):
        if tomato[y][x] == 1:
            queue.append((y,x))
            visited[y][x] = 1

# 2. BFS Go
while queue:
    y, x = queue.popleft()
    for i in range(4):
        nx = x + dx[i]
        ny = y + dy[i]
        if nx >= n or nx < 0 or ny >= m or ny < 0:
            continue
        # 3. visited 체크 
        if tomato[ny][nx] != -1 and visited[ny][nx] == 0:
            visited[ny][nx] = 1
            # 4. 익게 된 토마토의 수를 옮겨지는 토마토의 수 + 1
            tomato[ny][nx] = tomato[y][x] + 1
            queue.append((ny, nx))

# 5. 0이 있으면 -1 출력
if any(0 in l for l in tomato):
    print(-1)
# 6. 최대값 찾아서 -1 하여 출력
else : 
    test = max(map(max, tomato))
    print(test - 1) 

📚 새롭게 알게된 내용

그리고 VSCode에서 하면 tempCodeRunnerFile이 한번씩 생겨서 아래 설정을 켜서 안생기게 해줬습니다!!!

첫 PR인데 잘부탁드립니다!!🔥🔥🔥🔥🔥🔥🔥🔥🔥🔥🔥🔥🔥🔥🔥🔥

[tgyuuAn]

민석시치 Empty File!!!!!!!

[tgyuuAn]

from collections import deque
import sys

def input(): return sys.stdin.readline().rstrip()

M, N = map(int ,input().split())

tomato_box = []
ripe_tomatoes = deque()
visited = set()
for row in range(N):
    row_tomato = list(map(int, input().split()))

    for col, tomato in enumerate(row_tomato):
        if tomato == 1: 
            visited.add((row, col))
            ripe_tomatoes.append((row, col, 0))

    tomato_box.append(row_tomato)

dx = [0, 0, -1, 1]
dy = [-1, 1, 0, 0]

answer = 0
while ripe_tomatoes:
    now_row, now_col, now_day = ripe_tomatoes.popleft()
    answer = now_day
    
    for dir in range(4):
        new_row = now_row + dy[dir]
        new_col = now_col + dx[dir]
        
        if new_row < 0 or new_row >= N: continue
        if new_col < 0 or new_col >= M: continue
        if (new_row, new_col) in visited: continue
        if tomato_box[new_row][new_col] != 0: continue
    
        tomato_box[new_row][new_col] = 1
        visited.add((new_row, new_col))
        ripe_tomatoes.append((new_row, new_col, now_day+1))

flag = False
for row in range(N):
    for col in range(M):
        if tomato_box[row][col] == 0:
            flag = True
            break
    
    if flag: break
    
print(-1) if flag == True else print(answer)

짠!!!!!!!!!!!!!!!!

[alstjr7437]

짠!!!!!!!!!!!!!!!!

아니 10시 50분에 푸려고 하니까 댓글에 이미 스포가 되있더라구요
바로 눈 질끈 감고 풀러 갔습니다 덕분에

@tgyuuAn
다시 리뷰 하십쇼

[tgyuuAn]

잘한다....................

👍👍👍👍

[tgyuuAn]

오...................................................................... any 맛있는데요 ...?

[H0ngJu]

22 이렇게 하면 굳이 for문을 두개 중첩할 필요가 없네요 👏👏

[alstjr7437]

저도 처음에는 이중 반복문으로 만들어서 했다가 그렇게 찾으면 변수를 하나 더만들어야 겠더라구요..
그래서 any라는 함수를 써봤습니다~~~~

[tgyuuAn]

오호 날짜 정보를 tomato 에 더하는 식으로 저장했군요 좋은디요 ?????????

[H0ngJu]

22 아 그냥 이전 tamao 에서 +1 해버리면 따로 days 저장할 필요가 없겠네요 ㄷㄷ

[alstjr7437]

이것도 처음에 max_days해서 넣었는데 큐에 같이 저장을 안하면 그냥 BFS 돌때마다 +1이 되버려서 출력이 이상해지길래 어떡하지 하다가 이방법으로 풀었습니다!!!~!

[H0ngJu]

나중에 풀어봐야지 ~ 하고 까먹고 있었는데 이번 기회에 풀었습니닷

max_days 처리에서 고민 쫌 했네용.. 근데 민석님 코드 보니까 굳이 따로 저장안하고 바로 방향 탐색 전 tomato에 +1만 해주면 되네요 ㄷㄷ

문제 푸느라 수고하셨습니닷 ~~ ‼️‼️💪

import sys
from collections import deque

def input(): return sys.stdin.readline().rstrip()

M, N = map(int, input().split())
tomatos = [[int(i) for i in input().split()] for _ in range(N)]
visited = [[0 for _ in range(M)] for _ in range(N)]
dir = [(0,1), (0,-1), (1,0), (-1,0)]
q = deque()
max_days = 0
check = True # 익지 않은 토마토가 있는지

# 익은 토마토 넣기
for i in range(N):
    for k in range(M):
        if tomatos[i][k] == 1:
            q.append((0, i,k))
            visited[i][k] == 1

while q:
    days, x, y = q.popleft()
    for dir_x, dir_y in dir:
        if 0<= x+dir_x < N and 0<= y+dir_y < M:
            if tomatos[x+dir_x][y+dir_y] == 0 and visited[x+dir_x][y+dir_y] == 0:
                tomatos[x+dir_x][y+dir_y] = 1 # 익은 토마토 만들기
                visited[x+dir_x][y+dir_y] = 1
                max_days = days+1
                q.append((days +1, x+dir_x, y+dir_y))

for i in range(N):
    for k in range(M):
        if tomatos[i][k] == 0:
            check = False

if check:
    print(max_days)
else:
    print(-1)
        

[H0ngJu]

22 아 그냥 이전 tamao 에서 +1 해버리면 따로 days 저장할 필요가 없겠네요 ㄷㄷ

[H0ngJu]

22 이렇게 하면 굳이 for문을 두개 중첩할 필요가 없네요 👏👏

[H0ngJu]

오호라.. 요렇게 하면 tomato에 대한 최대값을 바로 찾을 수 있군요..
코드가 간결해서 좋네요 ㄷㄷ 👍 배워갑니다 (_ _)

[tgyuuAn]

나중에 풀어봐야지 ~ 하고 까먹고 있었는데 이번 기회에 풀었습니닷

max_days 처리에서 고민 쫌 했네용.. 근데 민석님 코드 보니까 굳이 따로 저장안하고 바로 방향 탐색 전 tomato에 +1만 해주면 되네요 ㄷㄷ

문제 푸느라 수고하셨습니닷 ~~ ‼️‼️💪

import sys

from collections import deque



def input(): return sys.stdin.readline().rstrip()



M, N = map(int, input().split())

tomatos = [[int(i) for i in input().split()] for _ in range(N)]

visited = [[0 for _ in range(M)] for _ in range(N)]

dir = [(0,1), (0,-1), (1,0), (-1,0)]

q = deque()

max_days = 0

check = True # 익지 않은 토마토가 있는지



# 익은 토마토 넣기

for i in range(N):

    for k in range(M):

        if tomatos[i][k] == 1:

            q.append((0, i,k))

            visited[i][k] == 1



while q:

    days, x, y = q.popleft()

    for dir_x, dir_y in dir:

        if 0<= x+dir_x < N and 0<= y+dir_y < M:

            if tomatos[x+dir_x][y+dir_y] == 0 and visited[x+dir_x][y+dir_y] == 0:

                tomatos[x+dir_x][y+dir_y] = 1 # 익은 토마토 만들기

                visited[x+dir_x][y+dir_y] = 1

                max_days = days+1

                q.append((days +1, x+dir_x, y+dir_y))



for i in range(N):

    for k in range(M):

        if tomatos[i][k] == 0:

            check = False



if check:

    print(max_days)

else:

    print(-1)

        

홍쥬르 나랑 완존 비슷 대박

[alstjr7437]

나중에 풀어봐야지 ~ 하고 까먹고 있었는데 이번 기회에 풀었습니닷

max_days 처리에서 고민 쫌 했네용.. 근데 민석님 코드 보니까 굳이 따로 저장안하고 바로 방향 탐색 전 tomato에 +1만 해주면 되네요 ㄷㄷ

문제 푸느라 수고하셨습니닷 ~~ ‼️‼️💪

저도 날짜 처리 부분에서 처음에 max_days를 다 1씩 추가하면서 했는데 그렇게 하니까 예제에서 25인가 나오더라구요 ㄷㄷ

그래서 위와 같이 방식을 바꿨습니다! 어짜피 0이 있으면 돌아가니까 1로 바꿔주지 않고 날짜로 바꿔줬습니당

[SeongHoonC]

먼저 첫번째 방법은 민석님이랑 거의 똑같이 풀었습니다. 대신 visited 는 쓰지 않았구요.

private val dx_1 = listOf(0, 0, 1, -1)
private val dy_1 = listOf(1, -1, 0, 0)
fun main() {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val (m, n) = br.readLine().split(" ").map { it.toInt() }
    val graph = Array(n) { Array(m) { 0 } }
    val q = ArrayDeque<Pair<Int, Int>>()

    repeat(n) {
        val line = br.readLine().split(" ").map { it.toInt() }
        line.forEachIndexed { index: Int, item: Int ->
            graph[it][index] = item
            if (item == 1) {
                q.add(it to index)
            }
        }
    }

    while (q.isNotEmpty()) {
        val now = q.removeFirst()
        for (i in 0..3) {
            val nextX = now.first + dx_1[i]
            val nextY = now.second + dy_1[i]
            if (nextX < 0 || nextY < 0 || nextX >= n || nextY >= m) {
                continue
            }
            if (graph[nextX][nextY] != 0) {
                continue
            }
            graph[nextX][nextY] = graph[now.first][now.second] + 1
            q.add(nextX to nextY)
        }
    }

//flatten() -> 2차원을 1차원으로 바꾼다
    val flattenGraph = graph.flatten()
    if (flattenGraph.contains(0)) println(-1) else println(flattenGraph.max() - 1)
}

마지막에 max 값 계산이나 0을 찾는 부분에서 시간을 더 빠르게 할 수 있는 방법이 없을까 생각했습니다.

1. 처음 데이터 받을 때 0 인 것들 (안익은 토마토) 들 의 개수를 저장해놓고 하나 탐색할 때마다 count --
2. max 를 1로 초기화하고 계속 업데이트

private val dx_1 = listOf(0, 0, 1, -1)
private val dy_1 = listOf(1, -1, 0, 0)
fun main() {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val (m, n) = br.readLine().split(" ").map { it.toInt() }
    val graph = Array(n) { Array(m) { 0 } }
    val q = ArrayDeque<Pair<Int, Int>>()

    // 요거 추가
    var tomatoCount = 0
    var maxDate = 1

    repeat(n) {
        val line = br.readLine().split(" ").map { it.toInt() }
        line.forEachIndexed { index: Int, item: Int ->
            graph[it][index] = item
            if (item == 1) q.add(it to index)
            if (item == 0) tomatoCount++ // 요거 추가
        }
    }

    while (q.isNotEmpty()) {
        val now = q.removeFirst()
        for (i in 0..3) {
            val nextX = now.first + dx_1[i]
            val nextY = now.second + dy_1[i]
            if (nextX < 0 || nextY < 0 || nextX >= n || nextY >= m) {
                continue
            }
            if (graph[nextX][nextY] != 0) {
                continue
            }
            val date = graph[now.first][now.second] + 1
            graph[nextX][nextY] = date
            q.add(nextX to nextY)
            tomatoCount-- // 요거 추가
            maxDate = date // 요거 추가
        }
    }

    if (tomatoCount > 0) println(-1) else println(maxDate - 1) // 이제 계산 안해도 됨!
}

실제로 조금 줄었습니다..유의미한 차이인지는 모르겠지만

토마토 재밌었네요!

[SeongHoonC]

날짜 정보를 토마토에 더한다면 visited 없이 가능할 것 같네요! 어차피 0이면 탐색하지 않은 토마토일테니까요~

Suggested change

	if tomato[ny][nx] != -1 and visited[ny][nx] == 0:
	visited[ny][nx] = 1
	tomato[ny][nx] = tomato[y][x] + 1
	queue.append((ny, nx))
	if tomato[ny][nx] == 0:
	tomato[ny][nx] = tomato[y][x] + 1
	queue.append((ny, nx))