[Java] 22. 등차수열의 특정한 항만 더하기

1. 문제 설명, 제한 사항, 입출력 예시

2. 아이디어

3. 풀이 코드

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

import java.util.stream.IntStream;

class Solution {
    public int solution(int a, int d, boolean[] included) {
        return IntStream.range(0, included.length)
                .filter(i -> included[i])
                .map(i -> a + d * i)
                .sum();
    }
}