[프로그래머스] Python - Lv.0 등차수열의 특정한 항만 더하기

https://school.programmers.co.kr/learn/courses/30/lessons/181931

프로그래머스

문제

코드 1 (내꺼.. 못생겼다..)

def solution(a, d, included):
    answer = 0
    cal = 0
    perm = []
    
    for n in range(len(included)) :
        cal = a + n*d
        perm.append(cal)
    
    for i in range(len(included)) :
        if included[i] == 1 :
            answer += perm[i]
        
    return answer

코드 2 (남들꺼.. 예쁘다..)

def solution(a, d, included):
    answer = 0
    for i in range(len(included)):
        if included[i]==True:
            answer+=a+i*d
    return answer