CS/알고리즘 문제 풀이

[백준] 14725 - 개미굴 [Python(파이썬)]

2021. 1. 1. 19:32
목차
  1. 문제
  2. 풀이
  3. 코드

문제

www.acmicpc.net/problem/14725

 

14725번: 개미굴

첫 번째 줄은 로봇 개미가 각 층을 따라 내려오면서 알게 된 먹이의 정보 개수 N개가 주어진다.  (1 ≤ N ≤ 1000) 두 번째 줄부터 N+1 번째 줄까지, 각 줄의 시작은 로봇 개미 한마리가 보내준 먹이

www.acmicpc.net

 

풀이

먹이(단어)를 하나의 노드로 간략하게 트라이(Trie)를 구현한다. 처음엔 스택을 이용해서 탐색하려 했으나 백트래킹하는 과정에서 재귀를 사용하는 것이 더 나을 것 같아서 변경했다. 이해를 돕기 위해 아래에 트라이 구현 코드까지 올렸다.

 

코드

문제풀이

import sys

class Trie:
    def __init__(self):
        self.root = {} # children

    def add(self, foods):
        cur = self.root
    
        for food in foods:
            if food not in cur:
                cur[food] = {} # dictonary => children
            cur = cur[food]
        cur[0] = True # leaf node
    
    def travel(self, level, cur):
        if 0 in cur:
            return

        cur_children = sorted(cur)

        for child in cur_children:
            print("--"*level + child)
            self.travel(level+1, cur[child])      

N = int(sys.stdin.readline().strip())
nest = Trie()
for _ in range(N):
    input_line = sys.stdin.readline().split()
    K = input_line[0 ]
    foods = input_line[1:]
    nest.add(foods)
nest.travel(0, nest.root)

 

 

트라이

class Node:
    def __init__(self):
        self.word = False
        self.children = {}
    
class Trie:
    def __init__(self):
        self.root = Node()
    
    def insert(self, word: str) -> None:
        node = self.root
        for char in word:
            if char not in node.children:
                node.children[char] = Node()
            node = node.children[char]
        node.word = True
    
    def search(self, word: str) -> bool:
        node = self.root
        for char in word:
            if char not in node.children:
                return False
            node = node.children[char]

        return node.word
    
    def startsWith(self, prefix: str) -> bool:
        node = self.root
        for char in prefix:
            if char not in node.children:
                return False
            node = node.children[char]
        
        return True
    
    def travel(self, level: int, node: Node) -> None: 
        if node.word:
            return

        cur_children = sorted(node.children)

        for child in cur_children:
            print("--"*level + child)
            self.travel(level+1, node.children[child])

 

저작자표시 (새창열림)

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

[백준] 14502 - 연구소 [Python(파이썬)]  (0) 2021.01.06
[백준] 11052 - 카드 구매하기 [Python(파이썬)]  (0) 2021.01.06
[백준] 10844 - 쉬운 계단 수 [Python(파이썬)]  (3) 2021.01.06
[백준] 1149 - RGB거리 [Python(파이썬)]  (0) 2021.01.06
[백준] 4948 - 베르트랑 공준 [Python(파이썬)]  (0) 2021.01.02
  1. 문제
  2. 풀이
  3. 코드
'CS/알고리즘 문제 풀이' 카테고리의 다른 글
  • [백준] 11052 - 카드 구매하기 [Python(파이썬)]
  • [백준] 10844 - 쉬운 계단 수 [Python(파이썬)]
  • [백준] 1149 - RGB거리 [Python(파이썬)]
  • [백준] 4948 - 베르트랑 공준 [Python(파이썬)]
코택
코택
코택
TaxFree
코택
전체
오늘
어제
  • 분류 전체보기 (369)
    • Spring (29)
      • Spring (18)
      • 스프링 핵심 원리 - 고급편 (11)
    • Spring Batch (4)
    • JPA (4)
    • CS (89)
      • 자료구조 (2)
      • 네트워크 (5)
      • 운영체제 (1)
      • 데이터베이스 (4)
      • SQL (7)
      • 알고리즘 이론 (4)
      • 알고리즘 문제 풀이 (66)
    • 웹 (28)
      • React.js (4)
      • Next.js (1)
      • Node.js (14)
      • FastAPI (4)
      • Django (5)
    • 프로그래밍 언어 (45)
      • Python (5)
      • Java + Kotlin (29)
      • JavaScript + TypeScript (11)
    • 테스트코드 (26)
      • ATDD, 클린 코드 with Spring (4)
      • 이규원의 현실 세상의 TDD: 안정감을 주는 코드.. (20)
    • 인프라 (6)
      • AWS (2)
      • Kubernetes (4)
    • 트러블슈팅 (25)
    • 책 (89)
      • Effective Java (54)
      • Effective Kotlin (14)
      • 도메인 주도 개발 시작하기: DDD 핵심 개념 정.. (11)
      • 웹 프로그래머를 위한 데이터베이스를 지탱하는 기술 (6)
      • 도메인 주도 설계 첫걸음 (4)
    • Git (10)
    • 회고 (5)
    • etc (8)

블로그 메뉴

  • 홈
  • 방명록
  • 관리
  • GitHub
  • LinkedIn

공지사항

  • 스킨 관련

인기 글

태그

  • dp
  • 그래프
  • atdd
  • http
  • Shortest Path
  • 깊이 우선 탐색
  • 브루트포스
  • 장고
  • mysql
  • BOJ
  • fastapi
  • 파이썬
  • 그래프 탐색
  • Git
  • 백준

최근 댓글

최근 글

hELLO · Designed By 정상우.
코택
[백준] 14725 - 개미굴 [Python(파이썬)]
상단으로

티스토리툴바

단축키

내 블로그

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

블로그 게시글

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

모든 영역

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

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