結果

問題 No.2206 Popcount Sum 2
ユーザー Taiki0715Taiki0715
提出日時 2024-09-28 16:50:38
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 548 ms / 4,000 ms
コード長 8,151 bytes
コンパイル時間 4,739 ms
コンパイル使用メモリ 317,996 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-09-28 16:50:55
合計ジャッジ時間 15,686 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,820 KB
testcase_02 AC 135 ms
6,816 KB
testcase_03 AC 143 ms
6,820 KB
testcase_04 AC 139 ms
6,816 KB
testcase_05 AC 451 ms
6,824 KB
testcase_06 AC 470 ms
6,820 KB
testcase_07 AC 454 ms
6,820 KB
testcase_08 AC 445 ms
6,816 KB
testcase_09 AC 446 ms
6,816 KB
testcase_10 AC 547 ms
6,820 KB
testcase_11 AC 504 ms
6,816 KB
testcase_12 AC 508 ms
6,820 KB
testcase_13 AC 548 ms
6,816 KB
testcase_14 AC 537 ms
6,816 KB
testcase_15 AC 531 ms
6,816 KB
testcase_16 AC 505 ms
6,820 KB
testcase_17 AC 534 ms
6,820 KB
testcase_18 AC 515 ms
6,820 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#if __has_include(<atcoder/all>)
#include <atcoder/all>
#endif
using ll=long long;
using ull=unsigned long long;
using P=pair<ll,ll>;
template<typename T>using minque=priority_queue<T,vector<T>,greater<T>>;
template<typename T>bool chmax(T &a,const T &b){return (a<b?(a=b,true):false);}
template<typename T>bool chmin(T &a,const T &b){return (a>b?(a=b,true):false);}
template<typename T1,typename T2>istream &operator>>(istream &is,pair<T1,T2>&p){is>>p.first>>p.second;return is;}
template<typename T>istream &operator>>(istream &is,vector<T> &a){for(auto &i:a)is>>i;return is;}
template<typename T1,typename T2>void operator++(pair<T1,T2>&a,int n){a.first++,a.second++;}
template<typename T1,typename T2>void operator--(pair<T1,T2>&a,int n){a.first--,a.second--;}
template<typename T>void operator++(vector<T>&a,int n){for(auto &i:a)i++;}
template<typename T>void operator--(vector<T>&a,int n){for(auto &i:a)i--;}
#define reps(i,a,n) for(int i=(a);i<int(n);i++)
#define rep(i,n) reps(i,0,n)
#define all(x) x.begin(),x.end()
#define pcnt(x) __builtin_popcountll(x)
#define fin(x) return cout<<x<<'\n',static_cast<void>(0)
ll myceil(ll a,ll b){return (a+b-1)/b;}
template<typename T,size_t n,size_t id=0>
auto vec(const int (&d)[n],const T &init=T()){
  if constexpr (id<n)return vector(d[id],vec<T,n,id+1>(d,init));
  else return init;
}
#ifdef LOCAL
#include<debug.h>
#else
#define debug(...) static_cast<void>(0)
#define debugg(...) static_cast<void>(0)
template<typename T1,typename T2>ostream &operator<<(ostream &os,const pair<T1,T2>&p){os<<p.first<<' '<<p.second;return os;}
#endif
void SOLVE();
int main(){
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  cout<<fixed<<setprecision(16);
  #ifdef LOCAL
  clock_t start=clock();
  #endif
  int testcase=1;
  cin>>testcase;
  for(int i=0;i<testcase;i++){
    SOLVE();
  }
  #ifdef LOCAL
  cerr<<"time:";
  cerr<<(clock()-start)/1000;
  cerr<<"ms\n";
  #endif
}
template<int m>
struct mod_int{
private:
  using uint=unsigned int;
  using mint=mod_int;
  uint v;
  static_assert(m<uint(1)<<31);
public:
  mod_int():v(0){}
  mod_int(ll a){
    a%=m;
    if(a<0)v=a+m;
    else v=a;
  }
  static mint raw(int a){
    mint ret;
    ret.v=a;
    return ret;
  }
  inline uint val()const{return this->v;}
  static constexpr int mod(){return m;}
  inline mint &operator+=(const mint &b){
    this->v+=b.v;
    if(this->v>=m)this->v-=m;
    return *this;
  }
  inline mint &operator-=(const mint &b){
    this->v-=b.v;
    if(this->v>=m)this->v+=m;
    return *this;
  }
  inline mint &operator*=(const mint &b){
    this->v=(uint64_t(this->v)*uint64_t(b.v))%m;
    return *this;
  }
  inline mint &operator/=(const mint &b){
    *this*=b.inv();
    return *this;
  }
  inline mint operator+()const{return *this;}
  inline mint operator-()const{return mint()-*this;}
  friend inline mint operator+(const mint &a,const mint &b){return mint(a)+=b;}
  friend inline mint operator-(const mint &a,const mint &b){return mint(a)-=b;}
  friend inline mint operator*(const mint &a,const mint &b){return mint(a)*=b;}
  friend inline mint operator/(const mint &a,const mint &b){return mint(a)/=b;}
  friend inline bool operator==(const mint &a,const mint &b){return a.val()==b.val();}
  friend inline bool operator!=(const mint &a,const mint &b){return !(a==b);}
  inline mint operator++(int){
    mint ret=*this;
    *this+=1;
    return ret;
  }
  inline mint operator--(int){
    mint ret=*this;
    *this-=1;
    return ret;
  }
  mint pow(ll n)const{
    mint ret=1,a(*this);
    while(n){
      if(n&1)ret*=a;
      a*=a;
      n>>=1;
    }
    return ret;
  }
  inline mint inv()const{
    assert(this->v!=0);
    return pow(m-2);
  }
  mint sqrt()const{
    if(this->val()<=1)return *this;
    if constexpr(m%8==1){
      mint b=2;
      while(b.pow((m-1)/2).val()==1)b++;
      int m2=m-1,e=0;
      while(m2%2==0)m2>>=1,e++;
      mint x=this->pow((m2-1)/2);
      mint y=(*this)*x*x;
      x*=*this;
      mint z=b.pow(m2);
      while(y.val()!=1){
        int j=0;
        mint t=y;
        while(t.val()!=1)t*=t,j++;
        z=z.pow(1<<(e-j-1));
        x*=z;
        z*=z;
        y*=z;e=j;
      }
      return x;
    }
    else if constexpr(m%8==5){
      mint ret=this->pow((m+3)/8);
      if((ret*ret).val()==this->val())return ret;
      else return ret*mint::raw(2).pow((m-1)/4);
    }
    else{
      return this->pow((m+1)/4);
    }
  }
  pair<bool,mint> safe_sqrt()const{
    if(this->pow((m-1)/2)==1)return make_pair(true,this->sqrt());
    else return make_pair(false,0);
  }
  friend istream &operator>>(istream &is,mint &b){
    ll a;
    is>>a;
    b=mint(a);
    return is;
  }
  friend ostream &operator<<(ostream &os,const mint &b){
    os<<b.val();
    return os;
  }
};
using mint998=mod_int<998244353>;
using mint107=mod_int<1000000007>;
template<typename T>
struct F{
private:
  static int capacity;
  static vector<T>fact,factinv,inv;
  static void resize(int n){
    fact.resize(n+1),factinv.resize(n+1),inv.resize(n+1);
    for(int i=capacity+1;i<=n;i++){
      fact[i]=fact[i-1]*i;
      inv[i]=-inv[T::mod()%i]*(T::mod()/i);
      factinv[i]=factinv[i-1]*inv[i];
    }
    capacity=n;
  }
public:
  static T C(int n,int k){
    if(n<k)return 0;
    if(k<0)return 0;
    if(capacity<n)resize(n);
    return fact[n]*factinv[k]*factinv[n-k];
  }
  static T P(int n,int k){
    if(n<k)return 0;
    if(k<0)return 0;
    if(capacity<n)resize(n);
    return fact[n]*factinv[n-k];
  }
  static T H(int n,int k){
    if(n==0&&k==0)return 1;
    return C(n+k-1,k);
  }
  static T factorial(int n){
    if(capacity<n)resize(n);
    return fact[n];
  }
  static T factorial_inv(int n){
    if(capacity<n)resize(n);
    return factinv[n];
  }
  static T S(ll n,int k){
    if(n<0)return 0;
    if(n<k)return 0;
    T ret=0;
    if(capacity<k)resize(k);
    rep(i,k+1){
      ret+=fact[k]*factinv[i]*factinv[k-i]*T::raw(i).pow(n)*((k-i)&1?-1:1);
    }
    return ret*factinv[k];
  }
  template<typename... INT>
  static T O(INT...k){
    int n=0;
    for(int i:initializer_list<int>{k...}){
      if(i<0)return 0;
      n+=i;
    }
    if(capacity<n)resize(n);
    T ret=fact[n];
    for(int i:initializer_list<int>{k...})ret*=factinv[i];
    return ret;
  }
};
template<typename T>int F<T>::capacity=1;
template<typename T>vector<T>F<T>::fact{1,1};
template<typename T>vector<T>F<T>::factinv{1,1};
template<typename T>vector<T>F<T>::inv{0,1};
using namespace std;
template<typename T>
struct BinomialPrefixSum{
private:
  static vector<vector<T>>table;
  static const T inv2;
  static void resize(int n){
    if(table.size()>n)return;
    int pre=table.size();
    table.resize(n+1);
    for(int i=pre;i<=n;i++){
      table[i].resize(i+1);
      table[i][0]=1;
      for(int j=1;j<=i;j++){
        table[i][j]=table[i][j-1];
        for(int k=1;k<=i;k++)table[i][j]+=F<T>::C(i*i,i*(j-1)+k);
      }
    }
  }
public:
  static T get(int n,int k){
    if(n<k)k=n;
    if(k<0)return 0;
    int sn=sqrt(n);
    resize(sn+1);
    if(sn>0&&n-sn*sn<(sn+1)*(sn+1)-n){
      int sk=min(sn,k/sn);
      int sk2=sk*sn;
      T now=table[sn][sk];
      for(int i=sn*sn;i<n;i++)now=now*2-F<T>::C(i,sk2);
      for(int i=sk2+1;i<=k;i++)now+=F<T>::C(n,i);
      return now;
    }
    else{
      sn++;
      int sk=k/sn;
      if(k-sk*sn<(sk+1)*sn-k){
        int sk2=sk*sn;
        T now=table[sn][sk];
        for(int i=sn*sn-1;i>=n;i--)now=(now+F<T>::C(i,sk2))*inv2;
        for(int i=sk2+1;i<=k;i++)now+=F<T>::C(n,i);
        return now;
      }
      else{
        sk++;
        int sk2=sk*sn;
        int sn2=sn*sn;
        T now=table[sn][sk];
        for(int i=sk2;i>k;i--)now-=F<T>::C(sn2,i);
        for(int i=sn2-1;i>=n;i--)now=(now+F<T>::C(i,k))*inv2;
        return now;
      }
    }
  }
  static inline T get(int n,int k1,int k2){
    return get(n,k2)-get(n,k1-1);
  }
};
template<typename T>vector<vector<T>>BinomialPrefixSum<T>::table;
template<typename T>const T BinomialPrefixSum<T>::inv2(T(2).inv());
using mint=mint998;
void SOLVE(){
  int n,m;
  cin>>n>>m;
  cout<<(mint(2).pow(n)-1)*BinomialPrefixSum<mint>::get(n-1,m-1)<<'\n';
}
0