提出 #58342253

---

ソースコード 拡げる

def w a; warn a.inspect end

N,M,K,*A = $<.read.split.map(&:to_i)
if N==M
	puts [0]*N*' '
	exit
end
I = N.times.group_by{|i| A[i] }
A.sort!.reverse! # A[]: A[i] より大きい値を数えるカウンター
B = A[0,M] # B[-1]: X 票を M+1 人に分配したときの底が満たせないライン
S = A.inject([0]){|s,a| s<<s[-1]+a }
X = K-S[-1] # x の最大値
Xs = [nil]*N # 答え 
I.keys.sort.each{|a|
	A.pop while A[0] && A[-1]<=a
	sm1 = M<=A.size ? S[M]+a : S[M+1] # M+1 人の既得票
	B.pop while B[0] && B[-1]*(M+1-B.size)-(sm1-S[B.size])<=X
#w [a,A,B,sm1,M+1-B.size,sm1-S[B.size],X]
	x = if M<=A.size && M==B.size # X 票足して M 人目に届かない
			-1
		elsif A.size<B.size # X 票足して M 人が a に届かない(a に届くのが M 人に満たない)
			0
		else
			x = sm1-S[B.size]+X
			k = M+1-B.size
			# x 票を k 人で均等に分ける
			(x+1)/k-a
		end
	I[a].each{|i|
		Xs[i] = x
	}
}
puts Xs*' '

提出情報

ジャッジ結果