結果
| 問題 | No.2375 watasou and hibit's baseball | 
| コンテスト | |
| ユーザー |  | 
| 提出日時 | 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 | 
ソースコード
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)
            
            
            
        