백준 1032 명령 프롬프트

문제

시작 -> 실행 -> cmd를 쳐보자. 검정 화면이 눈에 보인다. 여기서 dir이라고 치면 그 디렉토리에 있는 서브디렉토리와 파일이 모두 나온다. 이때 원하는 파일을 찾으려면 다음과 같이 하면 된다.

dir *.exe라고 치면 확장자가 exe인 파일이 다 나온다. "dir 패턴"과 같이 치면 그 패턴에 맞는 파일만 검색 결과로 나온다. 예를 들어, dir a?b.exe라고 검색하면 파일명의 첫 번째 글자가 a이고, 세 번째 글자가 b이고, 확장자가 exe인 것이 모두 나온다. 이때 두 번째 문자는 아무거나 나와도 된다. 예를 들어, acb.exe, aab.exe, apb.exe가 나온다.

이 문제는 검색 결과가 먼저 주어졌을 때, 패턴으로 뭘 쳐야 그 결과가 나오는지를 출력하는 문제이다. 패턴에는 알파벳과 "." 그리고 "?"만 넣을 수 있다. 가능하면 ?을 적게 써야 한다. 그 디렉토리에는 검색 결과에 나온 파일만 있다고 가정하고, 파일 이름의 길이는 모두 같다.

입력
첫째 줄에 파일 이름의 개수 N이 주어진다. 둘째 줄부터 N개의 줄에는 파일 이름이 주어진다. N은 50보다 작거나 같은 자연수이고 파일 이름의 길이는 모두 같고 길이는 최대 50이다. 파일이름은 알파벳과 "." 그리고 "?"로만 이루어져 있다.

출력
첫째 줄에 패턴을 출력하면 된다.

예제 입력 1
3
config.sys
config.inf
configures

예제 출력 1
config????

풀이 과정

파일 이름들을 받아 리스트로 만든다.
각각의 이름들의 첫 번째 글자부터 마지막 글자까지 각각 집합으로 만든다.
집합의 크기가 1이 아니면 이름의 해당 위치의 글자는 단 하나라도 서로 같지 않은 문자가 있으므로 그 부분은 물음표를 정답 변수에 붙인다.
집합의 크기가 1이라면 해당 문자를 정답 변수에 붙인다.

두 번째 풀이는 첫 번째 풀이를 조금 다듬었다. 집합으로 만드는 부분을 한 줄로 정리했다.

코드 1

N = int(input())
names = [input() for _ in range(N)]
length = len(names[0])

result = ''
for i in range(length):
    tmp = set()
    for name in names:
        tmp.add(name[i])
    if len(tmp) == 1:
        result += names[0][i]
    else:
        result += '?'
print(result)

코드 2

N = int(input())
names = [input() for _ in range(N)]
length = len(names[0])

result = ''
for i in range(length):
    tmp = set([name[i] for name in names])
    if len(tmp) == 1:
        result += tmp.pop()
    else:
        result += '?'
print(result)

백준 1032 명령 프롬프트

좋은 웹페이지 즐겨찾기