結果

問題 No.2512 Mountain Sequences
ユーザー momoyuu
提出日時 2023-10-29 12:18:47
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,433 bytes
コンパイル時間 3,085 ms
コンパイル使用メモリ 257,580 KB
実行使用メモリ 8,448 KB
最終ジャッジ日時 2024-09-25 16:39:59
合計ジャッジ時間 15,584 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other WA * 11 RE * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

#include<atcoder/modint>
using mint = atcoder::modint998244353;


const int B = 500;
mint fac[2<<17],ifac[2<<17];
mint comb(int n,int r){
    if(r>n) return 0;
    return fac[n] * ifac[r] * ifac[n-r];
}
int main(){
    cin.tie(nullptr);
    ios::sync_with_stdio(false);
    
    fac[0] = 1;
    for(int i = 1;i<2<<17;i++) fac[i] = fac[i-1] * i;
    for(int i = 0;i<2<<17;i++) ifac[i] = fac[i].inv();


    int t;
    cin>>t;
    vector<int> n(t),m(t);
    for(int i = 0;i<t;i++) cin>>n[i]>>m[i];
    vector<int> idx(t);
    for(int i = 0;i<t;i++) idx[i] = i;
    sort(idx.begin(),idx.end(),[&](int i,int j){
        if(n[i]/B!=n[j]/B) return n[i]/B < n[j]/B;
        if(n[i]/B%2) return m[i] < m[j];
        return m[i] > m[j];});

    int nn = 1,mm = 1;
    mint tmp = 1;
    vector<mint> ans(t);

    for(int i = 0;i<t;i++){
        int ni = idx[i];
        while(m[ni]>mm){
            tmp += comb(2*mm,nn-1);
            mm++;
        }
        while(m[ni]<mm){
            tmp -= comb(2*(mm-1),nn-1);
            mm--;
        }
        while(n[ni]>nn){
            mint now = comb(2*mm,nn+1);
            tmp = (now-tmp)/2;
            nn++;
        }
        while(n[ni]<nn){
            mint now = (2*mm,nn);
            tmp = now - 2 * tmp;
            nn--;
        }
        ans[ni] = tmp;

    }
    for(int i = 0;i<t;i++) cout<<ans[i].val()<<endl;


}
0