結果

問題 No.1105 Many Triplets
ユーザー QCFiumQCFium
提出日時 2020-07-03 21:58:07
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 3 ms / 2,000 ms
コード長 2,315 bytes
コンパイル時間 2,106 ms
コンパイル使用メモリ 178,268 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-17 00:44:40
合計ジャッジ時間 2,962 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 2 ms
5,376 KB
testcase_13 AC 2 ms
5,376 KB
testcase_14 AC 2 ms
5,376 KB
testcase_15 AC 2 ms
5,376 KB
testcase_16 AC 2 ms
5,376 KB
testcase_17 AC 2 ms
5,376 KB
testcase_18 AC 2 ms
5,376 KB
testcase_19 AC 2 ms
5,376 KB
testcase_20 AC 2 ms
5,376 KB
testcase_21 AC 2 ms
5,376 KB
testcase_22 AC 2 ms
5,376 KB
testcase_23 AC 2 ms
5,376 KB
testcase_24 AC 2 ms
5,376 KB
testcase_25 AC 3 ms
5,376 KB
testcase_26 AC 2 ms
5,376 KB
testcase_27 AC 2 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

int ri() {
	int n;
	scanf("%d", &n);
	return n;
}
template<int mod>
struct ModMatrixPower {
	int n;
	std::vector<std::vector<int> > data;
	std::vector<std::vector<int> > tmp;
	ModMatrixPower(int n) {
		this->n = n;
		data.resize(n, std::vector<int>(n));
		tmp.resize(n, std::vector<int>(n));
	}
	auto & operator [] (int x) { return data[x]; }
	void pow_self() {
		for (int i = 0; i < n; i++)
			for (int j = 0; j < n; j++) {
				uint64_t sum = 0;
				int upper1 = n & 0x7FFFFFF0;
				for (int k = 0; k < upper1; k += 0x10) {
					sum += (uint64_t) data[i][k] * data[k][j];
					sum += (uint64_t) data[i][k + 1] * data[k + 1][j];
					sum += (uint64_t) data[i][k + 2] * data[k + 2][j];
					sum += (uint64_t) data[i][k + 3] * data[k + 3][j];
					sum += (uint64_t) data[i][k + 4] * data[k + 4][j];
					sum += (uint64_t) data[i][k + 5] * data[k + 5][j];
					sum += (uint64_t) data[i][k + 6] * data[k + 6][j];
					sum += (uint64_t) data[i][k + 7] * data[k + 7][j];
					sum += (uint64_t) data[i][k + 8] * data[k + 8][j];
					sum += (uint64_t) data[i][k + 9] * data[k + 9][j];
					sum += (uint64_t) data[i][k + 10] * data[k + 10][j];
					sum += (uint64_t) data[i][k + 11] * data[k + 11][j];
					sum += (uint64_t) data[i][k + 12] * data[k + 12][j];
					sum += (uint64_t) data[i][k + 13] * data[k + 13][j];
					sum += (uint64_t) data[i][k + 14] * data[k + 14][j];
					sum += (uint64_t) data[i][k + 15] * data[k + 15][j];
					sum %= mod;
				}
				for (int k = upper1; k < n; k++) {
					sum += (uint64_t) data[i][k] * data[k][j];
				}
				tmp[i][j] = sum % mod;
			}
		std::swap(tmp, data);
	}
	void apply(std::vector<int> &vec) {
		assert(vec.size() == (size_t) n);
		std::vector<int> vec_tmp(n);
		for (int i = 0; i < n; i++)
			for (int j = 0; j < n; j++) {
				vec_tmp[j] += (int64_t) vec[i] * data[i][j] % mod;
				if (vec_tmp[j] >= mod) vec_tmp[j] -= mod;
			}
		std::swap(vec, vec_tmp);
	}
};
#define MOD 1000000007

int main() {
	int64_t n;
	scanf("%" SCNd64, &n);
	std::vector<int> vec(3);
	for (auto &i : vec) i = ri();
	n--;
	
	ModMatrixPower<MOD> mat(3);
	mat[0] = {1, 0, MOD - 1};
	mat[1] = {MOD - 1, 1, 0};
	mat[2] = {0, MOD - 1, 1};
	for (; n; n >>= 1) {
		if (n & 1) mat.apply(vec);
		mat.pow_self();
		
	}
	printf("%d %d %d\n", vec[0], vec[1], vec[2]);
	return 0;
}
0