結果

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

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 115 ms
89,472 KB
testcase_01 AC 115 ms
83,968 KB
testcase_02 AC 119 ms
84,120 KB
testcase_03 AC 126 ms
84,344 KB
testcase_04 AC 117 ms
84,520 KB
testcase_05 AC 122 ms
83,968 KB
testcase_06 AC 122 ms
84,224 KB
testcase_07 AC 118 ms
84,352 KB
testcase_08 AC 121 ms
83,840 KB
testcase_09 AC 114 ms
84,224 KB
testcase_10 AC 119 ms
84,328 KB
testcase_11 AC 118 ms
84,224 KB
testcase_12 AC 153 ms
89,216 KB
testcase_13 AC 171 ms
89,856 KB
testcase_14 AC 177 ms
90,112 KB
testcase_15 AC 155 ms
89,856 KB
testcase_16 AC 157 ms
89,728 KB
testcase_17 AC 178 ms
90,548 KB
testcase_18 AC 150 ms
89,300 KB
testcase_19 AC 162 ms
89,684 KB
testcase_20 AC 189 ms
90,300 KB
testcase_21 AC 157 ms
89,472 KB
testcase_22 AC 426 ms
90,240 KB
testcase_23 AC 240 ms
90,172 KB
testcase_24 AC 424 ms
90,240 KB
testcase_25 AC 420 ms
90,136 KB
testcase_26 AC 351 ms
90,240 KB
testcase_27 AC 867 ms
150,912 KB
testcase_28 AC 916 ms
153,504 KB
testcase_29 AC 813 ms
153,084 KB
testcase_30 AC 835 ms
152,288 KB
testcase_31 AC 874 ms
153,984 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