斐波那契数列 for循环

编写计算斐波那契（Fibonacci）数列的第n项函数fib（n）(n < 40)。
数列描述：
f1=f2==1;
fn=fn-1+fn-2(n>=3)。

Input

输入整数 n 的值（0 < n < 40)。

Output

输出fib(n)的值。

#include <stdio.h>

int fib(int n) {  
    if (n == 1 || n == 2) {  
        return 1;  
    } else {  
        int a = 1, b = 1, temp;  
        for (int i = 3; i <= n; i++) {  
            temp = a + b;  
            a = b;  
            b = temp;  
        }  
        return b;  
    }  
}

int main() {  
    int n;  
    scanf("%d", &n);  
    printf("%d\n", fib(n));  
    return 0;  
}