Submission #14275560

---

Source Code Expand

(N,K,D),A = 2.times.map{ gets.split.map(&:to_i) }

Less = lambda{|a,b|
	next a[0]<b[0] || a[0]==b[0] && a[1]<b[1] ? a : b
}
L = 1<<(A.size-1).bit_length
H = [nil]*(2*L-1)
H.fill([A[-1],A.size-1],A.size-L)
A.each_with_index{|a,i|
	H[i-L] = [a,i]
}
(L-2).downto(0){|i|
	j = 2*i+1
	H[i] = Less[H[j],H[j+1]]
}
Min = lambda{|s,e|
	s = 0 if s<0
	e = N-1 if N<=e
	next nil if e<s

	s,e = s+L-1,e+L-1
	m1,m2 = H[s],H[e]
	while 0
		sp,ep = s/2,e/2-1
		break if ep<sp
		m1 = Less[m1,H[sp]]
		m2 = Less[m2,H[ep]]
		s,e = sp,ep
	end
	next Less[m1,m2]
}

c,C = -D,[]
K.downto(1){|k|
	m,i = Min[c+D, N-1-D*(k-1)]
	break unless m
	C << m
	c = i
}
puts(C.size==K ? C.join(" ") : -1)

Submission Info

Compile Error

./Main.rb:29: warning: literal in condition

Judge Result