본문 바로가기

인공지능/알고리즘

[알고리즘] n의 배수 고르기 리스트 내 반복문, 조건문 사용 프로그래머스 level 0

앞선 포스팅에서 리스트 내 반복문을 사용할 수 있다는 내용을 적어두었다. 

그래서 코딩 할때 가급적 리스트의 결과를 리턴해야 할 경우에는 한번에 리스트내 반복문, 조건문을 사용하는

구조가 안되더라고 바꿔보기로 했다. 

 

아래는 처음에 그냥 또 직관적으로 코딩 했던 것

def solution(n, numlist):
    result = []
    for i, num in enumerate(numlist):
        if num % n == 0:
            result.append(num)
            print(result)
    return result

 

아래의 코드는 리스트 내 반복문, 조건문을 사용해서 문제를 해결했다. 

def solution(n, numlist):
    return [num for num in numlist if num % n == 0]

 

조금씩 나아지고 있군..