[ BOJ ] 1032 : 명령 프롬프트 ( BRONZE 1 ) / C, Python

문제

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

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

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

입력

첫째 줄에 파일 이름의 개수 N이 주어진다. 둘째 줄부터 N개의 줄에는 파일 이름이 주어진다. N은 50보다 작거나 같은 자연수이고 파일 이름의 길이는 모두 같고 길이는 최대 50이다. 파일이름은 알파벳 소문자와 '.' 로만 이루어져 있다.

출력

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

풀이 과정

특정 위치에서 모든 문자열이 똑같은 문자를 가지고 있다면, 그 문자는 '?' 패턴이 들어가는 곳이 아니다.

하나라도 다른 문자면, 다른 파일 이름이 검색될 가능성이 있기에 '?' 로 표시해야 한다.

문자열 모두를 검사하며 패턴을 찾는다.

C

1. 2차원 문자열 배열을 이용하여 문자열 모두 입력받고 브루트포스로 검토하기

#include <stdio.h>
#include <string.h>

char str[50][51];
int main(void) {
	int n; scanf("%d", &n);
	for (int i = 0; i < n; i++) scanf("%s", str[i]);
	
	char result[51];
	for (int i = 0; i < strlen(str[0]); i++) {
		// i번째 문자 패턴 찾기
		for (int j = 1; j < n; j++) {
			if (str[j][i] != str[0][i]) {
				// 0번째 문자열의 i번째 문자와 j번째 문자열의 i번째 문자가 다르면 ?
				result[i] = '?';
				break;
			}
		}

		if (result[i] != '?') result[i] = str[0][i];
	}
	result[strlen(str[0])] = 0; // 문자열은 뒤에 널 문자를 넣는 것 유의!

	printf("%s", result);
	return 0;
}

2. 먼저 첫 번째 문자열을 받고, 이후 들어오는 문자열이 첫 번째 문자열과 다른지 검토

#include <stdio.h>
#include <string.h>

char str[50][51];
int main(void) {
	int n; scanf("%d", &n);
    char result[51];
	scanf("%s", result); // 첫 번째 문장 입력받기
	
    char tmp[51];
	for (int i = 0; i < n - 1; i++) {
        scanf("%s", tmp); // 1 + i 번째 문장 입력받기
		
        for (int j = 0; j < strlen(result); j++) {
            if (result[j] != '?' && tmp[j] != result[j]) result[j] = '?';
        }
	}

	printf("%s", result);
	return 0;
}

Python

import sys
input = sys.stdin.readline

n = int(input().rstrip())
arr = []
for _ in range(n): arr.append(input().rstrip())

for i in range(len(arr[0])):
    now = {}
    for k in range(n):
        if not now.get(arr[k][i]): now[arr[k][i]] = 1
    
    if len(now) == 1:
        print(list(now.keys())[0], end='')
    else:
        print('?', end='')