結果

問題 No.2654 [Cherry 6th Tune] Re: start! (Black Sheep)
ユーザー ゼットゼット
提出日時 2024-02-24 05:00:35
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 5,705 ms / 7,000 ms
コード長 3,461 bytes
コンパイル時間 344 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 439,352 KB
最終ジャッジ日時 2024-09-29 09:50:33
合計ジャッジ時間 83,568 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 42 ms
52,992 KB
testcase_01 AC 44 ms
53,120 KB
testcase_02 AC 42 ms
53,376 KB
testcase_03 AC 2,940 ms
156,792 KB
testcase_04 AC 2,714 ms
155,600 KB
testcase_05 AC 1,582 ms
126,020 KB
testcase_06 AC 527 ms
90,112 KB
testcase_07 AC 1,308 ms
119,880 KB
testcase_08 AC 1,815 ms
131,200 KB
testcase_09 AC 2,926 ms
166,880 KB
testcase_10 AC 503 ms
89,688 KB
testcase_11 AC 2,028 ms
137,216 KB
testcase_12 AC 444 ms
87,424 KB
testcase_13 AC 1,866 ms
132,972 KB
testcase_14 AC 3,892 ms
201,244 KB
testcase_15 AC 3,616 ms
193,392 KB
testcase_16 AC 1,689 ms
126,208 KB
testcase_17 AC 3,013 ms
168,432 KB
testcase_18 AC 1,197 ms
112,228 KB
testcase_19 AC 3,640 ms
192,816 KB
testcase_20 AC 1,872 ms
133,076 KB
testcase_21 AC 2,238 ms
135,812 KB
testcase_22 AC 2,239 ms
137,876 KB
testcase_23 AC 91 ms
76,032 KB
testcase_24 AC 101 ms
76,776 KB
testcase_25 AC 105 ms
76,928 KB
testcase_26 AC 99 ms
76,436 KB
testcase_27 AC 127 ms
76,672 KB
testcase_28 AC 4,041 ms
203,576 KB
testcase_29 AC 4,036 ms
203,872 KB
testcase_30 AC 4,077 ms
203,740 KB
testcase_31 AC 3,914 ms
203,568 KB
testcase_32 AC 4,271 ms
203,692 KB
testcase_33 AC 5,705 ms
439,352 KB
testcase_34 AC 1,347 ms
216,776 KB
testcase_35 AC 1,254 ms
140,816 KB
testcase_36 AC 1,918 ms
341,344 KB
testcase_37 AC 2,589 ms
147,468 KB
testcase_38 AC 3,541 ms
386,584 KB
testcase_39 AC 40 ms
54,152 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

class segtreemin:
  def __init__(self,n):
    self.size=1
    while self.size<n:
      self.size*=2
    self.dat=[10**10]*(self.size*2)
  def update(self,x,a):
    x+=self.size
    self.dat[x]=a
    while x>1:
      x//=2
      self.dat[x]=min(self.dat[2*x],self.dat[2*x+1])
  def querry(self,u,v):
    u+=self.size
    v+=self.size
    score=10**10
    while u<v:
      if u&1:
        score=min(score,self.dat[u])
        u+=1
      if v&1:
        v-=1
        score=min(score,self.dat[v])
      u//=2
      v//=2
    return score
class segtreemax:
  def __init__(self,n):
    self.size=1
    while self.size<n:
      self.size*=2
    self.dat=[0]*(self.size*2)
  def update(self,x,a):
    x+=self.size
    self.dat[x]=a
    while x>1:
      x//=2
      self.dat[x]=max(self.dat[2*x],self.dat[2*x+1])
  def querry(self,u,v):
    u+=self.size
    v+=self.size
    score=0
    while u<v:
      if u&1:
        score=max(score,self.dat[u])
        u+=1
      if v&1:
        v-=1
        score=max(score,self.dat[v])
      u//=2
      v//=2
    return score
class segtree:
  def __init__(self,n):
    self.size=1
    while self.size<n:
      self.size*=2
    self.dat=[0]*(self.size*2)
  def update(self,x,a):
    x+=self.size
    self.dat[x]+=a
    while x>1:
      x//=2
      self.dat[x]=(self.dat[2*x]+self.dat[2*x+1])
  def querry(self,u,v):
    u+=self.size
    v+=self.size
    score=0
    while u<v:
      if u&1:
        score+=self.dat[u]
        u+=1
      if v&1:
        v-=1
        score+=self.dat[v]
      u//=2
      v//=2
    return score
N=int(input())
Zmax=segtreemax(N+1)
Zmin=segtreemin(N+1)
Zsum=segtree(N+1)
Zcount=segtree(N+1)
import sys
sys.setrecursionlimit(10**8)
import pypyjit
pypyjit.set_param('max_unroll_recursion=-1')
A=list(map(int,input().split()))
result=[10**10]*(N+1)
G=[[] for i in range(N+1)]
for i in range(N):
  a,b=map(int,input().split())
  G[a].append(b)
  G[b].append(a)
T={}
R={}
B=A[:]
B=set(B)
B=list(B)
B.sort()
for i in range(len(B)):
  R[B[i]]=i
used=[False]*(N+1)
dist=[-1]*(N+1)
dist[0]=0
def dfs(x):
  used[x]=True
  Zmax.update(R[A[x]],A[x])
  Zmin.update(R[A[x]],A[x])
  Zcount.update(R[A[x]],1)
  Zsum.update(R[A[x]],A[x])
  if not A[x] in T:
    T[A[x]]=1
  else:
    T[A[x]]+=1
  if dist[x]<=1:
    result[x]=-1
  else:
    k1=Zmax.querry(0,N+1)
    k2=Zmin.querry(0,N+1)
    if k1==k2:
      result[x]=1
    else:
      count=Zcount.querry(0,N+1)
      b=(count+1)//2
      l=0
      r=N
      while True:
        m=(l+r)//2
        c=Zcount.querry(0,m+1)
        if c>=b:
          r=m
        else:
          l=m+1
        if l==r:
          break
      c1=Zcount.querry(0,l)
      d1=Zsum.querry(0,l)
      c2=Zcount.querry(l+1,N+1)
      d2=Zsum.querry(l+1,N+1)
      result1=(c1*B[l]-d1)+((d2-k1)-(c2-1)*B[l])
      l=0
      r=N
      b=count//2+1
      while True:
        m=(l+r)//2
        c=Zcount.querry(0,m+1)
        if c>=b:
          r=m
        else:
          l=m+1
        if l==r:
          break
      c1=Zcount.querry(0,l)
      d1=Zsum.querry(0,l)
      c2=Zcount.querry(l+1,N+1)
      d2=Zsum.querry(l+1,N+1)
      result2=((c1-1)*B[l]-(d1-k2))+(d2-c2*B[l])
      result[x]=min(result1,result2)
  for y in G[x]:
    if used[y]==True:
      continue
    dist[y]=dist[x]+1
    dfs(y)
  Zcount.update(R[A[x]],-1)
  Zsum.update(R[A[x]],-A[x])
  T[A[x]]-=1
  if T[A[x]]==0:
    Zmax.update(R[A[x]],0)
    Zmin.update(R[A[x]],10**10)
dfs(0)
for i in range(1,N+1):
  print(result[i])
0