백준 1013(Contact) - 해결

Contact

 

문제

“무한히 넓은 저 우주에 인류만이 홀로 존재한다면, 그건 정말 슬픈 일이 아닐까요”

푸에르토리코 아레시보에 위치한 아레시보 전파망원경(Arecibo radio telescope)은 수십 년째 존재하지 않을 지도 모르는 외계 문명으로부터의 전파를 수신하기 위해 밤하늘을 바라보고 있다.

이 망원경이 수집한 전파 속에서 자연적으로 발생하기 힘든 패턴들을 찾아내어, 그것을 증거로 외계 문명의 존재 여부를 가리려는 노력은 줄곧 이어져왔지만 아직까지도 그러한 패턴은 발견되지 않았다. 한국 천문학계의 자존심 김동혁 박사는 국내 기술로 이러한 탐사를 진행하기 위하여 다음의 전파 표기를 표준으로 삼았다.

전파의 기본 단위는 { 0 , 1 } 두 가지로 구성되어있으며, x+ (  ) 는 임의의 개수(최소 1개) x의 반복으로 이루어진 전파의 집합을 나타낸다.

(xyx)+ (  ) 는 괄호 내의 xyx의 반복으로 이루어진 전파의 집합을 뜻한다. 아래는 이해를 돕기 위한 예제이다.

• 1+ = { 1, 11, 111, 1111, 11111, … }
• 10+ = { 10, 100, 1000, 10000, 100000, … }
• (01)+ = { 01, 0101, 010101, 01010101, 0101010101, … }
• (1001)+ = { 1001, 10011001, 100110011001, … }
• 10+11 = { 1011, 10011, 100011, 1000011, 10000011, … }
• (10+1)+ = { 101, 1001, 10001, 1011001, 1001101, 100011011000001, … }

반복을 의미하는 + 외에도 or 를 의미하는 | 기호가 있다. { x | y } 는 x 혹은 y 를 의미하는 것으로, { 0+ | 1+ } 는 { 0 , 1 , 00 , 11 , 000 , 111 , … } 의 집합을 의미한다. 아래는 두 기호를 복합적으로 사용한 예이다.

• (100 | 11)+ = { 100 , 11 , 10011 , 11100 , 1110011100 , 100111111100100, … }

최근 김동혁 박사는 아레시보 전파망원경에서 star Vega(직녀성) 으로부터 수신한 전파 기록의 일부를 조사하여 그 전파들의 패턴을 분석하여 아래와 같이 기록하였다.

• (100+1+ | 01)+

김동혁 박사는 다양한 전파 기록 중에서 위의 패턴을 지니는 전파를 가려내는 프로그램을 필요로 한다. 이를 수행할 수 있는 프로그램을 작성하라.

 

입력

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

 

출력

각 테스트 케이스에 대해 주어진 전파가 문제에서 제시한 패턴이면 “YES”를 그렇지 않은 경우는 “NO”를 출력한다. 출력 문자열은 모두 대문자로 구성되어 있다.

---

분기식 처럼 표현하였다.

처음 2갈래는 0인지 1인지 파악하기

0이면 뒤가 1인지 아닌지 확인하기 

- 01이면 다음으로 

- 00 이면 NO

그리고 1이면뒤에 00이 오는지 확인

그리고 0이 더있는지 확인

그리고 1이 나오면 하나는 확인 

뒤에 1이 더있는 확인

더 있는 1이 지금의 패턴에 포함 되는지 다음 패턴에 포함되야 할지 파악

(10011001 과 1001111 의 차이)

import sys
input = sys.stdin.readline

N = int(input().strip())

for _ in range(N):
    pattern = list(map(int,input().strip()))
    idx = 0
    p_len = len(pattern)

    while idx < p_len:
        # 1 분기
        # 1
        if pattern[idx]:
            # 1-1 분기 남은 길이가 2이상이고 그 2개가 00일때
            # 100
            if idx + 2 < p_len and pattern[idx + 1] == pattern[idx + 2] == 0:
                idx += 2

                # 1-1-1 뒤에 0이 더있는지 확인
                while idx + 1 < p_len and pattern[idx] == 0:
                    idx += 1
                
                # 이미 길이를 넘으면 끝
                # 100..0
                if idx == p_len:
                    ans = "NO"
                    break

                # 뒤에 1이 있는지 확인
                # 100..1
                if idx + 1 < p_len and pattern[idx] == 1:
                    idx += 1
                
                # 1-1-2 뒤에 1이 더있는지 확인
                # 100..1
                while idx + 1 <= p_len and pattern[idx] == 1:
                    # 1-1-2-1 뒤에 길이가 2가 더 있는지 확인
                    # 다음이 0이 있는 확인
                    # 100..1..10
                    if idx + 2 < p_len and pattern[idx + 1] == 0:
                        # 1-1-2-1-1 뒤에 또 0이 있는지 확인
                        # 다시 첫번째 패턴이 나올 수 있는경우
                        # 100..1..100 
                        if pattern[idx + 2] == 0:
                            break
                        # 1-1-2-1-2
                        # 뒤에 2번째 패턴인 경우 해당 1은 현재 패턴의 것
                        # 100..1..101
                        else:
                            idx += 1
                            break
                    # 1-1-2-2 뒤에 1이 더있는 경우
                    # 100..11..
                    idx += 1

            # 1-2 분기 1다음에 00이 아니면 패턴에 맞지 않음
            # NO
            else:
                ans = "NO"
                break

        # 2 분기
        # 0
        else:
            # 2-1 0다음에는 무조건 1이 와야한다.
            # 01
            if idx + 1 < p_len and pattern[idx + 1]:
                idx += 2
            # 2-2 
            # 00 NO
            else:
                ans = "NO"
                break

    # 무사히 다 돌면 
    # YES
    else:
        ans = 'YES'

    print(ans)