結果

問題 No.783 門松計画
コンテスト
ユーザー titia
提出日時 2024-02-03 03:20:16
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,353 bytes
コンパイル時間 179 ms
コンパイル使用メモリ 82,316 KB
実行使用メモリ 67,776 KB
最終ジャッジ日時 2024-09-28 10:44:44
合計ジャッジ時間 2,316 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 8 WA * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

N,C=map(int,input().split())
L=list(map(int,input().split()))
W=list(map(int,input().split()))

from functools import lru_cache
@lru_cache(maxsize=None)
def calc(a,b,cost):
    ANS=0
    if a==b==0:
        for i in range(N):
            if W[i]<=cost:
                ANS=max(ANS,L[i]+calc(0,L[i],cost-W[i]))

        return ANS

    if a==0:
        for i in range(N):
            if L[i]!=b and W[i]<=cost:
                ANS=max(ANS,L[i]+calc(b,L[i],cost-W[i]))

        return ANS

    if a<b:
        for i in range(N):
            if W[i]<=cost and L[i]<b and L[i]!=a:
                ANS=max(ANS,L[i]+calc(b,L[i],cost-W[i]))
        return ANS
    else:
        for i in range(N):
            if W[i]<=cost and L[i]>b and L[i]!=a:
                ANS=max(ANS,L[i]+calc(b,L[i],cost-W[i]))
        return ANS

if len(set(sorted(L)))<=2:
    print(0)
else:
    ANS=0

    for i in range(N):
        for j in range(i+1,N):
            for k in range(j+1,N):
                if L[i]!=L[j] and L[j]!=L[k] and L[k]!=L[i] and W[i]+W[j]+W[k]<=C:
                    if L[i]<L[j] and L[j]>L[k]:
                        ANS=max(ANS,L[i]+L[j]+L[k]+calc(L[j],L[k],C-W[i]-W[j]-W[k]))
                        
                    if L[i]>L[j] and L[j]<L[k]:
                        ANS=max(ANS,L[i]+L[j]+L[k]+calc(L[j],L[k],C-W[i]-W[j]-W[k]))

    print(ANS)
0