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

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

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

프로그래머스

 

1. 문제 명

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

2. 문제 설명

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

3. 제한 사항

- 1 ≤ a ≤ 100

- 1 ≤ d ≤ 100

- 1 ≤ included의 길이 ≤ 100

- included에는 true가 적어도 하나 존재합니다.

4. 예시

a	d	included	result
3	4	[true, false, false, true, true]	37
7	1	[false, false, false, true, false, false, false]	10

5. 기본 제공 코드

function solution(a, d, included) {
    var answer = 0;
    return answer;
}

6. 제출한 내 답

const solution = (a, d, included) => {
  let answer = 0;
  included.forEach((v) => {
    v ? (answer += a) : "";
    a += d;
  });
  return answer;
};

 

6-2. VSC에 작성한 내용

const solution = (a, d, included) => {
  // let answer = 0;
  // for (i = 0; i < included.length; i++) {
  //   if (included[i]) {
  //     answer += a;
  //   }
  //   a += d;
  // }
  // return answer;
  //
  let answer = 0;
  included.forEach((v) => {
    v ? (answer += a) : "";
    a += d;
  });
  return answer;
};

// 테스트
console.log(solution(3, 4, [true, false, false, true, true]));
console.log(solution(7, 1, [false, false, false, true, false, false, false]));

7. 특이사항

없음

8. 다른 사람이 작성한 답

8-1. reduce

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