프로그래머스 LV0 등차수열의 특정한 항만 더하기

프로그래머스

 

문제

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

프로그래머스

 

풀이

두 정수 a, d와 길이 n인 boolean 배열 included가 주어질 때, 등차수열과 included을 비교하여 included의 요소가 참일 때의 등차수열의 모든 값들의 합을 구하는 문제이다. 

 

등차수열을 구해준다음 included 배열과 zip함수로 같이 비교해가면서 문제를 해결하면 쉽게 풀 수 있다. 

 

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