F i r s t S o l u t i o n First Solution FirstSolution
这种不断分支散叶的结构。不如逆向思维,把第n项看成一个根节点+2个第n-1项+1个第n-2项连接而成。
记dp[i]表示第i项的答案,那么dp[1]=dp[2]=0,dp[3]=1;那么第n项的答案是否就是2个第n-1项+1个第n-2项的答案相加呢?显然不是,因为你还有个根节点没有考虑在内,如果要用上这个根节点,你还得用上前两项的根节点,前提条件是他们的根节点尚未染色。
设ans[i]为第i项的爪型数量。
显然ans[1]和ans[2]的根节点未被染色,ans[3]就可以用前两项的根节点一起染色做出1的贡献。显然ans[4]和ans[5]没法利用已被染色的ans[3].
而ans[6]可以利用前两项,所以答案就出来了。
ans[i]=2*ans[i-2]+ans[i-1]+(i%3==0)
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define MAXN 2000005
#define rep(n) for(int i=1;i<=n;i++)
#define rall(x) for(int i=(x).size()-1;i>=0;i--)
#define all(x) for(int i=0;i<(x).size();i++)
int a[MAXN];
int b[MAXN];
ll ans[MAXN];
const long long mod=1000000007;
int main()
{
int t;
cin>>t;
for(int i=3;i<=2000000;i++)
ans[i]=(2ll*ans[i-2]+ans[i-1]+(i%3==0))%mod;
while(t--)
{
int n;
cin>>n;
cout<<4ll*ans[n]%mod<<endl;
}
}
S e c o n d S o l u t i o n Second Solution SecondSolution
另一种做法就是对题目中的结点分出三个状态。
第一个状态即单个节点。
第二个状态即带一个子节点的单个节点。
第三个状态即带三个子节点的单个节点。
记录dp[i]为第i轮新生成的结点。
那么,
dp[1]=1
dp[i]=2*dp[i-1]+dp[i-2]
现在考虑如果最后一轮生长时新出现的爪型数,显然它的染色会与他们的母节点重合,但我们贪心选取子节点部分进行染色。
ans[n]=dp[n%3+3
∗
*
∗j]
(n%3+3*j<=n),j可取自然数。
用前缀和维护答案即可。
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define MAXN 2000005
#define rep(n) for(int i=1;i<=n;i++)
#define rall(x) for(int i=(x).size()-1;i>=0;i--)
#define all(x) for(int i=0;i<(x).size();i++)
int a[MAXN];
int b[MAXN];
ll dp[MAXN];
ll cnt[MAXN];
const long long mod=1000000007;
int main()
{
int t;
cin>>t;
dp[1]=dp[2]=1;
for(int i=3;i<=2000000;i++)
{
dp[i]=(2ll*dp[i-2]+dp[i-1])%mod;
cnt[i]=(cnt[i-3]+dp[i-2])%mod;
}
while(t--)
{
int n;
cin>>n;
cout<<4ll*cnt[n]%mod<<endl;
}
}