Problem p02821 (python)

Fixed submission:
import sys
input = sys.stdin.readline
# 昇順ソートされた配列の挿入位置indexを返す O(logN)
from bisect import bisect_left

def isOver(a,n,x):
	cnt = 0
	for i in range(n):
		cnt += n-bisect_left(a,x-a[i])
	return cnt

def calX(a,n,m,l,r):
	while r-l>1:
		# x: 握手で得たい最低パワー値
		x = (l+r)//2
		if isOver(a,n,x) >= m:
			l = x
		else: #握手する限界回数すら満たせないほど大きいxなら
			r = x 
	return l

n,m = map(int, input().split())
a = list(map(int, input().split()))

#累積和の下準備
a.sort()
s = [0]*(n+1)
for i in range(n):
	s[i+1] = s[i]+a[i]

x = calX(a,n,m,a[0]*2,a[n-1]*2)

out = 0
for i in range(n):
	idx = bisect_left(a,x-a[i])
	out += s[n]-s[idx]+a[i]*(n-idx)

out -= (isOver(a,n,x)-m)*x

print(out)

Buggy submission (verdict: Runtime Error):
import sys
input = sys.stdin.readline
# 昇順ソートされた配列の挿入位置indexを返す O(logN)
from bisect import bisect_left

def isOver(A,x):
	cnt = 0
	for a in A:
		cnt += n-bisect_left(A,x-a)
	return cnt

def calOut(A,s,x):
	for a in A:
		idx = bisect_left(A,x-a)
		out += s[n]-s[idx]+a*(n-idx)
	return out - (isOver(A,n,x)-m)*x

def calX(A,m,l,r):
	while r-l>1:
		# x: 握手で得たい最低パワー値
		x = (l+r)//2
		if isOver(A,x) >= m:
			l = x
		else: #握手する限界回数すら満たせないほど大きいxなら
			r = x 
	return l

n,m = map(int, input().split())
a = list(map(int, input().split()))

#累積和の下準備
a.sort()
s = [0]*(n+1)
for i in range(n):
	s[i+1] = s[i]+a[i]

x = calX(a,m,a[0]*2,a[n-1]*2)

print(calOut(a,s,x))

Buggy submission (verdict: Runtime Error):
import sys
input = sys.stdin.readline

n,m = (int(x) for x in input().split())
a = list(map(int, input().split()))

b = [0]*(n**2)
idx = 0

for i in range(n):
	for j in range(n):
		b[i*n+j] = a[i]+a[j]

b.sort(reverse=True)

print(sum(b[:m]))

Buggy submission (verdict: Runtime Error):
import sys
input = sys.stdin.readline
import numpy as np

n,m = (int(x) for x in input().split())
a = list(map(int, input().split()))

b = np.zeros(m)
idx = 0

for i in range(n):
	for j in range(n):
		if b[idx] < a[i]+a[j]:
			b[idx] = a[i]+a[j]
			idx = np.argmin(b)

print(int(sum(b)))

Buggy submission (verdict: Runtime Error):
import sys
input = sys.stdin.readline

n,m = (int(x) for x in input().split())
a = list(map(int, input().split()))

b = [0]*m
idx = 0

for i in range(n):
	for j in range(n):
		tmp = a[i]+a[j]
		if b[idx] < tmp:
			b[idx] = tmp
			idx = b.index(min(b))

print(sum(b))

Buggy submission (verdict: Runtime Error):
import sys
input = sys.stdin.readline
import numpy as np

n,m = (int(x) for x in input().split())
a = list(map(int, input().split()))

b = [0]*m
tmp = 0

for i in range(n):
	for j in range(n):
		if b[tmp] < a[i]+a[j]:
			b[tmp] = a[i]+a[j]
			tmp = b.index(min(b))

print(sum(b))

Buggy submission (verdict: Runtime Error):
import sys
input = sys.stdin.readline
import numpy as np

n,m = (int(x) for x in input().split())
a = list(map(int, input().split()))

b = [0]*m

for i in range(n):
	for j in range(n):
		tmp = b.index(min(b))
		if b[tmp] < a[i]+a[j]:
			b[tmp] = a[i]+a[j]

print(sum(b))

Buggy submission (verdict: Runtime Error):
import sys
input = sys.stdin.readline

n,m = (int(x) for x in input().split())
a = list(map(int, input().split()))

b = [0]*(n*n)
ans = 0

for i in range(n):
	for j in range(n):
		b[i*n+j] = a[i]+a[j]

b.sort(reverse=True)

for i in range(m):
	ans += b[i]

print(ans)