본문 바로가기
알고리즘/[python] 백준 BOJ

[python] 백준 2671 잠수함식별

by Alan_Kim 2023. 8. 26.
728x90
반응형

 

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

 

2671번: 잠수함식별

입력에 들어있는 스트링을 읽고, 이것이 잠수함의 엔진소리를 나타내는 스트링인지 아니면 그냥 물속의 잡음인지를 판정한 후, 잠수함의 엔진 소리에 해당하는 스트링이면 "SUBMARINE"을 출력하고

www.acmicpc.net

 

문제 해결

  • 정규식을 사용할 수 있는지 문제
  • re.compile 로 주어진 정규식 조건을 확인하는 식을 만들 수 있고
  • fullmatch 로 조건을 만족하는지 만족하지 않는지 확인할 수 있다.

 

CODE

import sys
input = sys.stdin.readline
import re

n = str(input().rstrip())
pattern = re.compile("(100+1+|01)+")
res = pattern.fullmatch(n)

if res:
    print("SUBMARINE")
else:
    print("NOISE")
728x90
반응형

댓글