c++求數列6

鏈接：https://ac.nowcoder.com/acm/problem/22161
來源：牛客網

題目描述
有一個數列

已知

A1=0A_1=0A1​=0

A2=1A_2=1A2​=1

A3=1A_3=1A3​=1

A4=A1+2∗A2＋A3A_4=A_1+2*A_2＋A_3A4​=A1​+2∗A2​＋A3​

A5=A2+2∗A3+A4A_5=A_2+2*A_3+A_4A5​=A2​+2∗A3​+A4​

…

求AnA_nAn​
輸入描述:

輸入一個整數n (1≤n≤201 \le n \le 201≤n≤20)

輸出描述:

輸出一個整數

#include<iostream>
#include<iomanip>
#include<cmath>
using namespace std;
int main()
{
	int n;
	cin>>n;
	int a[n];
	a[0]=0;
	a[1]=a[2]=1;
	for(int i=3;i<n;i++)
	{
		a[i]=a[i-3]+2*a[i-2]+a[i-1];
	}
	cout<<a[n-1];
	return 0;
 }