結果

問題 No.1105 Many Triplets
ユーザー shibh308shibh308
提出日時 2020-06-02 20:56:53
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 3 ms / 2,000 ms
コード長 1,192 bytes
コンパイル時間 3,247 ms
コンパイル使用メモリ 203,760 KB
最終ジャッジ日時 2025-01-10 20:38:13
ジャッジサーバーID
（参考情報） 		judge3 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:21:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   21 |     scanf("%lld%lld%lld%lld", &n, &a, &b, &c);
      |     ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

const int64_t mod = 1000000007;

vector<vector<int>> mul(vector<vector<int>>& x, vector<vector<int>>& y){
    vector<vector<int>> z(3, vector<int>(3, 0));
    for(int i = 0; i < 3; ++i)
        for(int j = 0; j < 3; ++j)
            for(int k = 0; k < 3; ++k)
                z[i][j] = (z[i][j] + 1LL * x[i][k] * y[k][j] + mod) % mod;
    return z;
}


signed main(){
    long long n;
    long long a, b, c;

    scanf("%lld%lld%lld%lld", &n, &a, &b, &c);

    --n;

    vector<vector<vector<int>>> ar(63, vector<vector<int>>(3, vector<int>(3)));
    ar[0] = {
            {1, 0, -1},
            {-1, 1, 0},
            {0, -1, 1},
    };

    for(int k = 0; k < 62; ++k)
        ar[k + 1] = mul(ar[k], ar[k]);

    vector<vector<int>> v(3, vector<int>(3, 0));
    v[0][0] = v[1][1] = v[2][2] = 1;

    for(int k = 0; k < 63; ++k)
        if(n & (1uLL << k))
            v = mul(v, ar[k]);

    long long na = (a * v[0][0] + b * v[1][0] + c * v[2][0]) % mod;
    long long nb = (a * v[0][1] + b * v[1][1] + c * v[2][1]) % mod;
    long long nc = (a * v[0][2] + b * v[1][2] + c * v[2][2]) % mod;

    printf("%lld %lld %lld\n", na, nb, nc);

}
0