結果

問題 No.1170 Never Want to Walk
ユーザー Navier_BoltzmannNavier_Boltzmann
提出日時 2022-10-08 14:31:25
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 2,723 bytes
コンパイル時間 237 ms
コンパイル使用メモリ 82,356 KB
実行使用メモリ 627,944 KB
最終ジャッジ日時 2024-06-22 17:25:41
合計ジャッジ時間 23,700 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 48 ms
63,524 KB
testcase_01 AC 51 ms
56,192 KB
testcase_02 AC 49 ms
56,448 KB
testcase_03 AC 49 ms
56,064 KB
testcase_04 AC 48 ms
56,064 KB
testcase_05 AC 47 ms
55,808 KB
testcase_06 AC 48 ms
56,192 KB
testcase_07 AC 49 ms
56,320 KB
testcase_08 AC 51 ms
56,320 KB
testcase_09 AC 48 ms
56,192 KB
testcase_10 AC 49 ms
56,192 KB
testcase_11 AC 48 ms
56,252 KB
testcase_12 AC 106 ms
78,436 KB
testcase_13 AC 107 ms
78,304 KB
testcase_14 AC 107 ms
78,116 KB
testcase_15 AC 105 ms
77,996 KB
testcase_16 AC 113 ms
78,308 KB
testcase_17 AC 119 ms
78,080 KB
testcase_18 AC 104 ms
78,464 KB
testcase_19 AC 106 ms
78,080 KB
testcase_20 AC 112 ms
78,188 KB
testcase_21 AC 106 ms
78,044 KB
testcase_22 AC 126 ms
78,464 KB
testcase_23 AC 125 ms
78,696 KB
testcase_24 AC 128 ms
78,664 KB
testcase_25 AC 125 ms
78,592 KB
testcase_26 AC 132 ms
78,720 KB
testcase_27 AC 1,460 ms
347,176 KB
testcase_28 AC 1,261 ms
346,984 KB
testcase_29 AC 1,266 ms
346,424 KB
testcase_30 AC 1,385 ms
348,340 KB
testcase_31 AC 1,264 ms
346,528 KB
testcase_32 TLE -
testcase_33 TLE -
testcase_34 TLE -
testcase_35 MLE -
testcase_36 -- -
testcase_37 -- -
testcase_38 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import *
from itertools import *
from functools import *
from heapq import *
import sys,math
input = sys.stdin.readline

N,A,B = map(int,input().split())
X = list(map(int,input().split()))
Q = 2*N
# 頂点数N,追加エッジの個数Q
# 頂点iのインデックスはn+iになる(エッジを追加するときは気にする必要はない)
n = 1
while n < N:
    n *= 2
N3 = 3*n + 1
e = [[] for _ in range(N3+2*Q)]
for i in range(n-1,0,-1):
    
    a = 2*i
    b = 2*i+1
    e[i].append((a,0))
    e[i].append((b,0))
    if n<=a<2*n:
        e[a].append((i+2*n,0))
    else:
        e[a+2*n].append((i+2*n,0))
        
        
    if n<=b<2*n:
        e[b].append((i+2*n,0))
    else:
        e[b+2*n].append((i+2*n,0))

q = 0
def add(l1,r1,l2,r2,c):
    ##[l1,r1)から[l2,r2)にコストcの辺を張る
    global q,e
    source = []
    l1 += n
    r1 += n
    while l1<r1:
        if l1&1:
            if n<=l1<2*n:
                source.append(l1)
            else:
                source.append(l1+2*n)
            l1 += 1
        if r1&1:
            if n<=r1-1<2*n:
                source.append(r1-1)
            else:
                source.append(r1+2*n-1)
        l1 >>= 1
        r1 >>= 1
    
    target = []
    l2 += n
    r2 += n
    while l2<r2:
        if l2&1:
            target.append(l2)
            l2 += 1
        if r2&1:
            target.append(r2-1)
        l2 >>= 1
        r2 >>= 1
    
    for s in source:
        e[s].append((N3+q,0))
    for t in target:
        e[N3+q+1].append((t,0))
    
    e[N3+q].append((N3+q+1,c))
    q += 2
    return
        
for i in range(N):
    
    xi = X[i]
    l = N
    ng = i
    while l-ng>1:
        
        mid = (l+ng)//2
        
        if X[mid]>=A+xi:
            
            l = mid
        else:
            
            ng = mid
            
            
    r = i
    ng = N
    while ng-r>1:
        mid = (ng+r)//2
        
        if X[mid]<=xi+B:
            r = mid
        else:
            ng = mid
    r += 1
    add(i,i+1,l,r,1)
    add(l,r,i,i+1,1)
            

        
visited = [False]*(N3+2*Q)
ans = [1]*(N3+2*Q)

for i in range(N):
    
    if visited[i+n]:
        continue
    
    visited[i+n]=True
    v = deque()
    v.append(i+n)
    target = deque()
    target.append(i+n)
    cnt = 1
    while v:
        
        x = v.popleft()
        
        for ix,_ in e[x]:
            
            if visited[ix]:
                continue
            visited[ix] = True
            if n<=ix<n+N:
                cnt += 1
                target.append(ix)
            v.append(ix)
    
    while target:
        y = target.pop()
        ans[y] = cnt
        
        
print(*ans[n:n+N],sep='\n')
0