結果

問題 No.2060 AND Sequence
ユーザー Nzt3Nzt3
提出日時 2022-08-27 01:01:35
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 11 ms / 2,000 ms
コード長 802 bytes
コンパイル時間 2,036 ms
コンパイル使用メモリ 194,240 KB
最終ジャッジ日時 2025-01-31 06:03:17
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 43
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
const int mod=998244353;

long long comb[31][31];

long long modpow(int a,int n){
  long long ret=1,t=a;
  for(int i=0;i<30;i++){
    if(n>>i&1)ret=ret*t%mod;
    t=t*t%mod;
  }
  return ret;
}

int main(){
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  for(int i=0;i<=30;i++){
    comb[i][0]=1;
  }
  for(int i=1;i<=30;i++){
    for(int j=1;j<=i;j++){
      comb[i][j]=comb[i][j-1]*(i-j+1)%mod*modpow(j,mod-2)%mod;
    }
  }
  int N,M;
  cin>>N>>M;
  vector<int>cnt(31);
  for(int i=30,t=0;i>=0;i--){
    if((M>>i&1)==0)continue;
    for(int j=0;j<=i;j++){
      cnt[t+j]=(cnt[t+j]+comb[i][j])%mod;
    }
    ++t;
  }
  cnt[__popcount(M)]+=1;
  long long ans=0;
  for(int i=0;i<=30;i++){
    ans=(ans+modpow(N,i)*cnt[i])%mod;
  }
  cout<<ans<<'\n';
}
0