結果

問題 No.2561 みんな大好きmod 998
ユーザー AsahiAsahi
提出日時 2023-12-02 15:38:00
言語 Java21
(openjdk 21)
結果
AC  
実行時間 1,737 ms / 4,000 ms
コード長 14,646 bytes
コンパイル時間 3,955 ms
コンパイル使用メモリ 98,712 KB
実行使用メモリ 395,604 KB
最終ジャッジ日時 2023-12-02 15:38:31
合計ジャッジ時間 29,488 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 113 ms
57,324 KB
testcase_01 AC 241 ms
70,280 KB
testcase_02 AC 112 ms
57,464 KB
testcase_03 AC 1,737 ms
395,596 KB
testcase_04 AC 1,652 ms
395,576 KB
testcase_05 AC 1,649 ms
395,600 KB
testcase_06 AC 1,631 ms
395,556 KB
testcase_07 AC 120 ms
57,312 KB
testcase_08 AC 111 ms
56,948 KB
testcase_09 AC 153 ms
59,204 KB
testcase_10 AC 113 ms
57,204 KB
testcase_11 AC 353 ms
112,352 KB
testcase_12 AC 116 ms
57,436 KB
testcase_13 AC 114 ms
56,944 KB
testcase_14 AC 120 ms
57,448 KB
testcase_15 AC 1,677 ms
395,604 KB
testcase_16 AC 117 ms
57,452 KB
testcase_17 AC 113 ms
57,344 KB
testcase_18 AC 129 ms
57,592 KB
testcase_19 AC 1,015 ms
308,812 KB
testcase_20 AC 124 ms
57,056 KB
testcase_21 AC 141 ms
57,320 KB
testcase_22 AC 114 ms
56,940 KB
testcase_23 AC 117 ms
56,928 KB
testcase_24 AC 117 ms
56,940 KB
testcase_25 AC 120 ms
57,436 KB
testcase_26 AC 464 ms
130,900 KB
testcase_27 AC 1,675 ms
395,572 KB
testcase_28 AC 493 ms
167,692 KB
testcase_29 AC 272 ms
80,940 KB
testcase_30 AC 512 ms
128,024 KB
testcase_31 AC 881 ms
230,028 KB
testcase_32 AC 350 ms
102,108 KB
testcase_33 AC 281 ms
80,952 KB
testcase_34 AC 1,649 ms
395,564 KB
testcase_35 AC 300 ms
81,252 KB
testcase_36 AC 273 ms
81,268 KB
testcase_37 AC 189 ms
68,564 KB
testcase_38 AC 318 ms
112,360 KB
testcase_39 AC 507 ms
127,952 KB
testcase_40 AC 492 ms
127,988 KB
testcase_41 AC 344 ms
112,380 KB
testcase_42 AC 531 ms
167,704 KB
testcase_43 AC 210 ms
68,576 KB
testcase_44 AC 329 ms
95,140 KB
testcase_45 AC 1,018 ms
308,820 KB
testcase_46 AC 307 ms
86,060 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.*; 
import java.io.*;
import java.math.*;
import java.util.stream.*;
import java.util.function.*;

class Main implements Runnable {

    public void solve(){    
        int n = in.nextInt() , k = in.nextInt();
        int [] a = in.nextInt(n);
        int [] e = new int[n];
        for(int i = 0 ; i < n ; i ++) e[i] = i ;
        int ans = 0 ;
        List<List<Integer>> s = generateCombinations(e , k);
        for(List<Integer> r : s) {
            int s998 = 0 , s998244353 = 0 ;
            for(int v : r) {
                s998 = (s998 + a[v]) % 998 ;
                s998244353 = (s998244353 + a[v]) % 998244353 ;
            }
            if(s998244353 <= s998) ans = (ans + 1) % 998 ;
        }
        print(ans);
    }

    public List<List<Integer>> generateCombinations(int[] elements, int r) {
        List<List<Integer>> combinations = new ArrayList<>();
        combine(elements, 0, r, new ArrayList<>(), combinations);
        return combinations;
    }

    private void combine(int[] elements, int start, int r, List<Integer> current, List<List<Integer>> combinations) {
        if (r == 0) {
            combinations.add(new ArrayList<>(current));
            return;
        }
        for (int i = start; i <= elements.length - r; i++) {
            current.add(elements[i]);
            combine(elements, i + 1, r - 1, current, combinations);
            current.remove(current.size() - 1);
        }
    }

    public record Pair<T,S> (T fi , S se) { 
        public String toString() { 
            return "("+fi+","+se+")";
        }
    }

    public record Trio<T,S,U>(T fi , S se , U th) { 
        public String toString() { 
            return "("+fi+","+se+","+th+")";
        }
    }

    public PrintWriter out = new PrintWriter(System.out);
    public In in = new In();
    public static final int mod7 = 1000000007;
    public static final int mod9 = 998244353;
    public static final int  inf = (1  << 30);
    public static final long lnf = (1L << 60);
    public static final String yes = "Yes";
    public static final String no  = "No" ;
    public static final int [] dy4 = {-1,0,1,0};
    public static final int [] dx4 = {0,1,0,-1};
    public static final int [] dy8 = {-1,-1,-1,0,1,1,1,0};
    public static final int [] dx8 = {-1,0,1,1,1,0,-1,-1};

    public boolean isOver(int y , int x , int h , int w) { 
        return y < 0 || x < 0 || y >= h || x >= w ;
    }

    public void swap(int [] array , int l , int r) { 
        int tmp = array[l] ; 
        array[l] = array[r] ; 
        array[r] = tmp ; 
    }

    public void swap(long [] array , int l , int r) { 
        long tmp = array[l] ; 
        array[l] = array[r] ; 
        array[r] = tmp ; 
    }
    
    public String swap(String string , int l , int r) { 
        StringBuilder m = new StringBuilder(string) ;
        m.setCharAt(l, string.charAt(r)); m.setCharAt(r, string.charAt(l)); 
        return m.toString(); 
    }

    public String binarytoString(int a , int len) { 
        String b = Integer.toBinaryString(a); 
        while(b.length() < len) b = "0" + b ; 
        return b ; 
    }

    public String binarytoString(long a , int len) { 
        String b = Long.toBinaryString(a); 
        while(b.length() < len) b = "0" + b ; 
        return b ; 
    }

    public <T extends Comparable<T>> int LowCountClosed(List<T> A , T key) { 
        return upperbound(A, key); 
    }

    public <T extends Comparable<T>> int LowCountOpen(List<T> A , T key) { 
        return lowerbound(A, key); 
    }

    public <T extends Comparable<T>> int HighCountClosed(List<T> A , T key) { 
        return A.size() - lowerbound(A, key); 
    }

    public <T extends Comparable<T>> int HighCountOpen(List<T> A , T key) { 
        return A.size() - upperbound(A, key); 
    }
    // [)
    public <T extends Comparable<T>> int CountClosedOpen(List<T> A, T a, T b) {
        return lowerbound(A, b) - lowerbound(A, a);
    }
    // []
    public <T extends Comparable<T>> int CountClosedClosed(List<T> A, T a, T b) {
        return upperbound(A, b) - lowerbound(A, a);
    }
    // (]
    public <T extends Comparable<T>> int CountOpenClosed(List<T> A, T a, T b) {
        return upperbound(A, b) - upperbound(A, a);
    }
    // ()
    public <T extends Comparable<T>> int CountOpenOpen(List<T> A, T a, T b) {
        return lowerbound(A, b) - upperbound(A, a);
    }

    private <T extends Comparable<T>> int lowerbound(List<T> A, T key) {
        int left = 0 , right = A.size();
        while (left < right) {
            int mid = (left + right) / 2;
            if (A.get(mid).compareTo(key) < 0) left = mid + 1;
            else right = mid;
        }
        return right;
    }

    private <T extends Comparable<T>> int upperbound(List<T> A, T key) { 
        int left = 0 , right = A.size();
        while (left < right) {
            int mid = (left + right) / 2;
            if (A.get(mid).compareTo(key) <= 0) left = mid + 1;
            else right = mid;
        }
        return right;
    }

    public Integer [] toInteger(int [] a) {
        return Arrays.stream(a).boxed().toArray(Integer[]::new);
    }

    public Long [] toLong(long [] a) {
        return Arrays.stream(a).boxed().toArray(Long[]::new);
    }

    public Double [] toDouble(double [] a) {
        return Arrays.stream(a).boxed().toArray(Double[]::new);
    }

    public int [] toIntArray(Collection<Integer> collection) {
        int [] array = new int[collection.size()];
        int cur = 0 ;
        for(int value : collection) array[cur++] = value ;
        return array ;
    }

    public long [] toLongArray(Collection<Long> collection) {
        long [] array = new long[collection.size()];
        int cur = 0 ;
        for(long value : collection) array[cur++] = value ;
        return array ;
    }

    public <T> List<T> ArrayList(T [] a) { 
        return Arrays.stream(a).collect(Collectors.toCollection(ArrayList::new));
    }

    public <T> HashSet<T> HashSet(T [] a) { 
        return Arrays.stream(a).collect(Collectors.toCollection(HashSet::new));
    }

    public <T> TreeSet<T> TreeSet(T [] a) { 
        return Arrays.stream(a).collect(Collectors.toCollection(TreeSet::new));
    }

    public <T> Deque<T> Deque(T [] a) { 
        return Arrays.stream(a).collect(Collectors.toCollection(ArrayDeque::new));
    }

    public <T> PriorityQueue<T> PriorityQueue(T [] a) { 
        return Arrays.stream(a).collect(Collectors.toCollection(PriorityQueue::new));
    }

    public <T> List <T> [] ArrayList(int n) { 
        @SuppressWarnings("unchecked")
        List<T> [] G = new ArrayList[n]; 
        for(int i = 0 ; i < n ; i ++ ) G[i] = new ArrayList<>();
        return G ; 
    }

    public <T> HashSet <T> [] HashSet(int n) { 
        @SuppressWarnings("unchecked")
        HashSet<T> [] G = new HashSet[n]; 
        for(int i = 0 ; i < n ; i ++ ) G[i] = new HashSet<>()  ;
        return G ; 
    }

    public <T> TreeSet <T> [] TreeSet(int n) { 
        @SuppressWarnings("unchecked")
        TreeSet<T> [] G = new TreeSet[n]; 
        for(int i = 0 ; i < n ; i ++ ) G[i] = new TreeSet<>()  ; 
        return G ; 
    }

    public <T> TreeSet <T> [] TreeSet(int n , Comparator<? super T> comparator) { 
        @SuppressWarnings("unchecked")
        TreeSet<T> [] G = new TreeSet[n]; 
        for(int i = 0 ; i < n ; i ++ ) G[i] = new TreeSet<>(comparator)  ;
        return G ; 
    }

    public <T> Deque <T> [] Deque(int n) { 
        @SuppressWarnings("unchecked")
        Deque<T> [] G = new ArrayDeque[n]; 
        for(int i = 0 ; i < n ; i ++ ) G[i] = new ArrayDeque<>(); 
        return G ; 
    }

    public <T> PriorityQueue <T> [] PriorityQueue(int n) { 
        @SuppressWarnings("unchecked")
        PriorityQueue<T> [] G = new PriorityQueue[n]; 
        for(int i = 0 ; i < n ; i ++ ) G[i] = new PriorityQueue<>(); 
        return G ; 
    }

    public <T> PriorityQueue <T> [] PriorityQueue(int n , Comparator<? super T> comparator) { 
        @SuppressWarnings("unchecked")
        PriorityQueue<T> [] G = new PriorityQueue[n]; 
        for(int i = 0 ; i < n ; i ++ ) G[i] = new PriorityQueue<>(comparator); 
        return G ; 
    }

    public <K,V> HashMap <K,V> [] HashMap(int n) { 
        @SuppressWarnings("unchecked")
        HashMap<K,V> [] G = new HashMap[n];
        for(int i = 0 ; i < n ; i ++ ) G[i] = new HashMap<>()  ;
        return G ; 
    }

    public <K,V> TreeMap <K,V> [] TreeMap(int n) { 
        @SuppressWarnings("unchecked")
        TreeMap<K,V> [] G = new TreeMap[n];
        for(int i = 0 ; i < n ; i ++ ) G[i] = new TreeMap<>()  ;
        return G ;
    }

    public <K,V> TreeMap <K,V> [] TreeMap(int n , Comparator<? super K> comparator) { 
        @SuppressWarnings("unchecked")
        TreeMap<K,V> [] G = new TreeMap[n];
        for(int i = 0 ; i < n ; i ++ ) G[i] = new TreeMap<>(comparator)  ;
        return G ; 
    }

    //---------------------------------------------------------------------------------------//

    @SuppressWarnings("unchecked")
    public <T> void print(T ... data) {
        StringBuilder pr = new StringBuilder();
        for(var temp : data) pr.append(temp+" ");
        out.println(pr.toString());
        out.flush();
    }

    public void print(String [][] array) {
        StringBuilder pr = new StringBuilder();
        for(var temp : array) {
            for(var data : temp) {
                pr.append(data+" ");
            }
            pr.append("\n");
        }
        if(0 < pr.length()) pr.deleteCharAt(pr.length() - 1);
        out.println(pr.toString());
        out.flush();
    }

    public void print(char [][] array) {
        StringBuilder pr = new StringBuilder();
        for(var temp : array) {
            for(var data : temp) {
                pr.append(data);
            }
            pr.append("\n");
        }
        if(0 < pr.length()) pr.deleteCharAt(pr.length() - 1);
        out.println(pr.toString());
        out.flush();
    }

    public <T> void print(Collection<T> collection) {
        StringBuilder pr = new StringBuilder();
        for(T data : collection) pr.append(data +" ");
        out.println(pr.toString());
        out.flush();
    }   

    public static void main(String ... args) {
        new Thread(null, new Main(), "", Runtime.getRuntime().maxMemory()).start(); 
    }

    public void run() {   
        solve();
        out.flush();
    }

}

class In {

    private final InputStream in = System.in;
    private final Scanner sc = new Scanner(System.in);
    private final byte[] buffer = new byte[1024];
    private int ptr = 0;
    private int buflen = 0;

    private boolean hasNextByte() {
        if (ptr < buflen) {
            return true;
        }else{
            ptr = 0;
            try {
                buflen = in.read(buffer);
            } catch (IOException e) {
                e.printStackTrace();
            }
            if (buflen <= 0) {
                return false;
            }
        }
        return true;
    }

    private int readByte() { 
        if (hasNextByte()) return buffer[ptr++]; else return -1;
    }

    private static boolean isPrintableChar(int c) { 
        return 33 <= c && c <= 126;
    }

    private  boolean hasNext() { 
        while(hasNextByte() && !isPrintableChar(buffer[ptr])) {
            ptr++; 
        }
        return hasNextByte();
    }
    
    public String next() {
        if (!hasNext()) throw new NoSuchElementException();
        StringBuilder sb = new StringBuilder();
        int b = readByte();
        while(isPrintableChar(b)) {
            sb.appendCodePoint(b);
            b = readByte();
        }
        return sb.toString();
    }

    public long nextLong() {
        if (!hasNext()) throw new NoSuchElementException();
        long n = 0;
        boolean minus = false;
        int b = readByte();
        if (b == '-') {
            minus = true;
            b = readByte();
        }
        if (b < '0' || '9' < b) {
            throw new NumberFormatException();
        }
        while(true){
            if ('0' <= b && b <= '9') {
                n *= 10;
                n += b - '0';
            }else if(b == -1 || !isPrintableChar(b)){
                return minus ? -n : n;
            }else{
                throw new NumberFormatException();
            }
            b = readByte();
        }
    }

    public int nextInt() {
        long nl = nextLong();
        if (nl < Integer.MIN_VALUE || nl > Integer.MAX_VALUE) throw new NumberFormatException();
        return (int) nl;
    }

    public double nextDouble() { 
        return Double.parseDouble(next());
    }

    public char nextChar() {
        return next().charAt(0);
    }

    public BigInteger nexBigInteger() {
        return sc.nextBigInteger();
    }

    public int [] nextInt(int n) {
        int [] array = new int[n];
        for(int i = 0 ; i < n ; i ++) {
            array[i] = nextInt();
        }
        return array ;
    }

    public int [][] nextInt(int n , int m) {
        int [][] array = new int[n][m];
        for(int i = 0 ; i < n ; i ++) {
            array[i] = nextInt(m);
        }
        return array ;
    }

    public long [] nextLong(int n) {
        long [] array = new long[n];
        for(int i = 0 ; i < n ; i ++) {
            array[i] = nextLong();
        }
        return array ;
    }

    public long [][] nextLong(int n , int m) {
        long [][] array = new long[n][m];
        for(int i = 0 ; i < n ; i ++) {
            array[i] = nextLong(m);
        }
        return array ;
    }

    public double [] nextDouble(int n) {
        double [] array = new double[n];
        for(int i = 0 ; i < n ; i ++) {
            array[i] = nextDouble();
        }
        return array ;
    }
    
    public String [] next(int n) {
        String [] array = new String[n];
        for(int i = 0 ; i < n ; i ++) {
            array[i] = next();
        }
        return array ;
    }

    public String [][] next(int n , int m) {
        String [][] array = new String[n][m];
        for(int i = 0 ; i < n ; i ++) {
            array[i] = next(m);
        }
        return array ;
    }

    public char [] nextChar(int n) {
        char [] array = new char[n];
        String string = next() ;
        for(int i = 0 ; i < n ; i ++) {
            array[i] = string.charAt(i);
        }
        return array ;
    }

    public char [][] nextChar(int n , int m) {
        char [][] array = new char[n][m];
        for(int i = 0 ; i < n ; i ++) {
            array[i] = nextChar(m);
        }
        return array ;
    }

}
0