結果

問題 No.1170 Never Want to Walk
ユーザー Navier_BoltzmannNavier_Boltzmann
提出日時 2022-10-05 04:04:00
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 3,028 bytes
コンパイル時間 368 ms
コンパイル使用メモリ 82,396 KB
実行使用メモリ 1,698,484 KB
最終ジャッジ日時 2024-12-31 00:08:24
合計ジャッジ時間 25,027 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 MLE -
testcase_01 MLE -
testcase_02 AC 45 ms
57,196 KB
testcase_03 AC 46 ms
57,016 KB
testcase_04 AC 45 ms
56,636 KB
testcase_05 AC 47 ms
57,656 KB
testcase_06 AC 48 ms
56,204 KB
testcase_07 AC 46 ms
56,844 KB
testcase_08 AC 47 ms
56,336 KB
testcase_09 AC 47 ms
57,824 KB
testcase_10 AC 47 ms
57,040 KB
testcase_11 AC 47 ms
56,808 KB
testcase_12 AC 71 ms
71,688 KB
testcase_13 AC 78 ms
74,044 KB
testcase_14 AC 79 ms
74,016 KB
testcase_15 AC 71 ms
71,692 KB
testcase_16 AC 76 ms
73,260 KB
testcase_17 AC 79 ms
74,896 KB
testcase_18 AC 68 ms
70,516 KB
testcase_19 AC 78 ms
72,156 KB
testcase_20 AC 76 ms
74,688 KB
testcase_21 AC 70 ms
72,060 KB
testcase_22 AC 86 ms
77,780 KB
testcase_23 AC 89 ms
77,576 KB
testcase_24 AC 86 ms
77,668 KB
testcase_25 AC 86 ms
77,796 KB
testcase_26 AC 88 ms
78,008 KB
testcase_27 AC 420 ms
135,020 KB
testcase_28 AC 402 ms
135,412 KB
testcase_29 AC 412 ms
131,860 KB
testcase_30 AC 391 ms
135,280 KB
testcase_31 AC 397 ms
132,924 KB
testcase_32 TLE -
testcase_33 TLE -
testcase_34 TLE -
testcase_35 TLE -
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()))

INF = (1<<40)

n = math.ceil(math.sqrt(N))
M = n**2
X = X + [INF]*(M - len(X))

e = [[] for _ in range(M+n)]

for i in range(n):
    
    for j in range(n):
        
        e[M+i].append(n*i+j)
        
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
    if l < r:
    
        L = (l+n-1)//n
        R = (r+n-1)//n
        
        if L==R:
            for j in range(l,r):
                e[i].append(j)
                e[j].append(i)
        else:
        
            for j in range(l,n*L):
                e[i].append(j)
                e[j].append(i)
                
            for j in range(M+L,M+R-1):
                e[i].append(j)
                e[j].append(i)
                
            for j in range(n*(R-1),r):
                e[i].append(j)
                e[j].append(i)
            
            
    # ng = 0
    # l = 0
    # if xi - X[ng]>B:
        
    #     l = i
    #     while l-ng>1:
    #         mid = (ng+l)//2
            
    #         if xi - X[mid]>B:
                
    #             ng = mid
    #         else:
    #             l = mid
    
        
            
            
    # r = 0
    # ng = i+1
    # if xi-X[r]<A:
    #     continue
    
    # while ng-r>1:
        
    #     mid = (ng+r)//2
        
    #     if xi - X[mid] >= A:
    #         r = mid
    #     else:
    #         ng = mid
    # r += 1
    
    # if l>=r:
    #     continue
    
    
    
    # L = (l+n-1)//n
    # R = (r+n-1)//n


    # if L==R:
    #     for j in range(l,r):
    #         e[i].append(j)
    # else:
    
    #     for j in range(l,n*L):
    #         e[i].append(j)
            
    #     for j in range(M+L,M+R-1):
    #         e[i].append(j)
            
    #     for j in range(n*(R-1),r):
    #         e[i].append(j)

        
        
visited = [False]*(M+n)
ans = [1]*(M+n)

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