結果

問題 No.2561 みんな大好きmod 998
ユーザー Shibaken28Shibaken28
提出日時 2023-12-04 14:57:09
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 546 ms / 4,000 ms
コード長 4,724 bytes
コンパイル時間 1,909 ms
コンパイル使用メモリ 144,376 KB
実行使用メモリ 221,796 KB
最終ジャッジ日時 2024-09-26 23:00:51
合計ジャッジ時間 9,619 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 22 ms
12,000 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 512 ms
221,796 KB
testcase_04 AC 531 ms
221,672 KB
testcase_05 AC 526 ms
221,796 KB
testcase_06 AC 519 ms
221,796 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 3 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 77 ms
37,096 KB
testcase_12 AC 2 ms
5,376 KB
testcase_13 AC 2 ms
5,376 KB
testcase_14 AC 2 ms
5,376 KB
testcase_15 AC 546 ms
221,672 KB
testcase_16 AC 2 ms
5,376 KB
testcase_17 AC 2 ms
5,376 KB
testcase_18 AC 2 ms
5,376 KB
testcase_19 AC 408 ms
199,780 KB
testcase_20 AC 2 ms
5,376 KB
testcase_21 AC 2 ms
5,376 KB
testcase_22 AC 2 ms
5,376 KB
testcase_23 AC 2 ms
5,376 KB
testcase_24 AC 2 ms
5,376 KB
testcase_25 AC 2 ms
5,376 KB
testcase_26 AC 112 ms
52,324 KB
testcase_27 AC 521 ms
221,668 KB
testcase_28 AC 146 ms
65,508 KB
testcase_29 AC 41 ms
20,452 KB
testcase_30 AC 126 ms
55,008 KB
testcase_31 AC 295 ms
113,124 KB
testcase_32 AC 58 ms
23,780 KB
testcase_33 AC 40 ms
20,320 KB
testcase_34 AC 517 ms
221,752 KB
testcase_35 AC 35 ms
15,820 KB
testcase_36 AC 35 ms
15,848 KB
testcase_37 AC 17 ms
8,676 KB
testcase_38 AC 78 ms
36,968 KB
testcase_39 AC 126 ms
55,012 KB
testcase_40 AC 131 ms
55,016 KB
testcase_41 AC 77 ms
37,092 KB
testcase_42 AC 146 ms
65,636 KB
testcase_43 AC 17 ms
8,804 KB
testcase_44 AC 59 ms
27,876 KB
testcase_45 AC 394 ms
199,784 KB
testcase_46 AC 48 ms
23,652 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream> // cout, endl, cin
#include <string> // string, to_string, stoi
#include <vector> // vector
#include <algorithm> // min, max, swap, sort, reverse, lower_bound, upper_bound
#include <utility> // pair, make_pair
#include <tuple> // tuple, make_tuple
#include <cstdint> // int64_t, int*_t
#include <cstdio> // printf
#include <map> // map
#include <queue> // queue, priority_queue
#include <set> // set
#include <stack> // stack
#include <deque> // deque
#include <unordered_map> // unordered_map
#include <unordered_set> // unordered_set
#include <bitset> // bitset
#include <cctype> // isupper, islower, isdigit, toupper, tolower
#include <iomanip>
#include <climits>
#include <cmath>
#include <functional>
#include <numeric>
#include <regex>
#include <array>
#include <fstream>
#include <sstream>

//#include<atcoder/modint>
//using namespace atcoder;

#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define rep1(i, n) for (int i = 1; i <= (int)(n); i++)
#define repl(i,l,r) for (int i = l; i < (int)(r); i++)
#define all(a) a.begin(),a.end()
#define Pii pair<int,int>
#define Pll pair<long,long>
#define INFi 1000000001
#define INFl 1000000000000000001
#define ll long long
using namespace std;



template<class T> inline bool chmin(T& a, T b) {
    if (a > b) {
        a = b;
        return true;
    }
    return false;
}
template<class T> inline bool chmax(T& a, T b) {
    if (a < b) {
        a = b;
        return true;
    }
    return false;
}

template<class T> void printArray(vector<T>&A){
    for(T&a:A){
        cout<<a<<" ";
    }
    cout<<endl;
}
template<class T> void printArrayln(vector<T>&A){
    for(T&a:A){
        cout<<a<<endl;
    }
}


template<class T1,class T2> std::ostream &operator<<(std::ostream &out, const pair<T1,T2> &A){
    cout<<"{"<<A.first<<","<<A.second<<"}";
    return out;
}

template<class T1,class T2> std::ostream &operator<<(std::ostream &out, const map<T1,T2> &M){
    for(const auto&A:M){
        cout<<"{"<<A.first<<","<<A.second<<"}";
    }
    return out;
}

template<class T1> std::ostream &operator<<(std::ostream &out, const set<T1> &M){
    cout<<"{";
    for(const auto&A:M){
        cout<<A<<", ";
    }
    cout<<"}"<<endl;
    return out;
}


template<class T1> std::ostream &operator<<(std::ostream &out, const multiset<T1> &M){
    cout<<"{";
    for(const auto&A:M){
        cout<<A<<", ";
    }
    cout<<"}"<<endl;
    return out;
}

template<class T> std::ostream &operator<<(std::ostream &out, const vector<T> &A){
    for(const T &a:A){
        cout<<a<<" ";
    }
    return out;
}

void print() { cout << endl; }
 
template <typename Head, typename... Tail>
void print(Head H, Tail... T) {
  cout << H << " ";
  print(T...);
}


template<class T> std::istream &operator>>(std::istream &in,vector<T>&A){
    for(T&a:A){
        std::cin>>a;
    }
    return in;
}


vector<vector<int>> permutations(int N){
    vector<int> array(N);
    vector<vector<int>> A(0);
    for(int i=0;i<N;i++)array[i]=i;
    do{
        A.push_back(array);
    }while(next_permutation(array.begin(),array.end()));
    return A;
}



vector<vector<int>> permutation(int N,int U,bool h=false){
    //0,1,2,3,...,N-1縺ｮN蛟九°繧蔚蛟矩∈縺ｶ鬆・・
    vector<vector<int>> A(0);
    auto fun = [&h,&A,&N,&U](auto &fun,vector<int> &B)->void{
        if((int)B.size()==U){
            auto C=B;
            do{
                A.push_back(C);
            }while(next_permutation(C.begin(),C.end()));
            return;
        }
        int s = (h?0:-1);
        if(!B.empty())s = B.back();
        for(int x=s+(h?0:1);x<N;x++){
            B.push_back(x);
            fun(fun,B);
            B.pop_back();
        }
    };
    vector<int> C={};
    fun(fun,C);
    return A;
}

vector<vector<int>> combination(int N,int U,bool h=false){
    //0,1,2,3,...,N-1縺ｮN蛟九°繧蔚蛟矩∈縺ｶ邨・∩蜷医ｏ縺幃・・
    vector<vector<int>> A(0);
    auto fun = [&h,&A,&N,&U](auto &fun,vector<int> &B)->void{
        if((int)B.size()==U){
            auto C=B;
            A.push_back(B);
            return;
        }
        int s = (h?0:-1);
        if(!B.empty())s = B.back();
        for(int x=s+(h?0:1);x<N;x++){
            B.push_back(x);
            fun(fun,B);
            B.pop_back();
        }
    };
    vector<int> C={};
    fun(fun,C);
    return A;

}

int main(void){
    std::cin.tie(0)->sync_with_stdio(0);
    long N,K;cin>>N>>K;
    vector<long> A(N);
    rep(i,N)cin>>A[i];
    int cnt = 0;
    auto B = combination(N,K,false);
    for(auto&C:B){
        long sum = 0;
        for(auto&c:C){
            sum += A[c];
        }
        if(sum%998244353<=sum%998)cnt++;
    }
    cout<<cnt%998<<endl;
}
0