結果

問題 No.2561 みんな大好きmod 998
ユーザー twooimp2
提出日時 2023-12-02 14:52:38
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 128 ms / 4,000 ms
コード長 619 bytes
コンパイル時間 2,787 ms
コンパイル使用メモリ 194,572 KB
最終ジャッジ日時 2025-02-18 04:01:21
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 44
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>
using namespace std;
using ll=long long;

ll n,k;
ll ans=0;

void dfs(ll now,vector<ll> &a,vector<ll> &used){
  if(used.size()==k){
    ll sum1=0;
    ll sum2=0;
    for(ll i=0;i<k;i++){
      sum1+=a[used[i]];
      sum1%=998244353;
      sum2+=a[used[i]];
      sum2%=998;
    }
    if(sum1<=sum2){
      ans++;
      ans%=998;
    }
    return;
  }
  for(ll i=now+1;i<=n;i++){
    used.push_back(i);
    dfs(i,a,used);
    used.pop_back();
  }
}

int main(){
  cin>>n>>k;
  vector<ll> a(n+1);
  for(ll i=1;i<=n;i++){
    cin>>a[i];
  }
  vector<ll> used;
  dfs(0,a,used);
  cout<<ans<<endl;
}
0