[ 프로그래머스 ] 파이썬 - 배열 만들기

문제

 

풀이

def solution(n, k):
    answer = []		// 배열 준비
    for i in range(1, n+1):	// 1부터 n이하까지의 숫자를 i 지정
        if i % k == 0: // i를 k의 배수일때
            answer.append(i)	// 그 숫자들만(i) answer 배열에 넣어준다.
    return answer

처음에는 위와같은 방식으로 풀었지만 다른 사람들의 풀이를 보니 이보다 훨씬 더 간단한 방법으로 풀 수 있다는 것을 알게되었다!

리스트 컴프리헨션 사용하기
- 간결하고 쉬운 방식으로 리스트를 만드는 구문이다.

def solution(n, k):
    return [i for i in range(1, n+1) if i % k == 0]

 

마무리

이번 문제를 통해서 리스트 컴프리헨션에 대해서 알아보았다.
이 방법을 통해서 저렇게도 간단하게 코드를 짤 수 있다는것에 또한 감탄을 하고 간다..

파이썬으로 코딩테스트를 풀어본 적이 별로 없었지만, 이제부터 파이썬으로도 풀어보려고 한다. 파이썬은 다른 언어에 비해 코드의 가독성을 높여주고 간략하게 코드를 작성할 수 있다는 것이 파이썬만의 장점인 것 같다. 다음번에도 내가 미처 알지 못했던 정보들을 알아가는 시간이 되었으면 좋겠다!