結果

問題 No.1470 Mex Sum
コンテスト
ユーザー magurofly
提出日時 2021-04-09 21:33:06
言語 Ruby
(3.4.1)
結果
WA  
実行時間 -
コード長 787 bytes
コンパイル時間 210 ms
コンパイル使用メモリ 7,424 KB
実行使用メモリ 42,240 KB
最終ジャッジ日時 2024-06-25 04:23:46
合計ジャッジ時間 44,697 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 3 WA * 46
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff # 

class Range
    def upper_bound
        ac, wa = self.begin, self.end
        wa += 1 unless self.exclude_end?
        while ac + 1 < wa
            wj = (ac + wa) / 2
            if yield(wj)
                ac = wj
            else
                wa = wj
            end
        end
        ac
    end
end

N = gets.to_i
A = gets.split.map(&:to_i)

counts = [[0, 0]]
A.each do |x|
    a, b = counts[-1]
    case x
    when 1
        counts << [a + 1, b]
    when 2
        counts << [a, b + 1]
    else
        counts << [a, b]
    end
end

ans = 0
(0 .. N - 2).each do |l|
    a, b = counts[l]
    r1 = (l + 1 .. N).upper_bound { |i| a == counts[i][0] }
    r2 = (r1 .. N).upper_bound { |i| b == counts[i][1] }
    ans += 1 * (r1 - l - 1) + 2 * (r2 - r1) + 3 * (N - r2)
end

puts ans
0