티스토리 뷰

728x90

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

 

1013번: Contact

입력의 첫 줄에는 테스트 케이스의 개수 T가 주어진다. 그 다음 줄부터 각각의 테스트 케이스에 대해 전파를 표현하는, { 0, 1 }만으로 이루어진 문자열이 공백 없이 주어진다. 문자열 길이는 (1 ≤

www.acmicpc.net

 

정규식을 사용하기 위해 re 모듈을 import 합니다.

패턴을 만들고 일치하는지 확인할 때, match()를 사용하면 AttributeError가 발생하므로

시작과 끝이 패턴과 정확히 일치할때만 반환하는 fullmatch() 메소드를 이용하면 결과가 제대로 출력됩니다.

match() 메소드는 처음부터 일치하면 결과를 반환합니다.

1
2
3
4
5
6
7
8
9
import re
 
for _ in range(int(input())):
    S = input()
    P = re.compile('(100+1+|01)+')
    if P.fullmatch(S):
        print('YES')
    else:
        print('NO')
cs

 

댓글