結果
問題 | No.703 ゴミ拾い Easy |
ユーザー | vwxyz |
提出日時 | 2024-11-20 16:24:35 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 15,304 bytes |
コンパイル時間 | 279 ms |
コンパイル使用メモリ | 82,388 KB |
実行使用メモリ | 184,788 KB |
最終ジャッジ日時 | 2024-11-20 16:25:06 |
合計ジャッジ時間 | 25,866 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 65 ms
69,596 KB |
testcase_01 | AC | 64 ms
70,008 KB |
testcase_02 | AC | 63 ms
69,544 KB |
testcase_03 | AC | 65 ms
70,392 KB |
testcase_04 | AC | 64 ms
70,908 KB |
testcase_05 | AC | 66 ms
69,724 KB |
testcase_06 | AC | 67 ms
69,304 KB |
testcase_07 | AC | 64 ms
69,732 KB |
testcase_08 | AC | 65 ms
70,440 KB |
testcase_09 | AC | 67 ms
69,308 KB |
testcase_10 | AC | 64 ms
70,344 KB |
testcase_11 | AC | 75 ms
70,012 KB |
testcase_12 | AC | 64 ms
70,204 KB |
testcase_13 | AC | 65 ms
70,688 KB |
testcase_14 | AC | 144 ms
78,988 KB |
testcase_15 | AC | 154 ms
79,112 KB |
testcase_16 | AC | 146 ms
79,392 KB |
testcase_17 | AC | 155 ms
79,096 KB |
testcase_18 | AC | 148 ms
79,064 KB |
testcase_19 | AC | 143 ms
78,912 KB |
testcase_20 | AC | 167 ms
79,252 KB |
testcase_21 | AC | 153 ms
79,120 KB |
testcase_22 | AC | 148 ms
79,008 KB |
testcase_23 | AC | 152 ms
78,988 KB |
testcase_24 | AC | 1,434 ms
167,536 KB |
testcase_25 | TLE | - |
testcase_26 | AC | 1,484 ms
167,512 KB |
testcase_27 | AC | 1,404 ms
167,540 KB |
testcase_28 | TLE | - |
testcase_29 | TLE | - |
testcase_30 | AC | 1,439 ms
167,616 KB |
testcase_31 | TLE | - |
testcase_32 | AC | 1,437 ms
167,888 KB |
testcase_33 | AC | 1,422 ms
167,784 KB |
testcase_34 | AC | 331 ms
184,376 KB |
testcase_35 | AC | 319 ms
184,596 KB |
testcase_36 | AC | 353 ms
184,336 KB |
testcase_37 | AC | 317 ms
184,432 KB |
testcase_38 | AC | 329 ms
184,536 KB |
testcase_39 | AC | 323 ms
184,408 KB |
testcase_40 | AC | 360 ms
184,344 KB |
testcase_41 | AC | 318 ms
184,436 KB |
testcase_42 | AC | 326 ms
184,788 KB |
testcase_43 | AC | 325 ms
184,248 KB |
testcase_44 | AC | 76 ms
70,444 KB |
testcase_45 | AC | 64 ms
69,424 KB |
testcase_46 | AC | 245 ms
163,036 KB |
testcase_47 | AC | 324 ms
167,052 KB |
testcase_48 | AC | 150 ms
78,904 KB |
testcase_49 | AC | 146 ms
79,200 KB |
ソースコード
import sys readline=sys.stdin.readline from collections import deque,defaultdict import math from bisect import bisect_left, bisect_right from typing import Generic, Iterable, Iterator, TypeVar, Optional, List T = TypeVar('T') class AVL_Node_dict: """ノード Attributes: key (any): ノードのキー。比較可能なものであれば良い。(1, 4)などタプルも可。 val (any): ノードの値。 left (Node): 左の子ノード。 right (Node): 右の子ノード。 bias (int): 平衡度。(左部分木の高さ)-(右部分木の高さ)。 size (int): 自分を根とする部分木の大きさ """ def __init__(self,parent,key,value): self.parent=parent self.key=key self.value=value self.left=None self.right=None self.bias=0 self.size=1 class AVLTree_dict: def __init__(self): self.root=None def Rotate_Left(self,node): node_right=node.right node_right.size=node.size node.size-=1 if node_right.right!=None: node.size-=node_right.right.size if node_right.bias==-1: node_right.bias=0 node.bias=0 else: #assert node_right.bias==0 node_right.bias=1 node.bias=-1 node.right=node_right.left node_right.left=node return node_right def Rotate_Right(self,node): node_left=node.left node_left.size=node.size node.size-=1 if node_left.left!=None: node.size-=node_left.left.size if node_left.bias==1: node_left.bias=0 node.bias=0 else: #assert node_left.bias==0 node_left.bias=-1 node.bias=1 node.left=node_left.right node_left.right=node return node_left def Rotate_Left_Right(self,node): node_left=node.left node_left_right=node_left.right #assert node.bias==2 #assert node_left.bias==-1 #assert node_left_right.bias in (-1,0,1) node_left_right.size=node.size node.size-=node_left.size if node_left_right.right!=None: node.size+=node_left_right.right.size node_left.size-=1 if node_left_right.right!=None: node_left.size-=node_left_right.right.size node_left.right=node_left_right.left node_left_right.left=node_left node.left=node_left_right.right node_left_right.right=node self.Update_Bias_Double(node_left_right) return node_left_right def Rotate_Right_Left(self,node): node_right=node.right node_right_left=node_right.left #assert node.bias==-2 #assert node_right.bias==1 #assert node_right_left.bias in (-1,0,1) node_right_left.size=node.size node.size-=node_right.size if node_right_left.left!=None: node.size+=node_right_left.left.size node_right.size-=1 if node_right_left.left!=None: node_right.size-=node_right_left.left.size node_right.left=node_right_left.right node_right_left.right=node_right node.right=node_right_left.left node_right_left.left=node self.Update_Bias_Double(node_right_left) return node_right_left def Update_Bias_Double(self,node): #assert node.right.bias*node.left.bias==-2 #assert node.right.bias>0 if node.bias==1: node.right.bias=-1 node.left.bias=0 elif node.bias==-1: node.right.bias=0 node.left.bias=1 else: node.right.bias=0 node.left.bias=0 node.bias=0 def __getitem__(self,key): v=self.root while v!=None: if key<v.key: v=v.left elif v.key<key: v=v.right else: return v.value return None def __setitem__(self,key,value): if self.root==None: self.root=AVL_Node_dict(None,key,value) return v=self.root stack=[] while v!=None: if key<v.key: stack.append((v,1)) v=v.left elif v.key<key: stack.append((v,-1)) v=v.right elif v.key==key: v.value=value return p,direction=stack[-1] if direction==1: p.left=AVL_Node_dict(p,key,value) else: p.right=AVL_Node_dict(p,key,value) while stack: v,direction=stack.pop() v.bias+=direction v.size+=1 vv=None if v.bias==2: if v.left.bias==-1: vv=self.Rotate_Left_Right(v) else: vv=self.Rotate_Right(v) #assert vv!=None break if v.bias==-2: if v.right.bias==1: vv=self.Rotate_Right_Left(v) else: vv=self.Rotate_Left(v) #assert vv!=None break if v.bias==0: break if vv!=None: if len(stack)==0: self.root=vv return p,direction=stack.pop() p.size+=1 if direction==1: p.left=vv else: p.right=vv while stack: p,direction=stack.pop() p.size+=1 def __delitem__(self,key): v=self.root stack=[] while v!=None: if key<v.key: stack.append((v,1)) v=v.left elif v.key<key: stack.append((v,-1)) v=v.right else: break else: return False if v.left!=None: stack.append((v,1)) lmax=v.left while lmax.right!=None: stack.append((lmax,-1)) lmax=lmax.right v.key=lmax.key v.value=lmax.value v=lmax c=v.right if v.left==None else v.left if stack: p,direction=stack[-1] if direction==1: p.left=c else: p.right=c else: self.root=c return True while stack: pp=None p,direction=stack.pop() p.bias-=direction p.size-=1 if p.bias==2: if p.left.bias==-1: pp=self.Rotate_Left_Right(p) else: pp=self.Rotate_Right(p) elif p.bias==-2: if p.right.bias==1: pp=self.Rotate_Right_Left(p) else: pp=self.Rotate_Left(p) elif p.bias!=0: break if pp!=None: if len(stack)==0: self.root=pp return True p,direction=stack[-1] if direction==1: p.left=pp else: p.right=pp if pp.bias!=0: break while stack: p,direction=stack.pop() p.size-=1 return True def __contains__(self,key): v=self.root while v!=None: if key<v.key: v=v.left elif v.key<key: v=v.right else: return True return False def Bisect_Right(self,key): retu=None v=self.root while v!=None: if v.key>key: if retu==None or retu[0]>v.key: retu=(v.key,v.value) v=v.left else: v=v.right return retu def Bisect_Left(self,key): retu=None v=self.root while v!=None: if v.key<key: if retu==None or retu[0]<v.key: retu=(v.key,v.value) v=v.right else: v=v.left return retu def Find_Kth_Element(self,K): v=self.root s=0 while v!=None: t=s+v.left.size if v.left!=None else s if t==K: return v.key,v.value elif t<K: s=t+1 v=v.right else: v=v.left return None def keys(self): stack=[(self.root,True)] while stack: node,subtree=stack.pop() if subtree: if node.right!=None: stack.append((node.right,True)) stack.append((node,False)) if node.left!=None: stack.append((node.left,True)) else: yield node.key def values(self): stack=[(self.root,True)] while stack: node,subtree=stack.pop() if subtree: if node.right!=None: stack.append((node.right,True)) stack.append((node,False)) if node.left!=None: stack.append((node.left,True)) else: yield node.value def items(self): stack=[(self.root,True)] while stack: node,subtree=stack.pop() if subtree: if node.right!=None: stack.append((node.right,True)) stack.append((node,False)) if node.left!=None: stack.append((node.left,True)) else: yield (node.key,node.value) def __bool__(self): return self.root!=None def __len__(self): return 0 if self.root==None else self.root.size def __iter__(self): return iter(self.keys()) def __str__(self): if self.root==None: retu="{}" else: retu="{"+", ".join(f"{r}: {m}" for r,m in self.items())+"}" return retu class Convex_Hull_Trick: def __init__(self,avl=False): self.avl=avl if self.avl: self.lines=AVLTree_dict() else: self.lines_A=SortedSet() self.lines_B={} def is_removed(self,line0,line1,line2): if line0==None: return False if line2==None: return False a0,b0=line0 a1,b1=line1 a2,b2=line2 return (a1-a0)*(b2-b1)<=(b1-b0)*(a2-a1) def add_line(self,a,b): if self.avl: bb=self.lines[a] else: if a in self.lines_B: bb=self.lines_B[a] else: bb=None if bb!=None and bb<=b: return if self.avl: line0=self.lines.Bisect_Left(a) else: aa=self.lines_A.lt(a) if aa==None: line0=None else: line0=(aa,self.lines_B[aa]) line1=(a,b) if self.avl: line2=self.lines.Bisect_Right(a) else: aa=self.lines_A.gt(a) if aa==None: line2=None else: line2=(aa,self.lines_B[aa]) if self.is_removed(line0,line1,line2): return if self.avl: self.lines[a]=b else: self.lines_A.add(a) self.lines_B[a]=b line2=(a,b) if self.avl: line1=self.lines.Bisect_Left(line2[0]) line0=None if line1==None else self.lines.Bisect_Left(line1[0]) else: aa=self.lines_A.lt(line2[0]) if aa==None: line1=None else: line1=(aa,self.lines_B[aa]) if line1==None: line0=None else: aa=self.lines_A.lt(line1[0]) if aa==None: line0=None else: line0=(aa,self.lines_B[aa]) while self.is_removed(line0,line1,line2): if self.avl: del self.lines[line1[0]] else: assert self.lines_A.discard(line1[0]) del self.lines_B[line1[0]] line1=line0 if self.avl: line0=None if line1==None else self.lines.Bisect_Left(line1[0]) else: aa=self.lines_A.lt(line1[0]) if aa==None: line0=None else: line0=(aa,self.lines_B[aa]) line0=(a,b) if self.avl: line1=self.lines.Bisect_Right(line0[0]) line2=None if line1==None else self.lines.Bisect_Right(line1[0]) else: aa=self.lines_A.gt(line0[0]) if aa==None: line1=None else: line1=(aa,self.lines_B[aa]) if line1==None: line2=None else: aa=self.lines_A.gt(line1[0]) if aa==None: line2=None else: line2=(aa,self.lines_B[aa]) while self.is_removed(line0,line1,line2): if self.avl: del self.lines[line1[0]] else: assert self.lines_A.discard(line1[0]) del self.lines_B[line1[0]] line1=line2 if self.avl: line2=None if line1==None else self.lines.Bisect_Right(line1[0]) else: aa=self.lines_A.gt(line1[0]) if aa==None: line2=None else: line2=(aa,self.lines_B[aa]) def __call__(self,x): if self.avl: size=len(self.lines) else: size=len(self.lines_A) if not size: return None ok,ng=-1,size-1 while ng-ok>1: mid=(ok+ng)//2 if self.avl: a0,b0=self.lines.Find_Kth_Element(mid) a1,b1=self.lines.Find_Kth_Element(mid+1) else: a0=self.lines_A[mid] b0=self.lines_B[a0] a1=self.lines_A[mid+1] b1=self.lines_B[a1] if a0*x+b0>a1*x+b1: ok=mid else: ng=mid if self.avl: a,b=self.lines.Find_Kth_Element(ok+1) else: a=self.lines_A[ok+1] b=self.lines_B[a] return a*x+b def __getitem__(self,a): if self.avl: return self.lines[a] else: if a in self.lines_A: return self.lines_B[a] else: return None def __setitem__(self,a,b): self.add_line(a,b) N=int(readline()) A=list(map(int,readline().split())) X=list(map(int,readline().split())) Y=list(map(int,readline().split())) dp=Convex_Hull_Trick(avl=True) cost=0 for i in range(N): dp.add_line(-2*X[i],cost+X[i]**2+Y[i]**2) cost=dp(A[i])+A[i]**2 ans=cost print(ans)