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

프로그래머스 문제: https://school.programmers.co.kr/learn/courses/30/lessons/181931

프로그래머스

 

문제 풀이

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

 

1. 공차수열: 1, 3, 5, 7 ... 2씩 증가하는 것 처럼 일정하게 증가는 수열을 의미

2. 첫번째항과 공차를 받아서 list에 included의 개수만큼 수열을 만들기

3. included가 true인경우만 answer에서 더하기

 

* 공차수열 보고 당황했습니다! 오랫만에 수학을 하는 것 같아서요 ㅎㅎ