dp统计字符串个数

#include<iostream>
#include<string.h>
using namespace std;
//dp
int main()
{
    int a[21],b[21];//a是以0结尾的不含101的个数，b是以1结尾的不含101的个数
    a[0]=0;b[0]=0;
    a[1]=1;b[1]=1;  //base
    for(int i=2; i<21; i++)
    {
        a[i]=a[i-1]+b[i-1];//所有不含101的字符串后面加0
        b[i]=(a[i-1]-b[i-2])+b[i-1];//所有不以10结尾的不含101的字符串后面加1
    }
    int n;
    cin>>n;
    while (n>=0)
    {
        cout<<a[n]+b[n]<<endl;
        cin>>n;
    }

    return 0;
}