結果

問題 No.950 行列累乗
ユーザー LeonardoneLeonardone
提出日時 2019-12-13 04:03:51
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,873 bytes
コンパイル時間 1,142 ms
コンパイル使用メモリ 101,832 KB
実行使用メモリ 230,928 KB
最終ジャッジ日時 2024-06-26 06:00:58
合計ジャッジ時間 69,578 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 AC 646 ms
73,552 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 915 ms
74,208 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 138 ms
17,740 KB
testcase_09 AC 90 ms
15,708 KB
testcase_10 AC 3 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 2 ms
5,376 KB
testcase_13 WA -
testcase_14 WA -
testcase_15 AC 3 ms
5,376 KB
testcase_16 AC 86 ms
15,892 KB
testcase_17 AC 2 ms
5,376 KB
testcase_18 AC 4 ms
5,376 KB
testcase_19 AC 2 ms
5,376 KB
testcase_20 AC 2 ms
5,376 KB
testcase_21 TLE -
testcase_22 TLE -
testcase_23 WA -
testcase_24 TLE -
testcase_25 WA -
testcase_26 WA -
testcase_27 AC 899 ms
73,480 KB
testcase_28 TLE -
testcase_29 TLE -
testcase_30 TLE -
testcase_31 TLE -
testcase_32 TLE -
testcase_33 TLE -
testcase_34 TLE -
testcase_35 TLE -
testcase_36 AC 906 ms
73,680 KB
testcase_37 WA -
testcase_38 WA -
testcase_39 AC 904 ms
73,552 KB
testcase_40 WA -
testcase_41 WA -
testcase_42 TLE -
testcase_43 TLE -
testcase_44 TLE -
testcase_45 TLE -
testcase_46 TLE -
testcase_47 AC 2 ms
5,376 KB
testcase_48 TLE -
testcase_49 TLE -
testcase_50 TLE -
testcase_51 TLE -
testcase_52 WA -
testcase_53 WA -
testcase_54 AC 2 ms
6,940 KB
testcase_55 AC 2 ms
6,940 KB
testcase_56 WA -
testcase_57 AC 2 ms
6,944 KB
testcase_58 AC 2 ms
6,940 KB
testcase_59 AC 922 ms
73,712 KB
testcase_60 AC 927 ms
73,660 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <unordered_map>
#include <vector>

using namespace std;


string tos(vector<long long> &a) {
    return to_string(a[0])+","+to_string(a[1])+","+to_string(a[2])+","+to_string(a[3]);
}

void mul(vector<long long> &a, vector<long long> &c, long long p) {
    vector<long long> d(4);
    d[0] = (a[0] * c[0] % p + a[1] * c[2] % p) % p;
    d[1] = (a[0] * c[1] % p + a[1] * c[3] % p) % p;
    d[2] = (c[0] * a[2] % p + c[2] * a[3] % p) % p;
    d[3] = (c[1] * a[2] % p + c[3] * a[3] % p) % p;
    a = d;
}

void modpow(vector<long long> &a, long long k, long long p) {
    if (k > 1) {
        if (k&1) {
            vector<long long> c(a);
            modpow(a, k-1, p);
            mul(a, c, p);
        } else {
            mul(a, a, p);
            modpow(a, k/2, p);
        }
    }
}

int main() {
    long long p;
    vector<long long> a(4), b(4);
    cin >> p;
    for (int i = 0; i < 4; i++) cin >> a[i];
    for (int i = 0; i < 4; i++) cin >> b[i];
    if (a == b) {
        cout << 1 << endl; return 0;
    }
    vector<long long> c(a);
    vector<vector<long long>> s1;
    s1.push_back(a);
    vector<long long> e(a), f(b);
    modpow(e, p-1, p);
    unordered_map<string, int> s2;
    s2[tos(b)] = 0;
    for (int i=1; i <= 1e6; i++) {
        mul(a, c, p);
        s1.push_back(a);
        mul(f, e, p);
        s2[tos(f)] = i;
        if (a == b) {
            cout << (i+1) << endl;
            return 0;
        }
        if (a == c) {
            cout << (-1) << endl;
            return 0;
        }
        
        // cout << a[0] << "," << a[1] << "," << a[2] << "," << a[3] << endl;
    }
    for (int i = 0; i < s1.size(); i++) {
        auto y = s2.find(tos(s1[i]));
        if (y != s2.end()) {
            cout << ((i+1)*y->second) << endl;
            return 0;
        }
    }
    cout << (-1) << endl;
    return 0;
}
0