18. 홀짝에 따라 다른 값 반환하기

나의 코드

import Foundation

func solution(_ n:Int) -> Int {
    
    var answer = 0
    
   if n % 2 == 0 {
            for i in 1 ... n {
                if i % 2 == 0 {
                    print (i)
                    answer += i * i
                   
                }
            }
        } else {
            for i in 1 ... n {
                if i % 2 != 0 {
                     print (i)
                    answer += i 
                }
            }
        }
    
    return answer
}

이렇게 생각한 이유.

일단 if문을 통해서 풀었다.

swich case문으로도 가능한데 일단은 이렇게 풀었다.

나중에 내가 다시 이문제를 리마인드할땐 swich case로 풀어봐야겠다.