Codeforces Round 113 (Div. 2)E. Tetrahedron（dp、递推）

题面

链接

E. Tetrahedron

题意

从一个顶点出发走过路径长度为n回到出发点的方案总数

题解

考虑dp
$f[i][0|1|2|3]$:走了i步，现在在j点的方案总数
转移：
$f[i][0]=f[i-1][1]+f[i-1][2]+f[i-1][3]$
$f[i][1]=f[i-1][0]+f[i-1][2]+f[i-1][3]$
$f[i][2]=f[i-1][0]+f[i-1][1]+f[i-1][3]$
$f[i][3]=f[i-1][0]+f[i-1][1]+f[i-1][2]$

代码

#include <bits/stdc++.h> 
#define int long long
#define rep(i,a,b) for(int i = (a); i <= (b); ++i)
#define fep(i,a,b) for(int i = (a); i >= (b); --i)
#define pii pair<int, int>
#define pll pair<long long, long long>
#define ll long long
#define db double
#define endl '\n'
#define x first
#define y second
#define pb push_back

using namespace std;
const int mod=1e9+7;
int dp[3][5];
void solve()
{
	int n;
	cin>>n;
	dp[0][0]=1;
	rep(i,1,n){
		int u=i&1,v=(i-1)&1;
		dp[u][0]=(dp[v][1]+dp[v][2]+dp[v][3])%mod;
		dp[u][1]=(dp[v][0]+dp[v][2]+dp[v][3])%mod;
		dp[u][2]=(dp[v][0]+dp[v][1]+dp[v][3])%mod;
		dp[u][3]=(dp[v][0]+dp[v][1]+dp[v][2])%mod;
	}
	cout<<dp[n&1][0]<<endl;
}

signed main(){
	ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
//   	freopen("1.in", "r", stdin);
  	int _;
//	cin>>_;
//	while(_--)
	solve();
	return 0;
}

总结

一定要开long long，容易忘直接$definelonglongint$
爆空间开滚动数组优化。