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

문제 설명

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

 

제한사항

• 1 ≤ a ≤ 100
  1 ≤ d ≤ 100
  1 ≤ included의 길이 ≤ 100
  included에는 true가 적어도 하나 존재합니다.

 

---

문제 풀이

접근 전략

등차수열의 특성을 이용하여, 각 항의 값은 첫째항 a에 (공차 d * 항의 인덱스)를 더하면 되는데, 그냥 true / false에 int 붙여서 곱해도 풀어짐. 이를 통해 included가 true인 항의 값을 계산하고, 이를 모두 더하면 끝.

코드

def solution(a, d, included):
    answer = 0
    
    int_included = [int(value) for value in included]

    for i in range(len(included)):
        answer = answer + ((a + i*d) * int_included[i])
    
    return answer

결과 및 해석

이 코드는 주어진 등차수열의 각 항에 대해 included가 true인지 확인하고, true인 경우에만 해당 항의 값을 합계에 추가함. 시간 복잡도는 O(n)으로, 주어진 배열의 길이에 비례함.

 

chatGPT 형님은 리스트 내포와 sum 함수를 사용하여 코드를 간결하게 표현해버림. incl 이 true일때는 (a + i*d) 를 아니면 0을 주어서 한줄컷 해주셨다.

 

def solution(a, d, included):
    return sum((a + i*d) if incl else 0 for i, incl in enumerate(included))