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

• 問題: No.2561 みんな大好きmod 998
• ユーザー: loop0919
• 提出日時: 2023-11-05 23:28:15
• 言語: Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0)
• 結果: AC
• 実行時間: 1,977 ms / 4,000 ms
• コード長: 237 bytes
• コンパイル時間: 707 ms
• コンパイル使用メモリ: 12,288 KB
• 実行使用メモリ: 10,624 KB
• 最終ジャッジ日時: 2024-09-25 22:53:40
• 合計ジャッジ時間: 24,764 ms
• ジャッジサーバーID （参考情報）: judge1 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	29 ms; 10,496 KB
testcase_01	AC	109 ms; 10,496 KB
testcase_02	AC	29 ms; 10,496 KB
testcase_03	AC	1,906 ms; 10,496 KB
testcase_04	AC	1,837 ms; 10,496 KB
testcase_05	AC	1,977 ms; 10,624 KB
testcase_06	AC	1,788 ms; 10,496 KB
testcase_07	AC	29 ms; 10,496 KB
testcase_08	AC	29 ms; 10,496 KB
testcase_09	AC	32 ms; 10,496 KB
testcase_10	AC	29 ms; 10,624 KB
testcase_11	AC	319 ms; 10,496 KB
testcase_12	AC	29 ms; 10,624 KB
testcase_13	AC	30 ms; 10,496 KB
testcase_14	AC	30 ms; 10,496 KB
testcase_15	AC	1,905 ms; 10,624 KB
testcase_16	AC	29 ms; 10,496 KB
testcase_17	AC	30 ms; 10,496 KB
testcase_18	AC	30 ms; 10,496 KB
testcase_19	AC	1,371 ms; 10,496 KB
testcase_20	AC	30 ms; 10,496 KB
testcase_21	AC	30 ms; 10,496 KB
testcase_22	AC	29 ms; 10,624 KB
testcase_23	AC	30 ms; 10,496 KB
testcase_24	AC	29 ms; 10,496 KB
testcase_25	AC	29 ms; 10,496 KB
testcase_26	AC	417 ms; 10,624 KB
testcase_27	AC	1,714 ms; 10,496 KB
testcase_28	AC	494 ms; 10,496 KB
testcase_29	AC	161 ms; 10,496 KB
testcase_30	AC	438 ms; 10,496 KB
testcase_31	AC	890 ms; 10,496 KB
testcase_32	AC	220 ms; 10,496 KB
testcase_33	AC	160 ms; 10,496 KB
testcase_34	AC	1,698 ms; 10,496 KB
testcase_35	AC	149 ms; 10,496 KB
testcase_36	AC	147 ms; 10,624 KB
testcase_37	AC	90 ms; 10,368 KB
testcase_38	AC	330 ms; 10,624 KB
testcase_39	AC	483 ms; 10,496 KB
testcase_40	AC	497 ms; 10,496 KB
testcase_41	AC	326 ms; 10,496 KB
testcase_42	AC	572 ms; 10,496 KB
testcase_43	AC	89 ms; 10,496 KB
testcase_44	AC	241 ms; 10,368 KB
testcase_45	AC	1,404 ms; 10,496 KB
testcase_46	AC	205 ms; 10,496 KB

ソースコード

from itertools import combinations

N, K = map(int, input().split())
A = list(map(int, input().split()))

ans = 0
for cmb in combinations(A, K):
    S = sum(cmb)
    if  S % 998244353 <= S % 998:
        ans = (ans + 1) % 998

print(ans)