結果

問題 No.2561 みんな大好きmod 998
ユーザー poeMoon
提出日時 2023-12-02 19:25:23
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 224 ms / 4,000 ms
コード長 2,723 bytes
コンパイル時間 148 ms
コンパイル使用メモリ 82,112 KB
実行使用メモリ 77,520 KB
最終ジャッジ日時 2024-09-26 21:30:47
合計ジャッジ時間 6,373 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 44
権限があれば一括ダウンロードができます

ソースコード

diff # 

# 2と3の見間違いとかが起きるので制約のコピペ時は要確認(自戒)

N, K = map(int, input().split())
A = list(map(int, input().split()))
cnt = 0
for k1 in range(N):
    if K >= 2:
        for k2 in range(k1 + 1, N):
            if K >= 3:
                for k3 in range(k2 + 1, N):
                    if K >= 4:
                        for k4 in range(k3 + 1, N):
                            if K >= 5:
                                for k5 in range(k4 + 1, N):
                                    if K >= 6:
                                        for k6 in range(k5 + 1, N):
                                            if K >= 7:
                                                for k7 in range(k6 + 1, N):
                                                    if K >= 8:
                                                        for k8 in range(k7 + 1, N):
                                                            S = A[k1] + A[k2] + A[k3] + A[k4] + A[k5] + A[k6] + A[k7] + A[k8]
                                                            if S % 998244353 <= S % 998:
                                                                cnt = (cnt + 1) % 998
                                                    else:
                                                        S = A[k1] + A[k2] + A[k3] + A[k4] + A[k5] + A[k6] + A[k7]
                                                        if S % 998244353 <= S % 998:
                                                            cnt = (cnt + 1) % 998
                                            else:
                                                S = A[k1] + A[k2] + A[k3] + A[k4] + A[k5] + A[k6]
                                                if S % 998244353 <= S % 998:
                                                    cnt = (cnt + 1) % 998
                                    else:
                                        S = A[k1] + A[k2] + A[k3] + A[k4] + A[k5]
                                        if S % 998244353 <= S % 998:
                                            cnt = (cnt + 1) % 998
                            else:
                                S = A[k1] + A[k2] + A[k3] + A[k4]
                                if S % 998244353 <= S % 998:
                                    cnt = (cnt + 1) % 998
                    else:
                        S = A[k1] + A[k2] + A[k3]
                        if S % 998244353 <= S % 998:
                            cnt = (cnt + 1) % 998
            else:
                S = A[k1] + A[k2]
                if S % 998244353 <= S % 998:
                    cnt = (cnt + 1) % 998
    else:
        S = A[k1]
        if S % 998244353 <= S % 998:
            cnt = (cnt + 1) % 998
print(cnt)
0