結果

問題 No.3293 Golden Cross
ユーザー kona0001
提出日時 2025-09-09 04:04:05
言語 Ruby
(3.4.1)
結果
RE  
実行時間 -
コード長 1,641 bytes
コンパイル時間 165 ms
コンパイル使用メモリ 8,320 KB
実行使用メモリ 32,896 KB
最終ジャッジ日時 2025-10-03 13:04:14
合計ジャッジ時間 19,626 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample RE * 2
other RE * 49
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff # 

h,w,k = gets.split.map(&:to_i)
a = h.times.map{gets.split.map(&:to_i)}
b = h.times.map{gets.split.map(&:to_i)}

sum_row = Array.new(h,0)
sum_col = Array.new(w,0)
cost_row = Array.new(h){Array.new}
cost_col = Array.new(w){Array.new}
h.times do |i|
  w.times do |j|
    sum_row[i] += a[i][j]
    cost_row[i] << b[i][j]
    sum_col[j] += a[i][j]
    cost_col[j] << b[i][j]
  end
  cost_row[i].sort!
end
w.times do |i|
  cost_col[i].sort!
end

# Z を決め打ちした時の部分問題
# (a + b) * (a + c) に対応
def calc(a,b,c,y,z,k)
  b_opt = (z*(a+c) - y*(a+b) + k) / (2*y)
  ky = k/y
  kouho = [0,b_opt,b_opt+1,ky]
  kouho.each do |v|
    next if v < 0 || ky < v
    rest_k = k - v*y
    val = (a+b+v) * (a+c+rest_k/z)
    max = val if max < val
  end
  max
end

# 三分探索
def solve(x,y,z,x_cost,y_cost,z_cost,k)
  l = 0
  r = k / z_cost
  while r - l > 2 #精度
    dif = (r - l) / 3
    t1,t2 = l+dif,r-dif
    s1 = calc(z+t1,x,y,x_cost,y_cost,k-t1*z_cost)
    s2 = calc(z+t2,x,y,x_cost,y_cost,k-t2*z_cost)
    if s1 < s2 #上に凸
      l = t1
    else
      r = t2
    end
  end
  max = 0
  3.times do |i|
    t = l + i
    next if t*z_cost > k
    v = calc(z+t,x,y,x_cost,y_cost,k-t*z_cost)
    max = v if max < v
  end
  max
end

max = 0
h.times do |i|
  ai = a[i]
  bi = b[i]
  cost_rowi = cost_row[i]
  w.times do |j|
    z = ai[j]
    z_cost = bi[j]
    x = sum_row[i] - z
    x_cost = z_cost == cost_rowi[0] ? cost_rowi[1] : cost_rowi[0]
    y = sum_col[j] - z
    y_cost = z_cost == cost_col[j][0] ? cost_col[j][1] : cost_col[j][0]
    v = solve(x,y,z,x_cost,y_cost,z_cost,k)
    max = v if max < v
  end
end
puts max
0