[백준/BOJ] python 1032번 명령 프롬프

언어: python

번호: 1032

제목: 명령 프롬프트

등급: 브론즈 1

​

풀이 과정: 각 파일들을 입력 받아서 2차원 행렬로 만들고, 넘파이를 활용해 세로로 쪼개서 한 열이 모두 같은 문자면 그대로 사용하고 아니면 '?'를 추가 -> 런타임 에러 -> 백준에서는 외부 라이브러리인 numpy를 사용 못함...

문자열을 입력 받아 하나의 리스트에 담고 i번째 글자가 모두 같으면 그대로 사용하고 아니면 '?'를 추가 

​​

코드:

# run-time error
import numpy as np

n = int(input())

lst = []
for i in range(n):
    lst.append(list(input()))
lst = np.array(lst)

result = ''
for i in range(len(lst[0])):
    s = np.unique(lst[:, i])
    if len(s) == 1:
        result += s[0]
    else:
        result += '?'

print(result)

n = int(input())

lst = []
for i in range(n):
    lst.append(list(input()))

result = ''
for i in range(len(lst[0])):
    s = set([k[i] for k in lst])
    if len(s) == 1:
        result += str(list(s)[0])
    else:
        result += '?'

print(result)

 

메모: