結果
問題 | No.533 Mysterious Stairs |
ユーザー |
![]() |
提出日時 | 2018-08-20 21:44:48 |
言語 | C++11 (gcc 13.3.0) |
結果 |
AC
|
実行時間 | 25 ms / 5,000 ms |
コード長 | 2,434 bytes |
コンパイル時間 | 1,156 ms |
コンパイル使用メモリ | 79,336 KB |
実行使用メモリ | 26,880 KB |
最終ジャッジ日時 | 2024-11-27 11:52:40 |
合計ジャッジ時間 | 2,370 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 28 |
ソースコード
#include<cstdio>#include<queue>#include<utility>#include<cstring>#include<stack>#include<algorithm>#include<cmath>#include<iostream>#include<map>#define MAX_N 100001#define INF_INT 2147483647#define INF_LL 9223372036854775807#define REP(i,n) for(int i=0;i<(int)(n);i++)using namespace std;typedef long long int ll;typedef pair<ll,ll> P;void init(int n);int find(int n);void unite(int x,int y);bool same(int x, int y);ll bpow(ll,ll,ll);typedef vector<int> vec;typedef vector<vec> mat;mat mul(mat &A,mat &B);mat pow(mat A,ll n);int dx[4] = {1,0,0,-1};int dy[4] = {0,1,-1,0};bool cmp_P(const P &a,const P &b){return a.second < b.second;}int main(){ll N,dp[1000001][3];cin >> N;ll MOD = 1000000007;REP(i,1000001)REP(j,3)dp[i][j] = 0;dp[0][0] = 0;dp[0][1] = 0;dp[0][2] = 0;dp[1][0] = 1;dp[1][1] = 0;dp[1][2] = 0;dp[2][0] = 0;dp[2][1] = 1;dp[2][2] = 0;dp[3][0] = 1;dp[3][1] = 1;dp[3][2] = 1;for(int i=3;i<N;i++){REP(j,3){if(j == 0){dp[i+1][0] = (dp[i][1]+dp[i][2]) % MOD;}else if(j == 1){dp[i+1][1] = (dp[i-1][0]+dp[i-1][2])%MOD;}else{if(i-2 >= 0)dp[i+1][2] = (dp[i-2][0]+dp[i-2][1])%MOD;elsedp[i+1][2] = 0;}}}cout <<( dp[N][0] + dp[N][1] + dp[N][2])%MOD << endl;return 0;}int par[MAX_N];int ranks[MAX_N];//n要素で初期化void init(int n){REP(i,n){par[i] = i;ranks[i] = 0;}}//木の根を求めるint find(int x){if(par[x] == x){return x;}else{return par[x] = find(par[x]);}}void unite(int x,int y){x = find(x);y = find(y);if(x == y) return ;if(ranks[x] < ranks[y]){par[x] = y;}else{par[y] = x;if(ranks[x] == ranks[y]) ranks[x]++;}}bool same(int x, int y){return find(x) == find(y);}ll bpow(ll a, ll n,ll mod){int i = 0;ll res=1;while(n){if(n & 1)res = (res*a) % mod;a = (a*a) % mod;n >>= 1;}return res;}const int MOD = 1000000007;mat mul(mat &A, mat &B){mat C(A.size(),vec(B[0].size()));REP(i,A.size())REP(k,B.size())REP(j,B[0].size()){C[i][j] = (C[i][j] + A[i][k] * B[k][j]) % MOD;}return C;}mat pow(mat A,ll n){mat B(A.size(),vec(A.size()));REP(i,A.size()){B[i][i] = 1;}while(n > 0){if ( n & 1) B = mul(B,A);A = mul(A,A);n >>= 1;}return B;}