Codeforces Round #652 (Div. 2) D. TediousLee （1900）

D. TediousLee

题目大意：
在树第N次成长后，最多能选出多少颗爪型树，求出这个值*4（点的数量）

思路：
经过几次推导，能够发现，会新长出来的点的数量a[i]=a[i-1]+2*a[i-2]，因为爪型树的根节点来自a[i-2]，左右两边的点来自a[i]，下面的点来自a[i-1]，能够看出，每当选出一层的根节点为a[i]爪型树，就不能再选根节点层数为a[i-1]和a[i-2]的爪型树，所以能推出递推式ans[i]=ans[i-1]+2*ans[i-2]+(i%3==0);

该递推式的意思值的是从最下面那一层开始每3层3层往上取一层树

代码附：

#pragma GCC optimize("Ofast","inline","-ffast-math")
#pragma GCC target("avx,sse2,sse3,sse4,mmx")
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N = 2e6+10;
const int mod=1e9+7;
int a[N]= {0,0,0,1,1};
void init()
{
    for(int i=5; i<N; ++i)
    {
        a[i]=a[i-1]+2*a[i-2]+(i%3==0);
        a[i]%=mod;
    }
}
inline int mul(int a,int b)
{
    return (a*b-(int)((long double)a/mod*b)*mod+mod)%mod;
}
signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int t,n;
    init();
    cin>>t;
    while(t--)
    {
        cin>>n;
        cout<<mul(4,a[n])<<endl;
    }
    return 0;
}