結果

問題 No.2594 Mix shake!!
ユーザー hitonanode
提出日時 2023-12-20 12:15:08
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 2,021 bytes
コンパイル時間 323 ms
コンパイル使用メモリ 13,056 KB
実行使用メモリ 11,904 KB
最終ジャッジ日時 2024-09-27 09:46:17
合計ジャッジ時間 6,588 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 66 WA * 18 RE * 1
権限があれば一括ダウンロードができます

ソースコード

diff #

# バグりまくり
from fractions import Fraction


def to_slope(vec: list[tuple[Fraction, int, int, int]]) -> list[tuple[int, int]]:
    x, y = 0, 0
    ret: list[tuple[int, int]] = []
    for _, a, b, _ in vec:
        x += a
        y += b
        ret.append((x, y))
    return ret


def solve() -> bool:

    n = int(input())

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

    AB = [(Fraction(b, a), a, b, i) for i, (a, b) in enumerate(zip(A, B))]
    CD = [(Fraction(d, c), c, d, i) for i, (c, d) in enumerate(zip(C, D))]

    AB.sort()
    CD.sort()

    lb_xys = to_slope(AB)  # 初期状態の y = f(x) の折れ線
    ub_xys = to_slope(CD)  # 所望の最終状態の y = f(x) の折れ線

    # 最終状態の折れ線が初期状態の折れ線の下側を通っていたら明らかに No
    for (x0, y0), (x1, y1) in zip(lb_xys[:-1], lb_xys[1:], strict=True):
        for x, y in ub_xys:
            if (x1 - x0) * (y - y0) - (y1 - y0) * (x - x0) < 0:
                return False

    # 初期状態と最終状態で濃度の大小関係が一致していたら Yes
    if [i for _, _, _, i in AB] == [i for _, _, _, i in CD]:
        return True

    # n - 1 本以下の線分で (0, 0) から右上まで到達できたら Yes
    vx, vy = Fraction(0, 1), Fraction(0, 1)  # 現在位置
    for i in range(n - 1):
        # (vx, vy) から出て (px, py) を通る直線を考える
        px, py = ub_xys[i]

        # if px <= vx or (1 - vx) * (py - vy) - (1 - vy) * (px - vx) >= 0:
        if px <= vx:
            return True

        # y = a1 x + b1
        a1 = (py - vy) / (px - vx)
        b1 = vy - a1 * vx

        # y = a2 x + b2
        ax, ay = lb_xys[i]
        bx, by = lb_xys[i + 1]

        a2 = Fraction(by - ay, bx - ax)
        b2 = ay - a2 * ax

        vx = (b2 - b1) / (a1 - a2)
        vy = a1 * vx + b1

    return False

print('Yes' if solve() else 'No')
0