Peteris's blog

CF #72 Ruby issue
By Peteris, 13 years ago, In English

I wrote the following solution in CF #72 for Problem B in Ruby:

  1. n, k = gets.split.map{|_|_.to_i}
  2. a = gets.split.map{|_|_.to_i}

  4. b = a.sort
  5. if (a.length == 1 ? a[0] : a.inject{|x,y| x + y}) <= k
  6.   puts -1
  7. else
  8.   i = 0
  9.   s = 0
  10.   while (b[i] - s) * (n - i) <= k
  11.     k -= (b[i] - s) * (n - i)
  12.     s = b[i]
  13.     i += 1
  14.   end
  15.   while b[i] - s == 0
  16.     i += 1
  17.   end

  19.   t = k / (n - i)
  20.   k -= t * (n - i)

  22.   j = 0
  23.   while k > 0
  24.     if a[j] >= b[i]
  25.       a[j] -= 1
  26.       k -= 1
  27.     end
  28.     j += 1
  29.   end

  31.   ans = ""
  32.   for k in j...n
  33.     if a[k] > t + s
  34.       ans += (k + 1).to_s + " "
  35.     end
  36.   end
  37.   for k in 0...j
  38.     if a[k] > t + s
  39.       ans += (k + 1).to_s + " "
  40.     end
  41.   end
  42.   puts ans.strip
  43. end

It worked fine on my computer, but on the server, it fails for case 5:

Test: #5, time: 80 ms., memory: 3408 KB, exit code: 0, checker exit code: 1, verdict: WRONG_ANSWER
Input
1 1
1
Output
-1
Answer
Checker Log
wrong answer Output contains longer sequence [length = 1], but answer contains 0 elements

Any Ruby gurus have a clue?
  • Vote: I like it
  • 0
  • Vote: I do not like it

13 years ago, # |
← Rev. 3   Vote: I like it 0 Vote: I do not like it

I do not know ruby, but it seems you output -1, but actually you should output  nothing in this case.

In the problem statement:

Note that this sequence may be empty. This case is present in pretests. You can just print nothing or print one "End of line"-character. Both will be accepted.
→ Reply