[Python] Softeer : X marks the Spot

Softeer에서 X marks the Spot이라는 문제를 풀어보았다.

 

https://softeer.ai/practice/7703

Softeer - 현대자동차그룹 SW인재확보플랫폼

 

N을 입력받고 N번 왼쪽 오른쪽 단어를 입력받는다.

왼쪽 단어에서 x나 X가 있는 index를 추출해 오른쪽 단어에 해당 index에 있는 값을 뽑아 정답에 추가시킨다.

소문자는 대문자로 바꾼다(문제를 잘 못 읽어서 소문자 -> 대문자, 대문자 -> 소문자 로 풀었었습니다.....)

 

import sys

N = int(input())
answer = ''
for _ in range(N):
    s_lst, t_lst = map(str, input().split())
    for i, s in enumerate(s_lst):
        if s == 'x' or s == 'X':
            answer += t_lst[i].upper()
            break

print(answer)

처음에 제출하니 시간초과가 테스트케이스 2개에서 나서 틀렸다고 나와서 answer를 list로 바꿔주었다.

 

 

import sys
input = sys.stdin.readline
N = int(input())
answer = []
for _ in range(N):
    s_lst, t_lst = map(str, input().split())
    for i, s in enumerate(s_lst):
        if s == 'x' or s == 'X':
            answer.append(t_lst[i].upper())
            break

print(''.join(answer))

 

그리고 input 받는 것을 sys readline으로 불러와야 시간 초과가 안뜨는 것 같았다..... 20분을 헤맸다.....

 

 

 

오늘도 감사합니다.