結果

問題 No.1007 コイン集め
ユーザー gemmaro
提出日時 2020-03-20 10:24:46
言語 Ruby
(3.4.1)
結果
WA  
実行時間 -
コード長 353 bytes
コンパイル時間 443 ms
コンパイル使用メモリ 7,552 KB
実行使用メモリ 20,480 KB
最終ジャッジ日時 2024-12-14 03:59:59
合計ジャッジ時間 3,425 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 18 WA * 1
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff #

# frozen_string_literal: true

N, K = gets.chomp.split.map(&:to_i)
AS = gets.chomp.split.map(&:to_i)

if AS[K - 1].zero?
  puts(0)
  exit
end

count = 0
[AS[K..-1], AS[0..K - 1].reverse].each do |a|
  a.take_while do |i|
    if i >= 2
      count += i
      true
    elsif i == 1
      count += 1
      false
    else false
    end
  end
end

puts count
0