-
Notifications
You must be signed in to change notification settings - Fork 5
/
Copy pathmain.cpp
35 lines (30 loc) · 876 Bytes
/
main.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
// Author : Qi Zhang
// Date : 2018-12-11
#include <bits/stdc++.h>
using namespace std;
int main()
{
string line;
while (getline(cin, line)) {
long long n = stoi(line);
vector<vector<long long>> dp(n, vector<long long>(3, 0));
for(long long i = 0; i < n; i++){
for(long long j = 0; j < 3; j++){
if(i == 0) dp[i][j] = 1;
else if(i == 1){
for(long long k = 0; k < 3; k++) dp[i][j] += dp[i-1][k];
}
else {
for(long long k = 0; k < 3; k++){
if(j != k)
dp[i][j] += dp[i-1][k] + dp[i-2][k];
}
}
}
}
long long ans = 0;
for(auto val: dp[n-1]) ans += val;
cout << ans << endl;
}
return 0;
}