[프로그래머스 코딩테스트] 등차수열의 특정한 항만 더하기(Java)

  문제 설명

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

 

  제한사항

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

 

  입출력 예

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

class Solution {
    public int solution(int a, int d, boolean[] included) {
        int answer = 0;
        int[] num = new int[included.length];
        num[0] = a;
        for (int i = 1; i < included.length; i++) {
            num[i] = num[i - 1] + d;
        }
        for (int i = 0; i < included.length; i++) {
            if (included[i]) {
                answer += num[i];
            }
        }
        return answer;
    }
}

  num 배열을 생성합니다. included에 맞는 배열을 저장할 것입니다. num의 0번째 인덱스는 a 고정입니다. for문이 1번부터 included의 길이만큼 순회합니다. i번째 인덱스는 i - 1에 d를 더한 값입니다.

  for문이 included의 길이만큼 순회합니다. 만약 included의 i번째 인덱스가 true라면 answer에 num의 i번째 인덱스를 더하기 할당합니다. 조건에 해당하는 값만 더해지므로 문제에서 요구하는 값을 구할 수 있습니다.