結果

問題 No.2561 みんな大好きmod 998
コンテスト
ユーザー MM
提出日時 2023-12-02 14:47:46
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 194 ms / 4,000 ms
コード長 970 bytes
コンパイル時間 3,725 ms
コンパイル使用メモリ 233,220 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-26 17:25:28
合計ジャッジ時間 7,055 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 44
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function 'int main()':
main.cpp:33:48: warning: narrowing conversion of '((((__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type)mod998) + A.std::vector<long long int>::operator[](((std::vector<long long int>::size_type)nxt))) % 998)' from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' [-Wnarrowing]
   33 |       st.push({bit+(1<<nxt),nxt,(mod998+A[nxt])%998,(mod353+A[nxt])%mod});
      |                                 ~~~~~~~~~~~~~~~^~~~
main.cpp:33:48: warning: narrowing conversion of '((((__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type)mod998) + A.std::vector<long long int>::operator[](((std::vector<long long int>::size_type)nxt))) % 998)' from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' [-Wnarrowing]
main.cpp:33:68: warning: narrowing conversion of '((((__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type)mod353) + A.std::vector<long long int>::operator[](((std::vector<long long int>::size_type)nxt))) % ((long long int)mod))' from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' [-Wnarrowing]
   33 |       st.push({bit+(1<<nxt),nxt,(mod998+A[nxt])%998,(mod353+A[nxt])%mod});
      |                                                     ~~~~~~~~~~~~~~~^~~~
main.cpp:33:68: warning: narrowing conversion of '((((__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type)mod353) + A.std::vector<long long int>::operator[](((std::vector<long long int>::size_type)nxt))) % ((long long int)mod))' from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' [-Wnarrowing]

ソースコード

diff # 

#include<bits/stdc++.h>
#include<atcoder/all>
#define chmin(x,y) (x) = min((x),(y))
#define chmax(x,y) (x) = max((x),(y))
#define ld long double
using namespace std;
using namespace atcoder;
using ll = long long;
const ll mod = 998244353;
using mint = modint998244353;
using Graph = vector<vector<int>>;
const vector<int> dx = {1,0,-1,0}, dy = {0,1,0,-1};

int main(){
  // input
  int N,K; cin >> N >> K;
  vector<ll> A(N);
  for(int i = 0; i < N; i++) cin >> A[i];
  
  // solve
  ll ans = 0;
  
  stack<vector<int>> st;
  st.push({0,-1,0,0}); // v = {bit,cur,mod998, mod998244353}
  
  while(!st.empty()){
    int bit = st.top()[0], cur = st.top()[1], mod998 = st.top()[2], mod353 = st.top()[3];
    st.pop();
    if(__builtin_popcount(bit) == K){
      if(mod353 <= mod998) ans++;
    }
    else for(int nxt = cur + 1; nxt < N; nxt++){
      st.push({bit+(1<<nxt),nxt,(mod998+A[nxt])%998,(mod353+A[nxt])%mod});
    }
  }
  
  // output
  cout << ans % 998 << endl;
}
0