結果

問題 No.2512 Mountain Sequences
ユーザー maksim
提出日時 2023-10-20 22:53:08
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 807 ms / 3,000 ms
コード長 1,536 bytes
コンパイル時間 1,761 ms
コンパイル使用メモリ 175,384 KB
実行使用メモリ 19,220 KB
最終ジャッジ日時 2024-09-20 21:03:38
合計ジャッジ時間 19,202 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 29
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function 'int32_t main()':
main.cpp:49:14: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17' [-Wc++17-extensions]
   49 |         auto [n,m]=que[i];
      |              ^

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;
#define int long long
const int p=998244353;
int po(int a,int b) {if(b==0) return 1; if(b==1) return a; if(b%2==0) {int u=po(a,b/2);return (u*u)%p;} else {int u=po(a,b-1);return (a*u)%p;}}
int inv(int x) {return po(x,p-2);}
const int maxn=4e5+5;
int fact[maxn];int invf[maxn];
int c(int n,int k)
{
    if(k>n) return 0;
    int ans=fact[n];ans*=invf[k];ans%=p;ans*=invf[n-k];ans%=p;return ans;
}
const int sq=700;
vector<pair<int,int> > h[maxn/sq];
int res[maxn];
int32_t main()
{
    ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    fact[0]=1;for(int i=1;i<maxn;++i) {fact[i]=(fact[i-1]*i)%p;}
    for(int i=0;i<maxn;++i) {invf[i]=inv(fact[i]);}
    vector<pair<int,int> > que;
    int t;cin>>t;
    for(int cyc=0;cyc<t;++cyc)
    {
        int n,m;cin>>n>>m;
        h[(n+sq-1)/sq].push_back({m,cyc});
        que.push_back({n,m});
    }
    for(int i=1;i<=200005/sq+1;++i)
    {
        sort(h[i].begin(),h[i].end());
        int cur=0;int x=0;
        int r=0;
        while(cur!=h[i].size())
        {
            while(x!=h[i][cur].first)
            {
                r+=c(2*x,i*sq-1);r%=p;
                ++x;
            }
            res[h[i][cur].second]=r;
            ++cur;
        }
    }
    for(int i=0;i<que.size();++i)
    {
        auto [n,m]=que[i];
        int n1=((n+sq-1)/sq)*sq;
        int r=res[i];
        while(n1>n)
        {
            r=(-2*r+(c(2*m,n1)));r%=p;
            --n1;
        }
        cout<<(r%p+p)%p<<'\n';
    }
    return 0;
}
0