#574198 (PyPy3) No.1361 [Zelkova 4th Tune *] QUADRUPLE-SEQUENCEの詩

提出ソース

結果

問題	No.1361 [Zelkova 4th Tune *] QUADRUPLE-SEQUENCEの詩
ユーザー	👑 Kazun
提出日時	2020-11-01 17:56:47
言語	PyPy3 (7.3.15)
結果	AC
実行時間	902 ms / 2,000 ms
コード長	4,371 bytes
コンパイル時間	451 ms
コンパイル使用メモリ	87,208 KB
実行使用メモリ	182,724 KB
最終ジャッジ日時	2023-08-27 16:15:34
合計ジャッジ時間	31,123 ms
ジャッジサーバーID （参考情報）	judge12 / judge14

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	77 ms 71,316 KB
testcase_01	AC	73 ms 71,196 KB
testcase_02	AC	73 ms 71,332 KB
testcase_03	AC	73 ms 71,456 KB
testcase_04	AC	74 ms 71,456 KB
testcase_05	AC	77 ms 71,192 KB
testcase_06	AC	72 ms 71,384 KB
testcase_07	AC	71 ms 71,392 KB
testcase_08	AC	93 ms 76,064 KB
testcase_09	AC	89 ms 75,996 KB
testcase_10	AC	93 ms 75,952 KB
testcase_11	AC	92 ms 76,216 KB
testcase_12	AC	92 ms 75,856 KB
testcase_13	AC	90 ms 76,268 KB
testcase_14	AC	87 ms 75,800 KB
testcase_15	AC	99 ms 77,020 KB
testcase_16	AC	94 ms 76,196 KB
testcase_17	AC	87 ms 76,076 KB
testcase_18	AC	132 ms 78,072 KB
testcase_19	AC	115 ms 77,144 KB
testcase_20	AC	126 ms 78,064 KB
testcase_21	AC	112 ms 78,516 KB
testcase_22	AC	120 ms 78,360 KB
testcase_23	AC	135 ms 79,340 KB
testcase_24	AC	136 ms 80,600 KB
testcase_25	AC	132 ms 79,552 KB
testcase_26	AC	139 ms 79,696 KB
testcase_27	AC	116 ms 77,888 KB
testcase_28	AC	399 ms 108,072 KB
testcase_29	AC	352 ms 102,144 KB
testcase_30	AC	336 ms 100,260 KB
testcase_31	AC	420 ms 110,532 KB
testcase_32	AC	542 ms 134,632 KB
testcase_33	AC	330 ms 98,872 KB
testcase_34	AC	452 ms 119,684 KB
testcase_35	AC	443 ms 114,312 KB
testcase_36	AC	676 ms 137,196 KB
testcase_37	AC	254 ms 99,068 KB
testcase_38	AC	357 ms 107,692 KB
testcase_39	AC	283 ms 94,280 KB
testcase_40	AC	311 ms 101,908 KB
testcase_41	AC	469 ms 114,216 KB
testcase_42	AC	662 ms 135,932 KB
testcase_43	AC	433 ms 106,840 KB
testcase_44	AC	516 ms 117,664 KB
testcase_45	AC	379 ms 111,324 KB
testcase_46	AC	535 ms 124,260 KB
testcase_47	AC	328 ms 99,148 KB
testcase_48	AC	631 ms 182,704 KB
testcase_49	AC	420 ms 182,636 KB
testcase_50	AC	613 ms 182,724 KB
testcase_51	AC	513 ms 182,668 KB
testcase_52	AC	627 ms 182,712 KB
testcase_53	AC	902 ms 157,100 KB
testcase_54	AC	886 ms 157,096 KB
testcase_55	AC	886 ms 157,128 KB
testcase_56	AC	885 ms 157,184 KB
testcase_57	AC	714 ms 156,668 KB
testcase_58	AC	606 ms 153,780 KB
testcase_59	AC	611 ms 153,896 KB
testcase_60	AC	499 ms 153,784 KB
testcase_61	AC	505 ms 154,052 KB
testcase_62	AC	502 ms 154,000 KB
testcase_63	AC	130 ms 127,088 KB
testcase_64	AC	129 ms 126,752 KB
testcase_65	AC	131 ms 126,800 KB
testcase_66	AC	135 ms 127,084 KB
testcase_67	AC	131 ms 126,812 KB
testcase_68	AC	73 ms 71,232 KB
testcase_69	AC	74 ms 71,416 KB
testcase_70	AC	72 ms 71,300 KB
testcase_71	AC	72 ms 71,248 KB
testcase_72	AC	73 ms 71,332 KB
testcase_73	AC	74 ms 71,332 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

def General_Binary_Increase_Search(L,R,cond,Integer=True,ep=1/(1<<20)):
    """条件式が単調増加であるとき,一般的な二部探索を行う.
    L:解の下限
    R:解の上限
    cond:条件(1変数関数,広義単調減少 or 広義単調減少を満たす)
    Integer:解を整数に制限するか?
    ep:Integer=Falseのとき,解の許容する誤差
    """
    if not(cond(R)):
        return False

    if Integer:
        R+=1
        while R-L>1:
            C=L+(R-L)//2
            if cond(C):
                R=C
            else:
                L=C
        return R
    else:
        while (R-L)>=ep:
            C=L+(R-L)/2
            if cond(C):
                R=C
            else:
                L=C
        return R
#================================================
def f(x): #負の時の判定
    Z=0

    I=0
    for u in U_pos:
        while (I<V_negative and u*V_neg[I]<=x):
            I+=1
        Z+=I

    I=0
    for v in V_pos:
        while (I<U_negative and v*U_neg[I]<=x):
            I+=1
        Z+=I

    return Z

def g(x): #正の時の判定
    Z=0

    I=V_positive
    for u in U_pos:
        while (I>0 and u*V_pos[I-1]>x):
            I-=1
        Z+=I

    I=0
    for v in V_neg:
        while (I<U_negative and v*U_neg[-(I+1)]<=x):
            I+=1
        Z+=I

    return Z
#================================================
# pq=x なる p in G,q in H を求める.
def h(x,G,H):
    if x==0:
        if 0 in G:
            return (0,H[0])
        else:
            return (G[0],0)

    H=set(H)
    for g in G:
        if g==0:
            continue

        if (x%g==0) and (x//g in H):
            return (g,x//g)
    return None
#================================================
#入力
K,L,M,N,S=map(int,input().split())

A=list(map(int,input().split()))
B=list(map(int,input().split()))
C=list(map(int,input().split()))
D=list(map(int,input().split()))

#================================================
#制約確認
assert 1<=K<=600,"Kが制約外(K={})".format(K)
assert 1<=L<=600,"Lが制約外(L={})".format(L)
assert 1<=M<=600,"Mが制約外(M={})".format(M)
assert 1<=N<=600,"Nが制約外(N={})".format(N)
assert 1<=S<=K*L*M*N,"Sが制約外(S={})".format(S)

assert len(A)==K,"Aの長さが違う(K={},Aの長さ={})".format(K,len(A))
assert len(B)==L,"Bの長さが違う(L={},Bの長さ={})".format(L,len(B))
assert len(C)==M,"Cの長さが違う(M={},Cの長さ={})".format(M,len(C))
assert len(D)==N,"Dの長さが違う(N={},Dの長さ={})".format(N,len(D))

A_abs_max=abs(max(A,key=lambda a:abs(a)))
B_abs_max=abs(max(B,key=lambda b:abs(b)))
C_abs_max=abs(max(C,key=lambda c:abs(c)))
D_abs_max=abs(max(D,key=lambda d:abs(d)))
assert A_abs_max<=3*10**4,"Aが制約外(max |A|={})".format(A_abs_max)
assert B_abs_max<=3*10**4,"Bが制約外(max |B|={})".format(B_abs_max)
assert C_abs_max<=3*10**4,"Cが制約外(max |C|={})".format(C_abs_max)
assert D_abs_max<=3*10**4,"Dが制約外(max |D|={})".format(D_abs_max)
#================================================
# (A,B),(C,D)の2つに分けて考える.
U=[a*b for a in A for b in B]
U.sort()
V=[c*d for c in C for d in D]
V.sort()

U_pos=[u for u in U if u>0]
U_neg=[u for u in U if u<0]

V_pos=[v for v in V if v>0]
V_neg=[v for v in V if v<0]

U_positive=len(U_pos)
U_negative=len(U_neg)
U_zero=K*L-(U_positive+U_negative)

V_positive=len(V_pos)
V_negative=len(V_neg)
V_zero=M*N-(V_positive+V_negative)

#================================================
# Eの正,ゼロ,負の個数を求める
E_positive=U_positive*V_positive+U_negative*V_negative
E_negative=U_positive*V_negative+U_negative*V_positive
E_zero=K*L*M*N-(E_positive+E_negative)

#================================================
# Jを求める.
U_abs_max=abs(max(U,key=lambda u:abs(u)))
V_abs_max=abs(max(V,key=lambda v:abs(v)))
Abs_max=U_abs_max*V_abs_max+1

if S<=E_negative: #負確定
    Ans=General_Binary_Increase_Search(-Abs_max,0,lambda x:f(x)>=S)
elif E_negative+1<=S<=E_negative+E_zero: #ゼロ確定
    Ans=0
else: #正確定
    Ans=General_Binary_Increase_Search(0,Abs_max,lambda x:g(x)>=S-(E_negative+E_zero))

#================================================
# T=abcd なる a,b,c,dを求める.
alpha,beta=h(Ans,U,V)
a,b=h(alpha,A,B)
c,d=h(beta ,C,D)

#================================================
#出力
print(Ans)
print(a,b,c,d)

yukicoder

結果

テストケース

ソースコード