爬楼梯C++

class Solution {
public:
/**
* @param n: An integer
* @return: An integer
*/

int climbStairs(int n) {
// write your code here
int a = 1, b = 1, k = 0;
if(n == 1 || n ==0){
return 1;
}
while(--n > 0){
k = a+b;
b = a;
a = k ;
}
return k;
}
};