Code Kata(Python) week 1 - day 5

Jeongyun Heo·2021년 1월 30일
0
post-thumbnail

문제

strs은 단어가 담긴 배열입니다.

공통된 시작 단어(prefix)를 반환해주세요.

예를 들어

strs = ['start', 'stair', 'step']
return은 'st'
strs = ['start', 'wework', 'today']
return은 ''

문제풀이 1

def get_prefix(strs):
    if len(strs) == 0:
        return ''

    strs.sort()

    i = 0
    while True:
        try:
            if strs[0][i] != strs[-1][i]:
                break
        except:
            break
        i += 1

    return strs[0][:i]

문제풀이 2

def get_prefix(strs):
  temp_zero = [x[0] for x in strs]
  if len(strs) == 0 or len(set(temp_zero)) > 1:
    return ''

  i = 1
  while True:
    try:
      word_lst = [x[i] for x in strs]
    except:
      break
    if len(set(word_lst)) > 1:
      break
    i += 1

  return strs[0][:i]

0개의 댓글