結果

問題 No.1170 Never Want to Walk
ユーザー Risu_BasquiatRisu_Basquiat
提出日時 2020-08-14 22:30:04
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 4,231 bytes
コンパイル時間 437 ms
コンパイル使用メモリ 82,124 KB
実行使用メモリ 154,584 KB
最終ジャッジ日時 2024-04-18 22:21:06
合計ジャッジ時間 15,566 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 122 ms
89,752 KB
testcase_01 AC 128 ms
84,240 KB
testcase_02 AC 130 ms
84,224 KB
testcase_03 AC 132 ms
84,224 KB
testcase_04 AC 127 ms
84,228 KB
testcase_05 AC 125 ms
84,176 KB
testcase_06 AC 135 ms
84,224 KB
testcase_07 AC 133 ms
84,264 KB
testcase_08 AC 141 ms
84,176 KB
testcase_09 AC 134 ms
84,352 KB
testcase_10 AC 141 ms
83,968 KB
testcase_11 AC 131 ms
84,132 KB
testcase_12 AC 166 ms
89,264 KB
testcase_13 AC 187 ms
90,144 KB
testcase_14 AC 183 ms
90,028 KB
testcase_15 AC 172 ms
89,428 KB
testcase_16 AC 160 ms
89,472 KB
testcase_17 AC 175 ms
90,140 KB
testcase_18 AC 155 ms
89,460 KB
testcase_19 AC 169 ms
89,592 KB
testcase_20 AC 190 ms
90,312 KB
testcase_21 AC 161 ms
89,856 KB
testcase_22 AC 489 ms
89,772 KB
testcase_23 AC 282 ms
90,112 KB
testcase_24 AC 505 ms
90,068 KB
testcase_25 AC 496 ms
89,936 KB
testcase_26 AC 364 ms
90,176 KB
testcase_27 AC 946 ms
150,420 KB
testcase_28 AC 993 ms
153,856 KB
testcase_29 AC 890 ms
153,268 KB
testcase_30 AC 986 ms
151,896 KB
testcase_31 AC 937 ms
153,728 KB
testcase_32 TLE -
testcase_33 -- -
testcase_34 -- -
testcase_35 -- -
testcase_36 -- -
testcase_37 -- -
testcase_38 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
sys.setrecursionlimit(10**7) #再帰関数の上限,10**5以上の場合python
import math
from copy import copy, deepcopy
from copy import deepcopy as dcp
from operator import itemgetter
from bisect import bisect_left, bisect, bisect_right#2分探索
#bisect_left(l,x), bisect(l,x)#aはソート済みである必要あり。aの中からx未満の要素数を返す。rightだと以下
from collections import deque, defaultdict
#deque(l), pop(), append(x), popleft(), appendleft(x)
#q.rotate(n)で → にn回ローテート
from collections import Counter#文字列を個数カウント辞書に、
#S=Counter(l),S.most_common(x),S.keys(),S.values(),S.items()
from itertools import accumulate,combinations,permutations,product#累積和
#list(accumulate(l))
from heapq import heapify,heappop,heappush
#heapify(q),heappush(q,a),heappop(q) #q=heapify(q)としないこと、返り値はNone
from functools import reduce,lru_cache#pypyでもうごく
#@lru_cache(maxsize = None)#maxsizeは保存するデータ数の最大値、2**nが最も高効率
from decimal import Decimal

def input(): 
    x=sys.stdin.readline()
    return x[:-1] if x[-1]=="\n" else x
def printe(*x):print("## ",*x,file=sys.stderr)
def printl(li): _=print(*li, sep="\n") if li else None
def argsort(s, return_sorted=False): 
    inds=sorted(range(len(s)), key=lambda k: s[k])
    if return_sorted: return inds, [s[i] for i in inds]
    return inds
def alp2num(c,cap=False): return ord(c)-97 if not cap else ord(c)-65
def num2alp(i,cap=False): return chr(i+97) if not cap else chr(i+65)
def matmat(A,B):
    K,N,M=len(B),len(A),len(B[0])
    return [[sum([(A[i][k]*B[k][j]) for k in range(K)]) for j in range(M)] for i in range(N)]
def matvec(M,v):
    N,size=len(v),len(M)
    return [sum([M[i][j]*v[j] for j in range(N)]) for i in range(size)]
def T(M):
    n,m=len(M),len(M[0])
    return [[M[j][i] for j in range(n)] for i in range(m)]
def binr(x): return bin(x)[2:]
def bitcount(x): #xは64bit整数
    x= x - ((x >> 1) & 0x5555555555555555)
    x= (x & 0x3333333333333333) + ((x >> 2) & 0x3333333333333333)
    x= (x + (x >> 4)) & 0x0f0f0f0f0f0f0f0f 
    x+= (x >> 8); x+= (x >> 16); x+= (x >> 32) 
    return x & 0x7f
class unionfind:#早いunionfind,class[i]のように要素指定すると親を得ることができる
    def __init__(self, elements=None):#elementsで初期化
        if elements is None:
            elements = ()
        self.parents = {}
        self.weights = {}
        for x in elements:
            self.weights[x] = 1
            self.parents[x] = x

    def __getitem__(self, i):
        # check for previously unknown i
        if i not in self.parents:
            self.parents[i] = i
            self.weights[i] = 1
            return i
        path = [i]
        root = self.parents[i]
        while root != path[-1]:
            path.append(root)
            root = self.parents[root]
        for ancestor in path:#縮約
            self.parents[ancestor] = root
        return root

    def __iter__(self):#for parent in Class:
        return iter(self.parents)

    def union(self, *objects):#オブジェクトをすべて結合
        roots = [self[x] for x in objects]
        heaviest = max(roots, key=lambda r: self.weights[r])
        for r in roots:
            if r != heaviest:
                self.weights[heaviest] += self.weights[r]
                self.parents[r] = heaviest

    def __len__(self):
        return len(self.parents)

def main():
    mod = 1000000007
    #w.sort(key=itemgetter(1),reverse=True)  #二個目の要素で降順並び替え

    #N = int(input())
    N, a,b = map(int, input().split())
    X = tuple(map(int, input().split())) #1行ベクトル
    #L = tuple(int(input()) for i in range(N)) #改行ベクトル
    #S = tuple(tuple(map(int, input().split())) for i in range(N)) #改行行列
    
    uf=unionfind(list(range(N)))
    pr=-1
    for i,x in enumerate(X):
        l=bisect_left(X,x+a)
        r=bisect_right(X,x+b)

        if pr>l:
            uf.union(i,i-1)
        else:
            for j in range(l,min(r,N)):
                uf.union(i,j)
    for i in range(N):
        print(uf.weights[uf[i]])


if __name__ == "__main__":
    main()
0