[Python] Lv1.수박수박수박수박수박수?

https://programmers.co.kr/learn/courses/30/lessons/12922

 

 

 

문제 설명

길이가 n이고, "수박수박수박수...."와 같은 패턴을 유지하는 문자열을 리턴하는 함수, solution을 완성하세요. 예를들어 n이 4이면 "수박수박"을 리턴하고 3이라면 "수박수"를 리턴하면 됩니다.

 

제한 조건

• n은 길이 10,000이하인 자연수입니다.

 

나의 접근방법 및 풀이

수박 문제에는 짧은 스토리가 있다. 

처음 문제를 보고 '홀수랑 짝수일 때 생각하면 되잖아?'이러면서 생각보다 문제가 쉽게 풀려(Lv1이지만) 좋아했는데 시간 초과로 실패했다.

 

(1) 리스트 추가

첫 번째에는 홀수일 경우, 짝수일경우를 나누어 리스트에 한 글자씩 추가하고 → ex) ["수", "박", "수"]

나중에 join을 이용해 리스트내 요소들을 합쳤다 → ex) "수박수"

def solution(n):
    ## 홀수에는 수, 짝수에는 박
    
    sstr = []
    while (n != 0):
        if n%2 != 0:
            sstr.append("수")
        else:
            sstr.append("박")
    answer = str("".join(sstr))
    return answer

 

(2) 문자열 추가

'그래, 리스트가 무거웠나보다'라는 생각으로 문자열에 추가하는 방식으로 방법을 바꾸었지만 이 경우에도 거절당했다.

def solution(n):
    ## 홀수에는 수, 짝수에는 박
    answer = ""
    while (n != 0):
        if n%2 != 0:
            answer += "수"
        else:
            answer += "박"
    return answer

 

(3) 최종 코드

'다 온 것 같은데 도대체 뭐가 문제지' 생각을 하다 코테 문제들을 배우며 for문과 if문을 한 줄로 표현하고 결과값을 리스트에 담는 방법이 생가가나 한 번 해보고 제출할때는 (1)에서 썼던 join을 써 제출했더니 성공했다. 야호!

def solution(n):
    ## 홀수에는 수, 짝수에는 박
    answer = ["박" if i%2 != 0 else "수" for i in range(n)]
    return str("".join(answer))

 

<다른 풀이>

def solution(n):
    s = "수박" * n
    return s[:n]

처음에 이 코드를 보고 '아... 맞다 슬라이싱' 했다. n만큼 수박을 반복하고 제출할때는 n까지 슬라이싱 해서 return하면 되니까. 그런데 다른 분들 댓글을 보며 메모리 공간의 효율성에 대해 생각하게 되었다. n이 커질 수록 그 만큼 불필요한 메모리와 CPU 연산이 필요하니까.

이 코드를 보며 시간 복잡도에 대해 진짜 공부해야겠다고 더 생각했다. 

 

def solution(n):
    return "수박"*(n//2) + "수"*(n%2)

개인적으로 이 코드가 맘에 들었다.

3 : "수박수"

4 :" "수박수박"

홀수는 "수", 짝수는 "박"이라는 생각했지만, 이걸 좀 더 생각했다면 짝수에서는 "수박", 홀수일 경우에는 짝수만큼+앞에다(1일때 제외하고) "수"만 붙이면 되니까!

 

 

여담으로, 전공시간에 시간 복잡도에 대해 배우면 대부분 주어진 함수에서 뭐가 제일 시간 복잡도 작니? 크니?에 대한 '문제' 위주로 공부하다 보니, 실질적으로 코드를 작성할 때 내 코드의 시간 복잡도를 어떻게 구해야하는 지 잘 몰랐고 일단 문제 풀기에 급급했다. 그런데 이 문제를 제외하고도 여러문제를 풀다보니, 생각보다 시간 초과로 실패하는 경우가 종종 있었다. 문제를 단순히 풀는 것이 아니라, 어떻게 더 효율적으로 구현하느냐에 초점을 맞춰야겠다.