結果

問題 No.121 傾向と対策：門松列（その２）
ユーザー gigurururugigurururu
提出日時 2015-01-09 16:27:50
言語 Ruby
(3.3.0)
結果
RE  
実行時間 -
コード長 811 bytes
コンパイル時間 223 ms
コンパイル使用メモリ 11,544 KB
実行使用メモリ 31,320 KB
最終ジャッジ日時 2023-09-03 22:53:11
合計ジャッジ時間 8,708 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge12
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 RE -
testcase_01 RE -
testcase_02 RE -
testcase_03 TLE -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff # 

class BIT
def initialize n;@n=n;@buf=[0]*(n+1)end
def add x,v;(@buf[x]+=v;x+=x&-x)while x<=@n;end
def sum x;s=0;(s+=@buf[x];x&=x-1)while x>0;s;end
end
n=gets.to_i
l=gets.split.map(&:to_i)
bl=BIT.new n#左にある値
br=BIT.new n#右にある値
#座標圧縮
z=Hash[l.sort.uniq.map.with_index{|i,j|[i,j+1]}]
t=l.map{|v|z[v]}
#最初は全部右にある
t.each{|v|br.add v,1}

cnt=t.map.with_index{|v,i|
  bl.add v,1
  br.add v,-1
  #左右の自分より小さい数同士、大きい数同士を掛けた数がパターン数
  bl.sum(v-1)*br.sum(v-1)+(i+1-bl.sum(v))*(n-i-1-br.sum(v))
}.inject(:+)
#左右が同じ数になるパターンを引く 解説(http://ideone.com/qVicVL)の通り
cnt-=(0...n).group_by{|i|t[i]}.values.flat_map{|a|(1...a.size).map{|i|i*(a.size-i)*(a[i]-a[i-1]-1)}}.inject(:+)
p cnt
0