백준 1013 : Contact

www.acmicpc.net/problem/1013

1013번: Contact

 

해결 방법 : 정규표현식

• 정규표현식 fullmatch로 전체 문자열이 해당 포맷으로 구성되는지 판별하면 된다.

import sys
input = sys.stdin.readline
import re

def solution(s):
    c = re.compile(r'(100+1+|01)+')
    if c.fullmatch(s):
        return "YES"
    return "NO"

if __name__ == '__main__':
    n = int(input())
    for i in range(n):
        result = solution(input().strip())
        print(result)

 

 

• 처음에는 다음과 같이 match를 이용해서 구현해봤는데 알 수 없는 이유로 몇몇 테스트케이스를 통과하지 못한다.
  • 정규표현식 +는 탐욕적 수량자이기 때문에 match 판단 이후, (match.group() == s)하여 match된 문자열이 원본 문자열과 일치하는지 확인하는식으로 했었다. 논리상 맞는 구현 같은데 이유를 잘 모르겠다. match의 경우 검사 가능 횟수가 제한되어 있는건지..(혹시 이유를 아는분이 계시다면 댓글로 알려주시면 감사드리겠습니다)

    c = re.compile(r'(100+1+|01)+')
    if c.match(s):
        if c.match(s) == s:
        	return "YES"
    return "NO"