stream이라는 것을 발견하고 유용하게 써먹어 보았다.
import java.util.*;
import java.util.stream.*;
class Solution {
public int solution(int n) {
int answer = 0;
if(n % 2 == 1){
answer = IntStream.rangeClosed(1,n)
.filter(num -> num%2 == 1)
.sum();
} else {
answer = IntStream.rangeClosed(1,n)
.filter(num -> num%2 == 0)
.map(i -> i * i)
.sum();
}
return answer;
}
}
https://school.programmers.co.kr/learn/courses/30/lessons/181935
'코딩테스트 > Programmers' 카테고리의 다른 글
정사각형으로 만들기 (0) | 2025.01.23 |
---|---|
[프로그래머스] 전국 대회 선발 고사 (0) | 2025.01.23 |
[프로그래머스] 배열 만들기 2 (0) | 2025.01.22 |
[프로그래머스] 주사위 게임 3 (1) | 2025.01.22 |
[프로그래머스] 정수를 나선형으로 배치하기 (0) | 2025.01.22 |