結果
問題 | No.2206 Popcount Sum 2 |
ユーザー | とりゐ |
提出日時 | 2023-02-03 16:18:32 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 533 ms / 4,000 ms |
コード長 | 1,500 bytes |
コンパイル時間 | 6,583 ms |
コンパイル使用メモリ | 219,716 KB |
実行使用メモリ | 10,044 KB |
最終ジャッジ日時 | 2023-09-15 11:45:38 |
合計ジャッジ時間 | 13,242 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge13 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 6 ms
5,544 KB |
testcase_01 | AC | 6 ms
5,388 KB |
testcase_02 | AC | 158 ms
5,488 KB |
testcase_03 | AC | 167 ms
5,488 KB |
testcase_04 | AC | 166 ms
5,452 KB |
testcase_05 | AC | 531 ms
9,688 KB |
testcase_06 | AC | 533 ms
9,620 KB |
testcase_07 | AC | 533 ms
9,648 KB |
testcase_08 | AC | 532 ms
9,648 KB |
testcase_09 | AC | 530 ms
9,592 KB |
testcase_10 | AC | 276 ms
9,336 KB |
testcase_11 | AC | 275 ms
9,440 KB |
testcase_12 | AC | 276 ms
9,348 KB |
testcase_13 | AC | 238 ms
9,860 KB |
testcase_14 | AC | 237 ms
9,852 KB |
testcase_15 | AC | 236 ms
9,872 KB |
testcase_16 | AC | 147 ms
10,044 KB |
testcase_17 | AC | 147 ms
9,952 KB |
testcase_18 | AC | 149 ms
9,612 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; #include <atcoder/modint> #define elif else if #define rep(i,n) for (int i=0;i<(n);i++) using mint=atcoder::modint998244353; vector<mint>pow2,fac,finv; mint binom(int n,int k){ if(n<0||k<0||k>n)return mint(0); return fac[n]*finv[k]*finv[n-k]; } int main(){ ios::sync_with_stdio(false); cin.tie(nullptr); int N=200010; int B=450; int mod=998244353; fac.resize(N); finv.resize(N); pow2.resize(N); fac[0]=1; for(int i=1;i<N;i++)fac[i]=fac[i-1]*i; finv[N-1]=1/fac[N-1]; for(int i=N-2;i>=0;i--)finv[i]=finv[i+1]*(i+1); pow2[0]=1; for(int i=1;i<N;i++)pow2[i]=pow2[i-1]*2; vector<vector<tuple<int,int,int>>> bucket(B); int T; cin>>T; vector<mint> ans(T,1); for(int i=0;i<T;i++){ int n,m; cin>>n>>m; bucket[m*B/N].push_back({n-1,m-1,i}); ans[i]=pow2[n]-1; } for(int i=0;i<B;i++){ if(i&1){ sort(bucket[i].rbegin(),bucket[i].rend()); } else{ sort(bucket[i].begin(),bucket[i].end()); } } mint res=1; int n=1; int m=0; mint inv2=mint(1)/2; for(auto b:bucket){ for(auto [ni,mi,i]:b){ while(m<mi){ res+=binom(n,m+1); m+=1; } while(m>mi){ res-=binom(n,m); m-=1; } while(n<ni){ res=2*res-binom(n,m); n+=1; } while(n>ni){ res=inv2*(res+binom(n-1,m)); n-=1; } ans[i]*=res; } } for(int i=0;i<T;i++){ cout<<ans[i].val()<<'\n'; } }