[25.12.11 코테(Python)]-배열의 원소 삭제하기&수박수박수박수박수?

Lv0. 배열의 원소 삭제하기 (Python)

링크: 코딩테스트 연습 - 배열의 원소 삭제하기 | 프로그래머스 스쿨

 

내가 작성한 코드

def solution(arr, delete_list):
    answer = []
    for i in arr:
        if i not in delete_list:
            answer.append(i)
    return answer

 

새로 배운 코드

def solution(arr, delete_list):
    return [i for i in arr if i not in delete_list]
      
# 리스트 컴프리헨션
# i for i in arr -> arr의 요소를 하나씩 i로 가져옴
# if i not in delete_list -> delete_list에 없는 값만 남겨서 반환함

---

Lv1. 수박수박수박수박수박수? (Python)

링크: 코딩테스트 연습 - 수박수박수박수박수박수? | 프로그래머스 스쿨

 

내가 작성한 코드

def solution(n):
    pattern = "수박" * (n//2 + 1)
    answer = pattern[:n]
    return answer

 

새로 배운 코드

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

# "수박"*(n//2) -> (몫) 짝수 길이만큼 패턴 생성
# "수"*(n%2) -> (나머지) n이 홀수일 때만 마지막 "수" 한 글자 추가
# 두 결과를 더해 길이 n의 문자열을 만듦