[Swift] for loop를 이용한 피보나치

func fibonacci(n: Int) {
    var res = [0, 1]
    
    if n < 2 {
        print("Enter a number greater than 1.")
    } else if n == 2 {
        print(res)
    } else {
        for _ in 1..<n-1 {
            res.append(res[res.count-2] + res[res.count-1])
        }
        print(res)
    }
}

fibonacci(n: 10)
// [0, 1, 1, 2, 3, 5, 8, 13, 21, 34]

 

class Assignment {
    
    func fibonacci(n: Int) {
        
        var n1 = 0
        var n2 = 1
        
        if n == 0 {
            print("Invalid")
        } else if n == 1 {
            print(n1)
        } else if n == 2 {
            print(n1, n2)
        } else {
            var array = [n1, n2]
            for _ in 2..<n {
                let n3 = n1 + n2
                n1 = n2
                n2 = n3
                array.append(n3)
            }
            print(array)
        }
    }
}