[JAVA] 프로그래머스 - 등차수열의 특정한 항만 더하기

프로그래머스 코딩테스트 연습 Lv.0 '등차수열의 특정한 항만 더하기' 문제 풀이와 풀이과정 입니다.

 

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

프로그래머스

---

❓ 문제 설명

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

 

 

< 제한사항 >

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

---

⚠️ 입출력 예시

입력 ▼

// a
3

// d
4

// included
[true, false, false, true, true]

 

출력 ▼

37

---

🔍 마주한 문제와 해결 과정

솔직히 처음에는 쉽게 풀 수 있을 줄 알았던 문제였다. 하지만 등차수열에 대한 이해가 조금 부족했던 탓인지 생각보다 헷갈렸다.

 

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

 

이렇게 문제를 푸니 점점 included의 i가 커질 수록 오류가 났다. 바로 전의 answer의 값과 d(공차) 곱하기 i(included의 인덱스)를 한 값을 더하는 과정에서 오류가 났던 것 같다.

 

answer = a;
answer += d*i

위에 코드처럼 바로 전의 answer 값에 계속 더하는 식으로 코드를 작성했더니 등차수열에 어긋나게 되었다.

---

✏️ 정답 코드

등차수열이란, 첫째항부터 같은 수씩 더해지는 수열을 뜻한다.

 

answer = 0;
answer += a+(d*i)

그에 따라 차근차근 다시 코드를 생각하여 이런 식으로 코드를 다시 작성했다.

 

최종적인 정답 코드이다.

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] == true){
                answer += a+d*i;
            }
        }
        
        return answer;
    }
}