[Python/Lv.0] 등차수열의 특정한 항만 더하기

---

풀이

included 개수만큼 for문 돌리기

T, F 값에 따라 더할지 말지 결정해야함.. → if문

True인 값에 항들만 더하기 → +=

 

제출

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

 

다른 풀이

def solution(a, d, included):
    answer = 0
    for i in range(len(included)):
        answer += (a + d * i) * int(included[i]) # int로 0 또는 1 반환
    return answer

 

다른 풀이 2

def solution(a, d, included):
    return sum(a + i * d for i, f in enumerate(included) if f)
    # enumerate 사용해서 i, f 두 변수 활용
    # i는 인덱스로 f는 true, false!
    # if f는 f 자체가 True 또는 False이므로 True일 때만 진행
	# True인 것만 반환해서 값들을 sum으로 합산

 

 

---

한줄 한줄 작성해보기 연습!!
다른 풀이들 보면서 여러 방안 익혀두기 아자!