IntStream.range / rangeClosed ?

프로그래머스에서 문제를 풀다가 다른 사람들의 풀이를 보면서 새롭게 알게 되어 알게 된 내용을 기록 해보려고 합니다.

 

우선 관련된 프로그래머스 문제는 짝수의 합 입니다. 

이 문제는 반복문을 이용하면 간단하게 풀리는 문제였지만, IntStream.rangeClosed 을 이용하면 더 간단하게 작성 할수 있다는 것을 알게 되었습니다.

 

IntStream.range vs rangeClosed ?

IntStream.range 와 rangeClosed 은 둘다 주어진 범위 내에서 순차적으로 정수 스트림을 반환 합니다.

차이점은 range 는 마지막 정수를 포함하지 않고, rangeClosed 는 마지막 정수를 포함합니다.

 

import java.util.stream.IntStream;

IntStream.range(0, 5).forEach(System.out::println);
IntStream.rangeClosed(0, 5).forEach(System.out::println);

 

 

 

 

그럼 처음에 언급했던 프로그래머스 문제 짝수의 합으로 코드가 어떤 식으로 변경이 될 수 있는 지 확인해 볼게요

 

우선 IntStream.rangeClosed 를 사용 안하고 반복문을 사용한 코드입니다.

class Solution {
    public int solution(int n) {
        int answer = 0;
        for (int i = 1; i <= n ; i++){
            if(i % 2 == 0) {
                answer += i;
            }
        }
        return answer;
    }
}

 

 

그 다음 IntStream.rangeClosed 사용한 코드입니다! 코드가 훨씬 간략해 진 것을 볼 수 있습니다.

import java.util.stream.IntStream;
class Solution {
    public int solution(int n) {
        return IntStream.rangeClosed(0, n)
            .filter(e -> e % 2 == 0)
            .sum();
    }
}