题解 | 牛牛学数列6

#include <iostream>
using namespace std;

int output(int n)
{
    int a1=0,a2=1,a3=1,a4=a1+2*a2+a3;
    if(n == 0)
    {
        return 0;
    }
    else if(n>1 && n<4){
        return 1;
    }
    else if (n == 4) {
        return a4;
    }
    else if(n>=5 && n<= 20)
    {
        for(int i =5;i<= n ;i ++)
        {
            a1=a2;
            a2=a3;
            a3=a4;
            a4=a1 + 2*a2 + a3;
        }
    }
    return a4;
}

int main() {
    int n;
    cin >> n;
    cout << output(n) << endl;
}
// 64 位输出请用 printf("%lld")