結果

問題 No.2561 みんな大好きmod 998
ユーザー nikoro256nikoro256
提出日時 2023-12-08 14:12:35
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 862 ms / 4,000 ms
コード長 1,121 bytes
コンパイル時間 401 ms
コンパイル使用メモリ 81,920 KB
実行使用メモリ 78,944 KB
最終ジャッジ日時 2024-09-27 02:49:07
合計ジャッジ時間 10,224 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 60 ms
67,328 KB
testcase_01 AC 89 ms
76,800 KB
testcase_02 AC 51 ms
64,000 KB
testcase_03 AC 249 ms
77,308 KB
testcase_04 AC 235 ms
77,440 KB
testcase_05 AC 237 ms
77,312 KB
testcase_06 AC 599 ms
78,248 KB
testcase_07 AC 68 ms
73,344 KB
testcase_08 AC 53 ms
63,488 KB
testcase_09 AC 86 ms
76,928 KB
testcase_10 AC 53 ms
63,232 KB
testcase_11 AC 262 ms
78,080 KB
testcase_12 AC 52 ms
63,472 KB
testcase_13 AC 68 ms
72,064 KB
testcase_14 AC 67 ms
71,168 KB
testcase_15 AC 862 ms
78,592 KB
testcase_16 AC 66 ms
72,832 KB
testcase_17 AC 51 ms
63,488 KB
testcase_18 AC 68 ms
72,704 KB
testcase_19 AC 588 ms
78,944 KB
testcase_20 AC 77 ms
76,800 KB
testcase_21 AC 50 ms
63,488 KB
testcase_22 AC 48 ms
62,976 KB
testcase_23 AC 77 ms
76,416 KB
testcase_24 AC 50 ms
63,616 KB
testcase_25 AC 50 ms
63,744 KB
testcase_26 AC 334 ms
77,952 KB
testcase_27 AC 324 ms
77,776 KB
testcase_28 AC 187 ms
77,696 KB
testcase_29 AC 121 ms
77,116 KB
testcase_30 AC 151 ms
77,204 KB
testcase_31 AC 216 ms
77,440 KB
testcase_32 AC 197 ms
77,312 KB
testcase_33 AC 123 ms
77,084 KB
testcase_34 AC 323 ms
77,696 KB
testcase_35 AC 155 ms
77,056 KB
testcase_36 AC 161 ms
77,056 KB
testcase_37 AC 118 ms
77,388 KB
testcase_38 AC 167 ms
77,312 KB
testcase_39 AC 185 ms
77,228 KB
testcase_40 AC 175 ms
77,112 KB
testcase_41 AC 167 ms
77,312 KB
testcase_42 AC 221 ms
77,056 KB
testcase_43 AC 118 ms
76,988 KB
testcase_44 AC 165 ms
77,348 KB
testcase_45 AC 254 ms
77,568 KB
testcase_46 AC 202 ms
77,440 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

N,K=map(int,input().split())
A=list(map(int,input().split()))
left=A[:N//2]
right=A[N//2:]
p=998244353
ps=998
exp2=[2**i for i in range(len(A)+1)]

def half(lis):
    lists=[[[] for _ in range(998)] for _ in range(K+1)]
    for i in range(exp2[len(lis)]):
        ans_long=0
        ans_short=0
        count=0
        for j in range(len(lis)):
            if i&1:
                count+=1
                ans_long+=lis[j]
                ans_long%=p
                ans_short+=lis[j]
                ans_short%=ps
            i>>=1
        if K<count:
            continue
        lists[count][ans_short].append(ans_long)
    for i in range(K+1):
        for j in range(998):
            lists[i][j].sort()
    return lists

import bisect
lefts=half(left)
rights=half(right)
ans=0
for i in range(K+1):
    for j in range(998):
        for t in lefts[i][j]:
            for k in range(998):
                lim=(j+k)%ps
                ans+=bisect.bisect_right(rights[K-i][k],lim-t)
                ans+=bisect.bisect_right(rights[K-i][k],lim-t+p)-bisect.bisect_left(rights[K-i][k],p-t)
                ans%=ps
print(ans)
0