結果

問題 No.526 フィボナッチ数列の第N項をMで割った余りを求める
ユーザー llc5pgllc5pg
提出日時 2021-12-15 18:22:06
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 3,696 bytes
コンパイル時間 2,100 ms
コンパイル使用メモリ 194,596 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-23 20:00:38
合計ジャッジ時間 2,757 ms
ジャッジサーバーID
(参考情報)
judge4 / judge5
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
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
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
using ll = long long;

#pragma GCC target ("avx2")
#pragma GCC optimization ("O3")

const int MOD = (1e9+7);
void printmat(const vector<vector<int>>& mat) {
    for (auto row : mat) {
        for (auto elem : row)
            cout << elem << " ";
        cout << endl;
    }
}
void printv(const vector<ll>& v) {
    for (auto elem : v)
        cout << elem << " ";
    cout << endl;
}
void printd(const deque<int>& v) {
    for (auto elem : v)
        cout << elem << " ";
    cout << endl;
}
void printvp(const vector<pair<int,int>>& vp) {
    for (auto pr : vp) {
        cout << pr.first << ", " << pr.second;
        cout << endl;
    }
}
void printvs(const vector<set<int>>& vs) {
    for (auto row : vs) {
        for (auto elem : row)
            cout << elem << ", ";
        cout << endl;
    }
}
void printht(const unordered_map<int, int>& ht) {
    for (auto elem : ht)
        cout << elem.first << " : " << elem.second << endl;
}
void printmp(const map<int, int>& ht) {
    for (auto elem : ht)
        cout << elem.first << " : " << elem.second << endl;
}
void printst(const set<int>& st) {
    for (auto elem : st)
        cout << elem << " ";
    cout << endl;
}
 
bool isPrime(long long n) {
    if (n <= 1)
        return false;
    if (n <= 3)
        return true;
    if (n % 2 == 0 || n % 3 == 0)
        return false;
    for (long long i = 5; i * i <= n; i = i + 6)
        if (n % i == 0 || n % (i + 2) == 0)
            return false;
    return true;
}
 
map<long long, long long> primeFactors(long long n) {
    map<long long, long long> ans;
    while (n % 2 == 0) {
        ans[2]++;
        n = n/2;
    }
    for (long long i = 3; i*i <= (n); i = i + 2) {
        while (n % i == 0) {
            ans[i]++;
            n = n/i;
        }
    }
    if (n > 2)
        ans[n]++;
    return ans;
}
 
int find_f(const vector<int>& uf, int i) {
    while (uf[i]!=i)
        i = uf[i];
    return i;
}
bool union_f(vector<int>& uf, vector<int>& sz, int a, int b) {
    a = find_f(uf, a);
    b = find_f(uf, b);
    //cout << "a, b = " << a << ", " << b << endl;
    if (a==b) return false;
    if (sz[a] < sz[b]) {
        //cout << "sz[a], sz[b] = " << sz[a] << ", " << sz[b] << endl;
        //cout << "a, b = " << a << ", " << b << endl;
        swap(a,b);
        //cout << "a, b = " << a << ", " << b << endl;
    }
    sz[a] += sz[b];
    uf[b] = a;
    return true;
}

long long modexp(long long b, long long e, long long M) {
    if (!e) return 1;
    b %= M;
    long long x = modexp(b * b % M, e / 2, M);
    if (e % 2) {
        return b * x % M;
    } else {
        return x;
    }
}

vector<ll> helper(vector<ll> v, ll e, ll M) {
    //cout << "e = " << e << endl;
    //printv(v);
    if (e==1) return v;
    vector<ll> w(4);
    w[0] = ((v[0]*v[0])%M + (v[1]*v[2])%M) % M;
    w[1] = ((v[0]*v[1])%M + (v[1]*v[3])%M) % M;
    w[2] = ((v[2]*v[0])%M + (v[3]*v[2])%M) % M;
    w[3] = ((v[2]*v[1])%M + (v[3]*v[3])%M) % M;
    w = helper(w, e/2, M);
#if 1    
    if (e%2) {
        vector<ll> x(4);
        x[0] = ((v[0]*w[0])%M + (v[1]*w[2])%M) % M;
        x[1] = ((v[0]*w[1])%M + (v[1]*w[3])%M) % M;
        x[2] = ((v[2]*w[0])%M + (v[3]*w[2])%M) % M;
        x[3] = ((v[2]*w[1])%M + (v[3]*w[3])%M) % M;
        w = x;
    }
#endif
    //printv(w);
    return w;
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);    
    int T=1, caseIdx=0;
    //cin >> T;
    while (T--) {
        //caseIdx++;
        ll n, m, ans=0;
        cin >> n >> m;
        vector<ll> v = {0,1,1,1};
        v = helper(v, n, m);
        ans = v[0];
        //printv(v);
        cout << ans << endl;
    }
}
 
 

0