[algorithm] 홀짝에 따라 다른 값 반환하기

인철·2024년 2월 26일
0

algorithm

목록 보기
76/91
post-thumbnail

양의 정수 n이 매개변수로 주어질 때, n이 홀수라면 n 이하의 홀수인 모든 양의 정수의 합을 return 하고 n이 짝수라면 n 이하의 짝수인 모든 양의 정수의 제곱의 합을 return 하는 solution 함수를 작성해 주세요.

class Solution {
    public int solution(int n) {
        int answer = 0; // 최종결과 값을 0으로 초기화
        int odd = 0; // 홀수 값을 0으로 초기화
        int jjj = 0; // 짝수 값을 0으로 초기화
        for(int i = 1; i <= n; i++){ // 1부터 시작하기
            if(i % 2 == 0){ // i가 짝수 라면
                jjj += i * i; // 짝수 값을 i 제곱을 더하기
            }
            else { // 홀수라면
                odd += i; // i값을 더하기
            }
        }
        answer = n % 2 == 0 ? jjj : odd;
        // n이 짝수 라면 jjj 값을, 홀수 라면 odd 값을 저장하기
        
        return answer;
        // answer 반환
    }
}
profile
같은글이있어도양해부탁드려요(킁킁)

0개의 댓글