#938491 (Python3) No.2561 みんな大好きmod 998

• 問題: No.2561 みんな大好きmod 998
• ユーザー: Theta
• 提出日時: 2023-12-15 15:46:36
• 言語: Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0)
• 結果: AC
• 実行時間: 3,479 ms / 4,000 ms
• コード長: 342 bytes
• コンパイル時間: 179 ms
• コンパイル使用メモリ: 12,672 KB
• 実行使用メモリ: 10,880 KB
• 最終ジャッジ日時: 2024-09-27 06:27:55
• 合計ジャッジ時間: 41,032 ms
• ジャッジサーバーID （参考情報）: judge3 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	28 ms; 10,752 KB
testcase_01	AC	167 ms; 10,752 KB
testcase_02	AC	28 ms; 10,752 KB
testcase_03	AC	3,242 ms; 10,752 KB
testcase_04	AC	3,287 ms; 10,752 KB
testcase_05	AC	3,479 ms; 10,752 KB
testcase_06	AC	3,274 ms; 10,752 KB
testcase_07	AC	27 ms; 10,752 KB
testcase_08	AC	27 ms; 10,752 KB
testcase_09	AC	32 ms; 10,752 KB
testcase_10	AC	27 ms; 10,752 KB
testcase_11	AC	531 ms; 10,752 KB
testcase_12	AC	28 ms; 10,880 KB
testcase_13	AC	27 ms; 10,752 KB
testcase_14	AC	27 ms; 10,880 KB
testcase_15	AC	3,471 ms; 10,752 KB
testcase_16	AC	29 ms; 10,752 KB
testcase_17	AC	27 ms; 10,624 KB
testcase_18	AC	27 ms; 10,752 KB
testcase_19	AC	2,441 ms; 10,752 KB
testcase_20	AC	29 ms; 10,752 KB
testcase_21	AC	27 ms; 10,752 KB
testcase_22	AC	28 ms; 10,752 KB
testcase_23	AC	28 ms; 10,752 KB
testcase_24	AC	28 ms; 10,880 KB
testcase_25	AC	28 ms; 10,752 KB
testcase_26	AC	724 ms; 10,752 KB
testcase_27	AC	3,204 ms; 10,752 KB
testcase_28	AC	915 ms; 10,880 KB
testcase_29	AC	276 ms; 10,752 KB
testcase_30	AC	798 ms; 10,752 KB
testcase_31	AC	1,616 ms; 10,752 KB
testcase_32	AC	375 ms; 10,752 KB
testcase_33	AC	262 ms; 10,752 KB
testcase_34	AC	3,228 ms; 10,752 KB
testcase_35	AC	242 ms; 10,752 KB
testcase_36	AC	237 ms; 10,752 KB
testcase_37	AC	129 ms; 10,880 KB
testcase_38	AC	532 ms; 10,752 KB
testcase_39	AC	831 ms; 10,752 KB
testcase_40	AC	853 ms; 10,752 KB
testcase_41	AC	549 ms; 10,880 KB
testcase_42	AC	984 ms; 10,752 KB
testcase_43	AC	135 ms; 10,752 KB
testcase_44	AC	409 ms; 10,752 KB
testcase_45	AC	2,495 ms; 10,752 KB
testcase_46	AC	322 ms; 10,880 KB

ソースコード

from itertools import combinations


def main():
    N, K = map(int, input().split())
    A = list(map(int, input().split()))
    ctr = 0
    for idxs in combinations(range(N), r=K):
        S = sum(A[idx] for idx in idxs)
        if S % 998 >= S % 998244353:
            ctr += 1
    print(ctr % 998)


if __name__ == "__main__":
    main()