結果

問題 No.2375 watasou and hibit's baseball
ユーザー 👑 rin204rin204
提出日時 2023-07-07 21:42:43
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,009 ms / 2,000 ms
コード長 1,474 bytes
コンパイル時間 455 ms
コンパイル使用メモリ 82,496 KB
実行使用メモリ 106,144 KB
最終ジャッジ日時 2024-07-21 17:30:23
合計ジャッジ時間 19,862 ms
ジャッジサーバーID
(参考情報)
judge4 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff #

def popcount(x):
    x = x - ((x >> 1) & 0x55555555)
    x = (x & 0x33333333) + ((x >> 2) & 0x33333333)
    x = (x + (x >> 4)) & 0x0F0F0F0F
    x += x >> 8
    x += x >> 16
    return x & 0x0000003F


n, a, b = map(int, input().split())
X = [0] * n
Y = [0] * n
K = [0] * n
for i in range(n):
    X[i], Y[i], K[i] = map(int, input().split())


def f(bit, i, j):
    return (bit * n + i) * n + j


dp = [False] * ((1 << n) * n * n)
for i in range(n):
    dp[f(1 << i, i, i)] = True


def dist(i, j):
    return abs(X[i] - X[j]) + abs(Y[i] - Y[j])


for bit in range(1 << n):
    ind = []
    for i in range(n):
        if bit >> i & 1:
            ind.append(i)

    for i in ind:
        for j in ind:
            if i == j:
                continue
            if len(ind) == 2 and a <= dist(i, j):
                dp[f(bit, i, j)] |= dp[f(bit ^ (1 << j), i, i)]
            if b <= abs(K[i] - K[j]):
                dp[f(bit, i, j)] |= dp[f(bit ^ (1 << j), i, i)]

            for k in ind:
                if k == i or k == j:
                    continue

                if a <= dist(i, k) + dist(j, k):
                    dp[f(bit, j, k)] |= dp[f(bit ^ (1 << k), i, j)]
                if b <= abs(K[j] - K[k]):
                    dp[f(bit, j, k)] |= dp[f(bit ^ (1 << k), i, j)]

ans = 0
for bit in range(1 << n):
    pc = popcount(bit)
    for i in range(n):
        for j in range(n):
            if dp[f(bit, i, j)]:
                ans = max(ans, pc)

print(ans)
0