[프로그래머스] 자동완성 (트라이)

2023. 9. 7. 16:02·Coding Test/programmers
728x90

https://school.programmers.co.kr/learn/courses/30/lessons/17685?language=python3 

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

분석

학습된 단어들이 아래와 같을 때
- go
- gone
- guild

go를 찾을 때 go를 모두 입력해야 한다.
gone을 찾을 때 gon 까지 입력해야 한다. (gon이 입력되기 전까지는 go 인지 gone인지 확신할 수 없다.)
guild를 찾을 때는 gu 까지만 입력하면 guild가 완성된다.

이 경우 총 입력해야 할 문자의 수는 7이다.

 

 2023.06.05 - [BOJ/Tree] - [14425] 문자열 집합 (트라이, Trie) 

 

[14425] 문자열 집합 (트라이, Trie)

14425번: 문자열 집합 첫째 줄에 문자열의 개수 N과 M (1 ≤ N ≤ 10,000, 1 ≤ M ≤ 10,000)이 주어진다. 다음 N개의 줄에는 집합 S에 포함되어 있는 문자열들이 주어진다. 다음 M개의 줄에는 검사해야 하

karla.tistory.com

 

풀이

# 노드 클래스 생성
class Node(object):
    def __init__(self, wordCnt):
        self.wordCnt=0 # 만들 수 있는 단어수
        self.childNode={}  # 자식노드

class Trie(object):
    def __init__(self):
        self.parent=Node(None)  # 부모노드 저장 변수

    # 문자 삽입
    def insert(self, string):
        nowNode=self.parent
        for char in string:
            nowNode.wordCnt+=1
            if char not in nowNode.childNode:
                nowNode.childNode[char]=Node(char)
            nowNode=nowNode.childNode[char]  # 자식노드로 이동
        nowNode.wordCnt+=1
        
    # 문자열 존재하는지 탐색
    def search(self, string):
        nowNode=self.parent
        for char in string:
            if char in nowNode.childNode:
                nowNode=nowNode.childNode[char]
            else:
                return False
        if nowNode.wordCnt==1:
            return True
        else:
            return False

def solution(words):
    answer = 0

    myTrie=Trie()  # Trie 생성
    for x in words: 
        myTrie.insert(x)   # 단어 삽입
        
    for word in words:
        already_find=False
        for i in range(1, len(word)+1):
            if myTrie.search(word[:i]):
                answer+=len(word[:i])
                already_find=True
                break
                
        if not already_find:
            answer+=len(word)
            
    return answer

 

 

728x90
'Coding Test/programmers' 카테고리의 다른 글
  • [Java] 삼각 달팽이 (2차원배열, 방향)
  • [Java] 교점에 별 만들기 (2차원배열, 최대사각형, 좌표 배열 전환)
  • [프로그래머스] 야근지수 (힙)
  • [프로그래머스] 매출 하락 최소화 (DP, 트리)
Karla Ko
Karla Ko
𝘾𝙤𝙣𝙩𝙞𝙣𝙪𝙤𝙪𝙨𝙡𝙮 𝙄𝙢𝙥𝙧𝙤𝙫𝙞𝙣𝙜, 𝘾𝙤𝙣𝙨𝙩𝙖𝙣𝙩𝙡𝙮 𝘿𝙚𝙫𝙚𝙡𝙤𝙥𝙞𝙣𝙜 𝙔𝙚𝙨!
    250x250
  • Karla Ko
    karlaLog
    Karla Ko
  • 전체
    오늘
    어제
    • Total (467)
      • Spring (19)
      • JPA (4)
      • Cloud & Architecture (15)
        • Kubernetes (5)
        • Docker (3)
        • MSA (2)
        • GCP (1)
        • AWS (4)
      • Devops (1)
      • Message Queue (4)
        • Kafka (2)
        • RabbitMQ (2)
      • Git (4)
      • DB (4)
      • Java (9)
      • Python (4)
      • CS (11)
        • OS (8)
        • Network (2)
        • Algorithm (1)
      • Coding Test (392)
        • programmers (156)
        • Graph (43)
        • DP (37)
        • Search (31)
        • Tree (13)
        • Data Structure (26)
        • Combination (12)
        • Implement (18)
        • Geedy (23)
        • Sort (7)
        • Math (21)
        • geometry (2)
  • 블로그 메뉴

    • 홈
  • 링크

  • 공지사항

  • 인기 글

  • 태그

    DP
    동적계획법
    최단거리
    BFS
    큐
    프로그래머스
    알고리즘
    그래프
    덱
    힙
    Algorithm
    트리
    구현
    스택
    플로이드워셜
    자료구조
    파이썬
    재귀
    최대공약수
    구간합
    백준
    월간코드챌린지
    조합
    다익스트라
    그리디
    LIS
    정렬
    DFS
    이분탐색
    최소신장트리
  • hELLO· Designed By정상우.v4.10.3
Karla Ko
[프로그래머스] 자동완성 (트라이)
상단으로

티스토리툴바