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

• 問題: No.2561 みんな大好きmod 998
• ユーザー: Meso Meso
• 提出日時: 2024-08-07 15:42:05
• 言語: Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0)
• 結果: AC
• 実行時間: 2,807 ms / 4,000 ms
• コード長: 254 bytes
• コンパイル時間: 278 ms
• コンパイル使用メモリ: 12,544 KB
• 実行使用メモリ: 35,072 KB
• 最終ジャッジ日時: 2024-08-07 15:42:41
• 合計ジャッジ時間: 33,506 ms
• ジャッジサーバーID （参考情報）: judge5 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	26 ms; 10,624 KB
testcase_01	AC	121 ms; 11,776 KB
testcase_02	AC	28 ms; 10,752 KB
testcase_03	AC	2,409 ms; 35,072 KB
testcase_04	AC	2,567 ms; 10,752 KB
testcase_05	AC	2,807 ms; 10,624 KB
testcase_06	AC	2,544 ms; 10,752 KB
testcase_07	AC	27 ms; 10,624 KB
testcase_08	AC	27 ms; 10,624 KB
testcase_09	AC	33 ms; 10,624 KB
testcase_10	AC	26 ms; 10,624 KB
testcase_11	AC	424 ms; 10,624 KB
testcase_12	AC	28 ms; 10,752 KB
testcase_13	AC	27 ms; 10,752 KB
testcase_14	AC	31 ms; 10,624 KB
testcase_15	AC	2,649 ms; 10,752 KB
testcase_16	AC	27 ms; 10,752 KB
testcase_17	AC	26 ms; 10,624 KB
testcase_18	AC	28 ms; 10,624 KB
testcase_19	AC	1,899 ms; 10,624 KB
testcase_20	AC	28 ms; 10,624 KB
testcase_21	AC	26 ms; 10,624 KB
testcase_22	AC	27 ms; 10,624 KB
testcase_23	AC	27 ms; 10,752 KB
testcase_24	AC	27 ms; 10,752 KB
testcase_25	AC	26 ms; 10,624 KB
testcase_26	AC	587 ms; 10,624 KB
testcase_27	AC	2,473 ms; 10,624 KB
testcase_28	AC	717 ms; 10,624 KB
testcase_29	AC	220 ms; 10,624 KB
testcase_30	AC	629 ms; 10,624 KB
testcase_31	AC	1,246 ms; 10,752 KB
testcase_32	AC	311 ms; 10,752 KB
testcase_33	AC	210 ms; 10,624 KB
testcase_34	AC	2,594 ms; 10,624 KB
testcase_35	AC	199 ms; 10,624 KB
testcase_36	AC	212 ms; 10,624 KB
testcase_37	AC	109 ms; 10,624 KB
testcase_38	AC	436 ms; 10,752 KB
testcase_39	AC	681 ms; 10,624 KB
testcase_40	AC	644 ms; 10,624 KB
testcase_41	AC	430 ms; 10,624 KB
testcase_42	AC	797 ms; 10,624 KB
testcase_43	AC	112 ms; 10,624 KB
testcase_44	AC	309 ms; 10,624 KB
testcase_45	AC	1,961 ms; 10,624 KB
testcase_46	AC	277 ms; 10,752 KB

ソースコード

import itertools

n, k = map(int, input().split())
A = list(map(int, input().split()))

ans = []
for i in itertools.combinations(A, k):
    t = sum(list(i))
    a = t % 998
    b = t % 998244353
    if a >= b:
        ans.append(1)

print(sum(ans) % 998)