結果

問題 No.461 三角形はいくつ？
ユーザー maspy
提出日時 2020-04-08 16:14:21
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 1,122 bytes
コンパイル時間 263 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 128,176 KB
最終ジャッジ日時 2024-07-18 14:50:41
合計ジャッジ時間 27,695 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3 WA * 1
other AC * 29 WA * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/ python3.8
import sys
read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines
import numpy as np

eps = 1e-16

N = int(readline())
X = [[0] for _ in range(3)]
m = map(int, read().split())
for p, a, b in zip(m, m, m):
    x = b / (a + b)
    X[p].append(x)

X.sort(key=len)


def solve_naive():
    import itertools
    ret = 0
    for x1, x2, x3 in itertools.product(*X):
        if 1 - eps < x1 + x2 + x3 < 1 + eps:
            continue
        if x1 + x3 > 1 + eps:
            continue
        if x1 + x2 > 1 + eps:
            continue
        if x2 + x3 > 1 + eps:
            continue
        ret += 1
    return ret


def solve(X, Y, Z):
    X.sort()
    Y.sort()
    Z.sort()
    add_XY = np.add.outer(X, Y).ravel()
    max_XY = np.maximum.outer(X, Y).ravel()
    cnt_Z = np.searchsorted(Z, 1 - max_XY + eps)
    ret = (cnt_Z * (add_XY < 1 + eps)).sum()
    # 1点で交わるものを除く
    ret -= (np.searchsorted(Z, 1 - add_XY + eps) - np.searchsorted(Z, 1 - add_XY - eps)).sum()
    return ret


X, Y, Z = map(np.array, X)

print(solve(X, Y, Z))
0