結果

問題 No.1919 Many Monster Battles
ユーザー googol_S0googol_S0
提出日時 2022-04-29 22:58:51
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,883 bytes
コンパイル時間 143 ms
コンパイル使用メモリ 82,420 KB
実行使用メモリ 241,096 KB
最終ジャッジ日時 2024-06-29 04:36:06
合計ジャッジ時間 6,565 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 42 ms
61,988 KB
testcase_01 AC 42 ms
54,904 KB
testcase_02 AC 45 ms
54,372 KB
testcase_03 AC 184 ms
79,316 KB
testcase_04 AC 180 ms
79,412 KB
testcase_05 AC 183 ms
79,540 KB
testcase_06 AC 178 ms
79,840 KB
testcase_07 AC 181 ms
79,280 KB
testcase_08 AC 177 ms
79,468 KB
testcase_09 AC 180 ms
79,204 KB
testcase_10 AC 183 ms
79,788 KB
testcase_11 AC 181 ms
80,048 KB
testcase_12 AC 181 ms
78,928 KB
testcase_13 TLE -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
testcase_34 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

from copy import *
def init(N,node,A,unit,func):
  n=1
  while n<N:
    n<<=1
  for i in range(n*2-1):
    if len(node)<=i:
      node.append(deepcopy(unit))
    else:
      node[i]=deepcopy(unit)
  for i in range(len(A)):
    node[n-1+i]=deepcopy(A[i])
  for i in range(n-2,-1,-1):
    node[i]=func(node[(i<<1)+1],node[(i<<1)+2])
  node.append(func)
  node.append(unit)
  node.append(n)

def upd(node,x,a):
  y=node[-1]+x
  node[y-1]+=a
  while y>1:
    y=y>>1
    node[y-1]=node[-3](node[(y<<1)-1],node[y<<1])

def query(node,l,r):
  x,y=l,r
  z=node[-1]-1
  rr=deepcopy(node[-2])
  rl=deepcopy(node[-2])
  while True:
    if x==y:
      return node[-3](rl,rr)
    if x&1:
      rl=node[-3](rl,node[x+z])
      x+=1
    if y&1:
      rr=node[-3](node[y+z-1],rr)
    if z==0:
      return node[-3](rl,rr)
    x>>=1
    y>>=1
    z>>=1

mod=10**9+7
def comp(x):
  y=sorted(set(x))
  d=dict()
  for i in range(len(y)):
    d[y[i]]=i
  return d

def solve(A,B):
  N=len(A)
  X=[(A[i],B[i]) for i in range(N)]
  X.sort()
  Y=[X[i][0]-X[i][1] for i in range(N)]
  Z=[X[i][0]+X[i][1] for i in range(N)]
  y,z=comp(Y),comp(Z)
  ANS=0
  segy=[]
  segz=[]
  init(N,segy,[],0,lambda x,y:x+y)
  init(N,segz,[],0,lambda x,y:x+y)
  c=0
  for i in range(N):
    if i and X[i]==X[i-1]:
      c+=1
    else:
      c=0
    a,b=y[Y[i]],z[Z[i]]
    ANS=(ANS+X[i][0]*(query(segy,0,a)+query(segz,0,b)-i+c))%mod
    upd(segy,a,1)
    upd(segz,b,1)
  segy=[]
  segz=[]
  init(N,segy,[],0,lambda x,y:x+y)
  init(N,segz,[],0,lambda x,y:x+y)
  c=0
  for i in range(N-1,-1,-1):
    if i+1<N and X[i]==X[i+1]:
      c+=1
    else:
      c=0
    a,b=y[Y[i]],z[Z[i]]
    ANS=(ANS-X[i][0]*(query(segy,a+1,N)+query(segz,b+1,N)-(N-1-i)+c))%mod
    upd(segy,a,1)
    upd(segz,b,1)
  return ANS

N=int(input())
A=list(map(int,input().split()))
B=list(map(int,input().split()))
print(solve(A,B)*2%mod,solve(B,A)*2%mod)
0