結果

問題 No.3310 mod998
コンテスト
ユーザー applejam
提出日時 2026-02-06 16:05:14
言語 C++23
(gcc 15.2.0 + boost 1.89.0)
結果
AC  
実行時間 835 ms / 2,000 ms
コード長 2,261 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 14,190 ms
コンパイル使用メモリ 723,068 KB
実行使用メモリ 7,972 KB
最終ジャッジ日時 2026-02-06 16:05:43
合計ジャッジ時間 28,419 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
#include <atcoder/all>
#include <boost/multiprecision/cpp_dec_float.hpp>
#include <boost/multiprecision/cpp_int.hpp>
namespace mp = boost::multiprecision;
// 任意長整数型
using Bint = mp::cpp_int;
// 仮数部が10進数で1024桁の浮動小数点数型(TLEしたら小さくする)
using Real = mp::number<mp::cpp_dec_float<1024>>;
using namespace std;
using namespace atcoder;
using ll = long long;
using mint = modint998244353;
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vll = vector<ll>;
using vvll = vector<vll>;
using vvvll = vector<vvll>;
using vmi = vector<mint>;
using vvmi = vector<vmi>;
using vvvmi = vector<vvmi>;
#define all(a) (a).begin(), (a).end()
#define rep2(i, m, n) for (int i = (m); i < (n); ++i)
#define rep(i, n) rep2(i, 0, n)
#define drep2(i, m, n) for (int i = (m)-1; i >= (n); --i)
#define drep(i, n) drep2(i, n, 0)

void solve(){
    ll n, m; cin >> n >> m;
    set<int> st;
    st.insert(1);
    vi v(1, 1);
    ll a = 1;
    int loopstart;
    while(true){
        a *= n; a %= 998;
        if(st.count(a)){
            loopstart = a;
            break;
        }
        v.push_back(a);
        st.insert(a);
    }
    int slen = v.size();
    vector<Bint> vs(slen, v[0]); rep2(i, 1, slen)vs[i] = vs[i-1] + v[i];
    vector<Bint> vl(1, loopstart);
    set<int> st2; st2.insert(loopstart);
    a = loopstart;
    while(true){
        a*=n; a %= 998;
        if(st2.count(a))break;
        vl.push_back(a);
        st2.insert(a);
    }
    int llen = vl.size(); Bint s = 0; rep(i, llen)s+= vl[i];
    vector<Bint> r(llen, 0);
    rep(i, llen-1){
        r[i+1] = r[i] + vl[i];
        r[i+1] %= 998; 
    }
    rep(i, m){
        Bint k; cin >> k;
        k++;
        if(n == 0){
            cout << 1 << endl;
            continue;
        }
        if(k <= slen){
            cout << vs[(int)(k-1)]%998 << endl;
            continue;
        }
        k -= slen;
        Bint loop = k/(Bint)llen;
        Bint ans = loop*s + vs[slen-1];
        int amari = (int)(k % (Bint)llen);
        cout << (ans + r[amari])%998 << endl;
    }
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int t; cin >> t;
    while(t--)solve();
    return 0;
}
0