結果
問題 | No.1105 Many Triplets |
ユーザー |
![]() |
提出日時 | 2020-07-03 21:53:23 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 2 ms / 2,000 ms |
コード長 | 2,158 bytes |
コンパイル時間 | 1,394 ms |
コンパイル使用メモリ | 121,756 KB |
実行使用メモリ | 6,944 KB |
最終ジャッジ日時 | 2024-09-17 00:25:02 |
合計ジャッジ時間 | 2,123 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge6 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 25 |
ソースコード
#include <iostream> #include <vector> #include <algorithm> #include <cmath> #include <string> #include <queue> #include <stack> #include <set> #include <map> #include <iomanip> #include <utility> #include <tuple> #include <functional> #include <bitset> #include <cassert> #include <complex> #include <stdio.h> #include <time.h> #include <numeric> #include <random> #include <unordered_map> #include <unordered_set> #define all(a) a.begin(),a.end() #define rep(i, n) for (ll i = 0; i < (n); i++) #define pb push_back #pragma GCC optimize("O3") #pragma GCC optimize("unroll-loops") using namespace std; typedef long long ll; typedef unsigned long long ull; typedef long double ld; typedef pair<ll, ll> P; typedef complex<ld> com; constexpr int inf = 1000000010; constexpr ll INF = 1000000000000000010; constexpr ld eps = 1e-12; constexpr ld pi = 3.141592653589793238; template<class T, class U> inline bool chmax(T &a, const U &b) { if (a < b) { a = b; return true; } return false; } template<class T, class U> inline bool chmin(T &a, const U &b) { if (a > b) { a = b; return true; } return false; } constexpr ll mod = 1000000007; vector<vector<ll>> mat_mul(vector<vector<ll>> a, vector<vector<ll>> b) { int n = a.size(); vector<vector<ll>> ans(n, vector<ll>(n)); for (int i = 0; i < n; i++) { for (int j = 0; j < n; j++) { for (int k = 0; k < n; k++) { ans[i][j] += a[i][k] * b[k][j]; ans[i][j] %= mod; } } } return ans; } vector<vector<ll>> mat_pow(vector<vector<ll>> a, ll b) { int n = a.size(); vector<vector<ll>> ans(n, vector<ll>(n)); for (int i = 0; i < n; i++) ans[i][i] = 1; while (b) { if (b & 1) ans = mat_mul(ans, a); a = mat_mul(a, a); b >>= 1; } return ans; } int main() { cin.tie(0); ios::sync_with_stdio(false); cout << fixed << setprecision(20); ll n; cin >> n; ll a, b, c; cin >> a >> b >> c; vector<vector<ll>> mat(3, vector<ll>(3)); rep(i, 3) mat[i][i] = 1; rep(i, 3) mat[i][(i + 1) % 3] = mod - 1; mat = mat_pow(mat, n - 1); vector<ll> ans(3); rep(i, 3) ans[i] = (a * mat[i][0] % mod + b * mat[i][1] % mod + c * mat[i][2] % mod) % mod; rep(i, 3) cout << ans[i] << ' '; cout << '\n'; }