오늘 한 일

• 알고리즘 코드카타
  • 등차수열의 특정한 항만 더하기

알고리즘 코드카타

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

문제

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

풀이

function solution(a, d, included) {
  let result = 0;
  for (let i = 0; i < included.length; i++) {
    if (included[i]) result += a + d * i;
  }
  return result;
}

반복문을 통해 요소가 true일 경우, result 변수에 해당 등차수열을 더해주고 result를 return하는 방식으로 문제를 해결했다.

다른 사람의 풀이

function solution(a, d, included) {
    return included.reduce((acc, flag, i) => {
        return flag ? acc + a + d * i : acc
    }, 0)
}