結果

問題 No.1112 冥界の音楽
ユーザー QCFium
提出日時 2020-07-10 21:38:40
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 4 ms / 2,000 ms
コード長 2,466 bytes
コンパイル時間 1,936 ms
コンパイル使用メモリ 177,496 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-11 08:24:02
合計ジャッジ時間 3,066 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

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() {
	int k = ri();
	int m = ri();
	int64_t n;
	scanf("%" SCNd64, &n);
	
	ModMatrixPower<MOD> mat(k * k);
	for (int i = 0; i < m; i++) {
		int p = ri() - 1;
		int q = ri() - 1;
		int r = ri() - 1;
		mat[p * k + q][q * k + r] = 1;
	}
	std::vector<int> all(k * k);
	for (int i = 0; i < k; i++) all[i] = 1;
	n -= 2;
	for (; n; n >>= 1) {
		if (n & 1) mat.apply(all);
		mat.pow_self();
	}
	int64_t res = 0;
	for (int i = 0; i < k; i++) res += all[i * k];
	printf("%d\n", (int) (res % MOD));
	return 0;
}
0