結果
問題 |
No.1170 Never Want to Walk
|
ユーザー |
![]() |
提出日時 | 2025-04-16 15:37:48 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,282 bytes |
コンパイル時間 | 549 ms |
コンパイル使用メモリ | 81,684 KB |
実行使用メモリ | 123,160 KB |
最終ジャッジ日時 | 2025-04-16 15:43:00 |
合計ジャッジ時間 | 7,507 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 13 WA * 24 |
ソースコード
import sys from bisect import bisect_left, bisect_right def main(): input = sys.stdin.read().split() idx = 0 N = int(input[idx]); idx +=1 A = int(input[idx]); idx +=1 B = int(input[idx]); idx +=1 x = list(map(int, input[idx:idx+N])) parent = list(range(N)) size = [1]*N def find(u): while parent[u] != u: parent[u] = parent[parent[u]] u = parent[u] return u def union(u, v): u_root = find(u) v_root = find(v) if u_root == v_root: return if size[u_root] < size[v_root]: u_root, v_root = v_root, u_root parent[v_root] = u_root size[u_root] += size[v_root] for i in range(N): xi = x[i] left = xi + A right_val = xi + B l = bisect_left(x, left) r = bisect_right(x, right_val) - 1 if l > r: continue # Connect to l and r if l < N: union(i, l) if r >=0 and r != l: union(i, r) # After all unions, find the root for each i and output the size result = [] for i in range(N): root = find(i) result.append(str(size[root])) print('\n'.join(result)) if __name__ == '__main__': main()