[백준] 14502번 연구소 - 파이썬(Python)

2021. 10. 1. 09:39· 알고리즘/문제풀이
목차
  1. 문제 (링크)
  2. 나의 풀이
  3. 코드 설명 및 풀이법
728x90

문제 (링크)

https://www.acmicpc.net/problem/14502

 

14502번: 연구소

인체에 치명적인 바이러스를 연구하던 연구소에서 바이러스가 유출되었다. 다행히 바이러스는 아직 퍼지지 않았고, 바이러스의 확산을 막기 위해서 연구소에 벽을 세우려고 한다. 연구소는 크

www.acmicpc.net

 

나의 풀이

import sys
from collections import deque
from itertools import combinations
import copy

input = sys.stdin.readline

n, m = map(int, input().split())
graph = []
virus_locs = []
empty_locs = []
moves = [
    (1, 0),
    (-1, 0),
    (0, 1),
    (0, -1)
]

for i in range(n):
    row = list(map(int, input().split()))
    for j in range(m):
        if row[j] == 2:
            virus_locs.append([i, j])
        
        elif row[j] == 0:
            empty_locs.append([i, j])

    graph.append(row)

def bfs(copied_graph, virus_locs):
    q = deque(virus_locs)

    while q:
        x, y = q.popleft()

        for move in moves:
            nx, ny = x + move[0], y + move[1]

            if 0 <= nx < n and 0 <= ny < m:
                if copied_graph[nx][ny] == 0:
                    copied_graph[nx][ny] = 2
                    q.append([nx, ny])

    return copied_graph

# 빈 공간에서 3개를 뽑는 경우의 수 (3개의 벽을 세우는 경우의 수)
comb = combinations(empty_locs, 3)
ans = -1

for walls in comb:
    x1, y1 = walls[0][0], walls[0][1]
    x2, y2 = walls[1][0], walls[1][1]
    x3, y3 = walls[2][0], walls[2][1]

    # 매번 기존 그래프에서 진행해줘야하므로 copy해서 진행
    copied_graph = copy.deepcopy(graph)

    copied_graph[x1][y1] = 1
    copied_graph[x2][y2] = 1
    copied_graph[x3][y3] = 1

    graph_with_walls = bfs(copied_graph, virus_locs)
    tmp_ans = 0

    for each in graph_with_walls:
        # 각 열에 대해서 0의 갯수 세서 더해줌
        tmp_ans += each.count(0)
    
    ans = max(ans, tmp_ans)

print(ans)

코드 설명 및 풀이법

처음에는 벽을 가장 효과적으로 세울 수 있는 방법에 대해서 오랜 시간 고민했지만, N과 M의 범위가 각각 3이상 8이하라는 점에서 브루트포스 방식을 통해 해결할 수 있는 문제였다.

 

처음 그래프를 입력받을 때 바이러스가 있는 위치와 빈 위치(벽을 세울 수 있는 위치)를 따로 저장해준다.

 

BFS함수에서는 바이러스 위치부터 시작해 BFS를 진행해주고, 진행된 곳에 대해서는 바이러스에 감염됐다는 뜻으로 값을 2로 업데이트 해줬다.

 

앞서 저장해둔 빈 위치에서 3가지 위치를 뽑는 경우의 수를 itertools.combinations함수를 통해서 뽑아준 후, 반복문을 돌며 해당 조합들의 값을 1로 업데이트 해주어 벽을 세워준다.

 

이렇게 벽이 세워진 그래프에 대해서 앞서 정의한 BFS 함수를 돌려준 후 리턴된 그래프에서 0의 갯수를 세어 최대값을 구하도록 작성해준다.

저작자표시 (새창열림)

'알고리즘 > 문제풀이' 카테고리의 다른 글

[백준] 14503번 로봇청소기 - 파이썬(Python)  (0) 2021.10.03
[백준] 1450번 냅색문제 - 파이썬(Python)  (0) 2021.10.01
[백준] 9663번 N-Queen - 파이썬(Python)  (13) 2021.09.30
[백준] 1208번 부분수열의 합 2 - 파이썬(Python)  (0) 2021.09.30
[백준] 12865번 평범한 배낭 - 파이썬(Python)  (0) 2021.09.29
  1. 문제 (링크)
  2. 나의 풀이
  3. 코드 설명 및 풀이법
'알고리즘/문제풀이' 카테고리의 다른 글
  • [백준] 14503번 로봇청소기 - 파이썬(Python)
  • [백준] 1450번 냅색문제 - 파이썬(Python)
  • [백준] 9663번 N-Queen - 파이썬(Python)
  • [백준] 1208번 부분수열의 합 2 - 파이썬(Python)
SeongOnion
SeongOnion
서버는 꺼지지 않아요
SeongOnion
조무래기 코딩
SeongOnion
전체
오늘
어제
  • 분류 전체보기 (167)
    • 알고리즘 (81)
      • 이론 (8)
      • 문제풀이 (73)
    • 언어 (15)
      • Python (9)
      • JavaScript (1)
      • JAVA (5)
    • 데이터베이스 (5)
    • 프레임워크 (15)
      • Django (7)
      • Spring (8)
    • 그 외 공부 (38)
      • 운영체제 (1)
      • 자료구조 (14)
      • 네트워크 (5)
      • CS (2)
      • 기타 (7)
      • 트러블 슈팅 (9)
    • 프로젝트 (0)
    • 개발자취 (8)
    • 회고 (3)
    • 주저리주저리 (1)
    • 기타 (비개발) (1)

블로그 메뉴

  • 홈
  • 태그
  • 방명록

공지사항

인기 글

태그

  • 컨트리뷰트
  • 파이썬
  • 알고리즘
  • 투 포인터 알고리즘
  • DRF
  • DP
  • 데이터베이스
  • 프로그래머스
  • BFS/DFS
  • Django
  • 웹
  • 회고
  • 그리디알고리즘
  • 큐
  • 에라토스테네스의 체
  • 백준
  • 코딩
  • 트러블 슈팅
  • 장고
  • 자바
  • 브루트포스
  • 개발자
  • 코딩테스트
  • 소수
  • 이진탐색
  • BFS
  • 정렬 알고리즘
  • 스택
  • spring
  • 오픈소스

최근 댓글

최근 글

hELLO · Designed By 정상우.v4.2.2
SeongOnion
[백준] 14502번 연구소 - 파이썬(Python)
상단으로

티스토리툴바

단축키

내 블로그

내 블로그 - 관리자 홈 전환
Q
Q
새 글 쓰기
W
W

블로그 게시글

글 수정 (권한 있는 경우)
E
E
댓글 영역으로 이동
C
C

모든 영역

이 페이지의 URL 복사
S
S
맨 위로 이동
T
T
티스토리 홈 이동
H
H
단축키 안내
Shift + /
⇧ + /

* 단축키는 한글/영문 대소문자로 이용 가능하며, 티스토리 기본 도메인에서만 동작합니다.