조건문 : 특정한 항만 더하기

문제

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

입출력 예

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

풀이

import java.util.ArrayList;

class Solution {
    public int solution(int a, int d, boolean[] included) {
int answer = 0;
        ArrayList<Integer> arr = new ArrayList<>();

        // true 가 몇번째 항인지 찾기
        for (int i = 0; i < included.length; i++){
            if (included[i] == true) {
                arr.add(i+1);
            }
        }

        System.out.println("[arr] : " + arr);
        // 해당 항의 숫자를 계산
        for (int i = 0; i < arr.size(); i++){
            // 각각의 항을 계산
            arr.get(i); // 4항 ?  3+4+4+4 i-1 만큼 d를 더해줌
            answer += a;
            for (int j = 1; j < arr.get(i); j++){
                answer += d;
            }
        }
        return answer;
    }
}

입력값 〉 3, 4, [true, false, false, true, true]
기댓값 〉 37
실행 결과 〉 테스트를 통과하였습니다.
출력 〉 [arr] : [1, 4, 5]