結果
問題 | No.1170 Never Want to Walk |
ユーザー | persimmon-persimmon |
提出日時 | 2021-06-24 10:09:58 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 651 ms / 2,000 ms |
コード長 | 2,186 bytes |
コンパイル時間 | 380 ms |
コンパイル使用メモリ | 86,960 KB |
実行使用メモリ | 187,744 KB |
最終ジャッジ日時 | 2023-09-07 13:21:43 |
合計ジャッジ時間 | 11,794 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 70 ms
71,200 KB |
testcase_01 | AC | 71 ms
71,112 KB |
testcase_02 | AC | 70 ms
71,276 KB |
testcase_03 | AC | 70 ms
71,348 KB |
testcase_04 | AC | 70 ms
71,372 KB |
testcase_05 | AC | 68 ms
71,328 KB |
testcase_06 | AC | 70 ms
71,304 KB |
testcase_07 | AC | 72 ms
71,412 KB |
testcase_08 | AC | 73 ms
71,384 KB |
testcase_09 | AC | 75 ms
71,448 KB |
testcase_10 | AC | 72 ms
71,344 KB |
testcase_11 | AC | 71 ms
71,416 KB |
testcase_12 | AC | 90 ms
75,712 KB |
testcase_13 | AC | 95 ms
76,200 KB |
testcase_14 | AC | 95 ms
75,992 KB |
testcase_15 | AC | 90 ms
76,200 KB |
testcase_16 | AC | 92 ms
75,852 KB |
testcase_17 | AC | 94 ms
75,932 KB |
testcase_18 | AC | 94 ms
75,856 KB |
testcase_19 | AC | 92 ms
76,000 KB |
testcase_20 | AC | 94 ms
75,808 KB |
testcase_21 | AC | 90 ms
76,092 KB |
testcase_22 | AC | 97 ms
76,188 KB |
testcase_23 | AC | 100 ms
76,444 KB |
testcase_24 | AC | 96 ms
76,024 KB |
testcase_25 | AC | 102 ms
76,964 KB |
testcase_26 | AC | 102 ms
76,820 KB |
testcase_27 | AC | 577 ms
157,584 KB |
testcase_28 | AC | 551 ms
154,844 KB |
testcase_29 | AC | 651 ms
158,412 KB |
testcase_30 | AC | 575 ms
157,704 KB |
testcase_31 | AC | 571 ms
156,756 KB |
testcase_32 | AC | 544 ms
186,780 KB |
testcase_33 | AC | 554 ms
163,328 KB |
testcase_34 | AC | 559 ms
187,424 KB |
testcase_35 | AC | 530 ms
183,452 KB |
testcase_36 | AC | 552 ms
180,144 KB |
testcase_37 | AC | 524 ms
182,944 KB |
testcase_38 | AC | 533 ms
187,744 KB |
ソースコード
# UnionFind class UnionFind: def __init__(self,n): self.n=n self.par=[-1]*n # par[i]:i根ならグループiの要素数に-1をかけたもの。i根じゃないならiの親 self.rank=[0]*n # iの根を返す def find(self,i): if self.par[i]<0:return i ii=i while self.par[i]>=0: i=self.par[i] while i!=self.par[ii]: ii,self.par[ii]=self.par[ii],i return i # iとjをunionし、根頂点を返す def union(self,i,j): i,j=self.find(i),self.find(j) if i==j:return i elif self.rank[i]>self.rank[j]: # par[i]:グループiの要素数で判断してもいい self.par[i]+=self.par[j] self.par[j]=i else: self.par[j]+=self.par[i] self.par[i]=j # 深さ(rank)が同じものを併合した場合1を足す if self.rank[i]==self.rank[j]: self.rank[j]+=1 return self.find(i) # iとjが同じグループに属するか判断 def same(self,i,j): return self.find(i)==self.find(j) # ノードiが属する木のサイズを返す def size(self,i): return -self.par[self.find(i)] from bisect import bisect_left as bl, bisect_right as br def main1(n,a,b,x): # ある駅から移動可能な駅が2つの区間で表せることを使う。 # 駅v->右区間[lv0,rv0],左区間[lv1,rv1]。これの列挙はO(NlogN)でできる。尺取り法を使えばO(N)で可能 # 辺の数がO(N^2)ある。これをO(N)程度にすれば解ける。 # i,i+1の辺とi,lv0の辺のみ考える。これでO(N)程度になる。 g=[set() for _ in range(n)] hq=list(range(n)) hq.reverse() edge=[0]*n uf=UnionFind(n) for i in range(n): l=bl(x,x[i]-b) r=br(x,x[i]-a) if l==r:continue # 区間[l,r)は行き来可能 # 前回のpl,prに対して常にpl<=l,pr<=r for j in reversed(range(l+1,r)): if j-1 in g[j]:break uf.union(j-1,j) g[j].add(j-1) g[j-1].add(j) uf.union(i,l) g[i].add(l) g[l].add(i) ret=[uf.size(i) for i in range(n)] return ret if __name__=='__main__': n,a,b=map(int,input().split()) x=list(map(int,input().split())) ret1=main1(n,a,b,x) print(*ret1,sep="\n")