등차수열의 특정한 항만 더하기

두 정수 a, d와
길이가 n인 boolean 배열 included가 주어집니다.
첫째항이 a,
공차가 d인 등차수열에서
included[i]가 i + 1항을 의미할 때,
이 등차수열의 1항부터 n항까지
included가 true인 항들만 더한 값을
return 하는 solution 함수를 작성해 주세요.

내 코드

def solution(a, d, included):
    answer = sum([a+d*i for i,e in enumerate(included) if e])
    return answer

예시 코드

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