結果

問題 No.2561 みんな大好きmod 998
ユーザー kazuppakazuppa
提出日時 2024-09-01 21:19:23
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 55 ms / 4,000 ms
コード長 2,464 bytes
コンパイル時間 6,028 ms
コンパイル使用メモリ 338,292 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-09-01 21:19:32
合計ジャッジ時間 8,456 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 44
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
#pragma GCC target ("avx")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
using namespace std;
using namespace atcoder;
using ll=long long;
#define rep(i,a,b) for(it i=(it)(a);i<=(it)b;i++)
#define nrep(i,a,b) for(it i=(it)(a);i>=(it)b;i--)
#define all(v) v.begin(), v.end()
#define rall(v) v.rbegin(), v.rend()
#define moda 998244353LL
#define modb 1000000007LL
#define gyaku 166374059LL
#define dai 2500000000000000000LL
#define tyu 2500000000
#define giri 1000000000
#define en 3.14159265358979
#define endl '\n'
#define sho -dai
#define eps 1e-14
#define yn(x) cout<<(x ? "Yes\n":"No\n");
#define YN(x) cout<<(x ? "YES\n":"NO\n");
#define cou(x) cout<<x<<endl;
#define ci(x) cin>>x;
using it=long long;
using un=unsigned long long;
using db=long double;
using st=string;
using ch=char;
using bo=bool;
using P=pair<it,it>;
using ip=pair<int,int>;
using mint=modint;
using mints=modint998244353;
using minto=modint1000000007;
using vi=vector<it>;
using ivi=vector<int>;
using vd=vector<db>;
using vs=vector<st>;
using vc=vector<ch>;
using vb=vector<bo>;
using vp=vector<P>;
using ivp=vector<ip>;
using vm=vector<mint>;
using vms=vector<mints>;
using vmo=vector<minto>;
using sp=set<P>;
using isp=set<ip>;
using ss=set<st>;
using sc=set<ch>;
using si=set<it>;
using isi=set<int>;
using svi=set<vi>;
using vvi=vector<vi>;
using ivvi=vector<ivi>;
using vvd=vector<vd>;
using vvs=vector<vs>;
using vvb=vector<vb>;
using vvc=vector<vc>;
using vvp=vector<vp>;
using ivvp=vector<ivp>;
using vvm=vector<vm>;
using vvms=vector<vms>;
using vvmo=vector<vmo>;
using vsi=vector<si>;
using ivsi=vector<isi>;
using vsp=vector<sp>;
using ivsp=vector<isp>;
using vvsi=vector<vsi>;
using ivvsi=vector<ivsi>;
using vvsp=vector<vsp>;
using ivvsp=vector<ivsp>;
using vvvi=vector<vvi>;
using ivvvi=vector<ivvi>;
using vvvd=vector<vvd>;
using vvvm=vector<vvm>;
using vvvvi=vector<vvvi>;
using ivvvvi=vector<ivvvi>;

int ans=0;
void func(int l,int n,int k,ivi &x,vi &a){
  if(x.size()==k){
    it sum=0;
    rep(i,0,k-1)
      sum+=a[x[i]];
    if(sum%998>=sum%moda)ans++;
    return;
  }
  rep(i,l+1,n-1){
    x.push_back(i);
    func(i,n,k,x,a);
    x.pop_back();
  }
}

int main(){
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  int n,k;cin>>n>>k;
  vi a(n);ivi x;
  rep(i,0,n-1)cin>>a[i];
  func(-1,n,k,x,a);
  cout<<ans%998<<endl;
}
0