結果

問題 No.658 テトラナッチ数列 Hard
ユーザー pazzle1230
提出日時 2018-03-02 22:43:08
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 556 ms / 2,000 ms
コード長 1,684 bytes
コンパイル時間 2,225 ms
コンパイル使用メモリ 184,820 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-06-22 04:14:14
合計ジャッジ時間 4,944 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 8
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

#define INF_LL (ll)1e18
#define INF (int)1e9
#define REP(i, n) for(int i = 0;i < (n);i++)
#define FOR(i, a, b) for(int i = (a);i < (b);i++)
#define all(x) x.begin(),x.end()
#define fs first
#define sc second

using int32 = int_fast32_t;
using uint32 = uint_fast32_t;
using int64 = int_fast64_t;
using uint64 = uint_fast64_t;
using PII = pair<int32, int32>;
using PLL = pair<int64, int64>;

const double eps = 1e-10;

template<typename A, typename B>inline void chmin(A &a, B b){if(a > b) a = b;}
template<typename A, typename B>inline void chmax(A &a, B b){if(a < b) a = b;}

using mat = vector<vector<int32>>;

mat mul(mat a, mat b){
	mat res = mat(a[0].size(), vector<int32>(b[0].size(), 0));
	REP(i, a.size()){
		REP(j, b[0].size()){
			REP(k, a[0].size()){
				res[i][j] = res[i][j]+a[i][k]*b[k][j];
			}
			res[i][j] %= 17;
		}
	}
	return res;
}

mat are(int32 x){
	mat res = mat(x, vector<int32>(x, 0));
	REP(i, x)
		res[i][i] = 1;
	return res;
}

mat matpow(mat a, int64 b){
	if(b == 0) return are(a.size());
	if(b%2) return mul(a, matpow(a, b-1));
	mat x = matpow(a, b/2);
	return mul(x, x);
}

map<vector<int32>, int32> mp;
map<int32, int32> mp2;
int32 dp[1123456] = {};
int32 l, r;

int main(void){
	cin.tie(0);
	ios::sync_with_stdio(false);

	vector<int32> x = {0, 0, 0, 1};
	mp[x] = 0;
	int32 Q;
	cin >> Q;
	mat org = {
		{0, 1, 0, 0},
		{0, 0, 1, 0},
		{0, 0, 0, 1},
		{1, 1, 1, 1}
	};
	mat start = {
		{0},
		{0},
		{0},
		{1}
	};
	REP(i, Q){
		int64 n;
		cin >> n;
		if(n < 4) cout << 0 << endl;
		else if(n == 4) cout << 1 << endl;
		else{
			mat res = mul(matpow(org, n-4), start);
			cout << res[3][0] << endl;
		}
	}
}
0