본문 바로가기

백준

백준 2671 잠수함식별(파이썬)

문제

코드

import re

def confirm(pattern):
    return bool(re.fullmatch('(100+1+|01)+', pattern))

def sol():
    answer = confirm(input())
    if answer:
        print("SUBMARINE")
    else:
        print("NOISE")

sol()

풀이

처음에는 입력값을 하나씩 풀어서 제공하려고 하였으나 제출하는 중 실패가 나와 검색을 해보았고 정규식 모듈(Regular Expression 줄여서 re)가 있는 걸을 알게 되었다. 해당 문제는 re 모듈을 사용하면 쉽게 해결이 가능한 문제였다. 이후 re모듈에 대해서 익숙해질 필요가 있을듯 하다.