結果

問題 No.1170 Never Want to Walk
ユーザー Navier_BoltzmannNavier_Boltzmann
提出日時 2022-10-08 14:32:04
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 2,783 bytes
コンパイル時間 152 ms
コンパイル使用メモリ 81,836 KB
実行使用メモリ 583,648 KB
最終ジャッジ日時 2024-06-22 17:26:56
合計ジャッジ時間 22,224 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 48 ms
56,524 KB
testcase_01 AC 45 ms
56,524 KB
testcase_02 AC 45 ms
57,880 KB
testcase_03 AC 44 ms
56,212 KB
testcase_04 AC 46 ms
56,680 KB
testcase_05 AC 46 ms
57,596 KB
testcase_06 AC 44 ms
56,316 KB
testcase_07 AC 45 ms
56,900 KB
testcase_08 AC 44 ms
58,092 KB
testcase_09 AC 46 ms
56,980 KB
testcase_10 AC 43 ms
57,128 KB
testcase_11 AC 47 ms
56,420 KB
testcase_12 AC 104 ms
77,984 KB
testcase_13 AC 107 ms
78,696 KB
testcase_14 AC 98 ms
78,120 KB
testcase_15 AC 97 ms
78,320 KB
testcase_16 AC 105 ms
77,920 KB
testcase_17 AC 105 ms
78,384 KB
testcase_18 AC 100 ms
77,984 KB
testcase_19 AC 100 ms
78,232 KB
testcase_20 AC 102 ms
78,032 KB
testcase_21 AC 99 ms
77,812 KB
testcase_22 AC 116 ms
78,084 KB
testcase_23 AC 112 ms
78,452 KB
testcase_24 AC 110 ms
78,648 KB
testcase_25 AC 104 ms
78,300 KB
testcase_26 AC 111 ms
78,728 KB
testcase_27 AC 1,351 ms
347,512 KB
testcase_28 AC 1,157 ms
348,324 KB
testcase_29 AC 1,101 ms
345,732 KB
testcase_30 AC 1,305 ms
349,072 KB
testcase_31 AC 1,106 ms
345,044 KB
testcase_32 TLE -
testcase_33 TLE -
testcase_34 TLE -
testcase_35 MLE -
testcase_36 TLE -
testcase_37 TLE -
testcase_38 MLE -
権限があれば一括ダウンロードができます

ソースコード

diff # 

import pypyjit
pypyjit.set_param('max_unroll_recursion=-1')
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