結果

問題 No.5005 3-SAT
ユーザー tomeruntomerun
提出日時 2022-04-29 16:20:49
言語 Crystal
(1.11.2)
結果
AC  
実行時間 1,557 ms / 2,000 ms
コード長 7,613 bytes
コンパイル時間 19,462 ms
実行使用メモリ 6,136 KB
スコア 75,503
最終ジャッジ日時 2022-04-29 16:23:51
合計ジャッジ時間 176,807 ms
ジャッジサーバーID
(参考情報)
judge14 / judge13
純コード判定しない問題か言語
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1,545 ms
5,704 KB
testcase_01 AC 1,527 ms
5,636 KB
testcase_02 AC 1,538 ms
5,688 KB
testcase_03 AC 1,546 ms
5,572 KB
testcase_04 AC 1,528 ms
5,616 KB
testcase_05 AC 1,545 ms
5,620 KB
testcase_06 AC 1,517 ms
5,732 KB
testcase_07 AC 1,543 ms
5,796 KB
testcase_08 AC 1,540 ms
5,564 KB
testcase_09 AC 1,541 ms
5,708 KB
testcase_10 AC 1,535 ms
5,692 KB
testcase_11 AC 1,538 ms
5,760 KB
testcase_12 AC 1,535 ms
5,580 KB
testcase_13 AC 1,533 ms
5,632 KB
testcase_14 AC 1,516 ms
5,616 KB
testcase_15 AC 1,529 ms
5,784 KB
testcase_16 AC 1,546 ms
5,620 KB
testcase_17 AC 1,536 ms
5,688 KB
testcase_18 AC 1,545 ms
5,560 KB
testcase_19 AC 1,520 ms
5,788 KB
testcase_20 AC 1,451 ms
5,992 KB
testcase_21 AC 1,544 ms
5,636 KB
testcase_22 AC 1,479 ms
5,620 KB
testcase_23 AC 1,523 ms
5,760 KB
testcase_24 AC 1,528 ms
5,808 KB
testcase_25 AC 1,526 ms
5,616 KB
testcase_26 AC 1,507 ms
5,720 KB
testcase_27 AC 1,536 ms
5,636 KB
testcase_28 AC 1,538 ms
5,700 KB
testcase_29 AC 1,528 ms
5,792 KB
testcase_30 AC 1,530 ms
5,752 KB
testcase_31 AC 1,534 ms
5,768 KB
testcase_32 AC 1,536 ms
5,760 KB
testcase_33 AC 1,547 ms
5,696 KB
testcase_34 AC 1,539 ms
5,668 KB
testcase_35 AC 1,522 ms
5,784 KB
testcase_36 AC 1,540 ms
5,720 KB
testcase_37 AC 1,528 ms
5,680 KB
testcase_38 AC 1,524 ms
5,724 KB
testcase_39 AC 1,486 ms
5,616 KB
testcase_40 AC 1,539 ms
5,644 KB
testcase_41 AC 1,547 ms
5,724 KB
testcase_42 AC 1,504 ms
5,788 KB
testcase_43 AC 1,538 ms
5,688 KB
testcase_44 AC 1,509 ms
5,636 KB
testcase_45 AC 1,521 ms
5,692 KB
testcase_46 AC 1,505 ms
5,764 KB
testcase_47 AC 1,545 ms
5,624 KB
testcase_48 AC 1,545 ms
5,708 KB
testcase_49 AC 1,529 ms
5,692 KB
testcase_50 AC 1,521 ms
5,552 KB
testcase_51 AC 1,556 ms
5,808 KB
testcase_52 AC 1,548 ms
5,628 KB
testcase_53 AC 1,531 ms
6,136 KB
testcase_54 AC 1,532 ms
5,764 KB
testcase_55 AC 1,517 ms
5,620 KB
testcase_56 AC 1,556 ms
5,616 KB
testcase_57 AC 1,535 ms
5,976 KB
testcase_58 AC 1,535 ms
5,560 KB
testcase_59 AC 1,513 ms
5,620 KB
testcase_60 AC 1,544 ms
5,616 KB
testcase_61 AC 1,535 ms
5,696 KB
testcase_62 AC 1,530 ms
5,792 KB
testcase_63 AC 1,556 ms
5,616 KB
testcase_64 AC 1,555 ms
5,684 KB
testcase_65 AC 1,542 ms
5,696 KB
testcase_66 AC 1,536 ms
5,620 KB
testcase_67 AC 1,488 ms
5,632 KB
testcase_68 AC 1,547 ms
5,648 KB
testcase_69 AC 1,524 ms
5,684 KB
testcase_70 AC 1,524 ms
5,552 KB
testcase_71 AC 1,523 ms
5,684 KB
testcase_72 AC 1,526 ms
5,732 KB
testcase_73 AC 1,533 ms
5,764 KB
testcase_74 AC 1,537 ms
5,696 KB
testcase_75 AC 1,552 ms
5,612 KB
testcase_76 AC 1,508 ms
5,764 KB
testcase_77 AC 1,536 ms
5,736 KB
testcase_78 AC 1,537 ms
5,740 KB
testcase_79 AC 1,545 ms
5,692 KB
testcase_80 AC 1,526 ms
5,764 KB
testcase_81 AC 1,549 ms
5,648 KB
testcase_82 AC 1,523 ms
5,644 KB
testcase_83 AC 1,532 ms
5,688 KB
testcase_84 AC 1,537 ms
5,784 KB
testcase_85 AC 1,495 ms
5,796 KB
testcase_86 AC 1,556 ms
5,692 KB
testcase_87 AC 1,548 ms
5,736 KB
testcase_88 AC 1,541 ms
5,692 KB
testcase_89 AC 1,531 ms
5,708 KB
testcase_90 AC 1,532 ms
5,712 KB
testcase_91 AC 1,546 ms
5,756 KB
testcase_92 AC 1,557 ms
5,552 KB
testcase_93 AC 1,550 ms
5,568 KB
testcase_94 AC 1,533 ms
5,644 KB
testcase_95 AC 1,544 ms
5,800 KB
testcase_96 AC 1,546 ms
5,636 KB
testcase_97 AC 1,533 ms
5,708 KB
testcase_98 AC 1,516 ms
5,572 KB
testcase_99 AC 1,526 ms
5,620 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

START_TIME = Time.utc.to_unix_ms
TL         = 1580
N          = 2048
M          =  256
INF        = 1 << 29
RND        = Random.new(2)

macro debug(msg)
  {% if flag?(:trace) %}
    STDERR.puts({{msg}})
  {% end %}
end

macro debugf(format_string, *args)
  {% if flag?(:trace) %}
    STDERR.printf({{format_string}}, {{*args}})
  {% end %}
end

class IndexSet
  getter :upper

  def initialize(@upper : Int32)
    @que = [] of Int32
    @pos = Array(Int32).new(@upper, -1)
  end

  def add(v)
    if @pos[v] == -1
      @pos[v] = @que.size
      @que << v
    end
  end

  def remove(v)
    p = @pos[v]
    b = @que[-1]
    @que[p] = b
    @que.pop
    @pos[b] = p
    @pos[v] = -1
  end

  def get_random
    @que[RND.rand(@que.size)]
  end

  def size
    @que.size
  end

  def to_s(io)
    @que.to_s(io)
  end
end

class Result
  property :bits, :score

  def initialize(@bits : Array(Int32), @score : Int32)
  end

  def to_s(io)
    io << @bits.map { |v| v == -1 ? 0 : v }.reverse.join
  end
end

struct Cond
  getter :var, :val

  def initialize(@var : Int32, @val : Int32)
  end
end

alias Term = Tuple(Cond, Cond, Cond)

class Solver
  def initialize
    @terms = Array(Term).new(N) do
      a, b, c, p, q, r = read_line.split.map(&.to_i)
      {Cond.new(a, p), Cond.new(b, q), Cond.new(c, r)}
    end
    @var_pos = Array(Array(Tuple(Int32, Int32))).new(M) { [] of Tuple(Int32, Int32) }
    N.times do |i|
      3.times do |j|
        term = @terms[i][j]
        @var_pos[term.var] << {i, term.val}
      end
    end
    @best_result = Result.new(Array.new(M, -1), 0)
    @ans = Array(Int32).new(M, -1)
    @search_order = [0, 1, 2]
    @dfs_count = 0
    @eval = Array(Int32).new(3, 0)
    @ok_count = Array(Int32).new(N, 0)
    @ng_count = Array(Int32).new(N, 0)
    @pass = Array(Bool).new(N) do |i|
      ts = @terms[i]
      [[0, 1], [1, 2], [2, 0]].any? { |ps| ts[ps[0]].var == ts[ps[1]].var && ts[ps[0]].var != ts[ps[1]].var }
    end
  end

  def solve(timelimit)
    start_time = Time.utc.to_unix_ms
    before_time = start_time
    solve_single()
    after_time = Time.utc.to_unix_ms
    worst_time = after_time - before_time
    turn = 0
    while true
      before_time = after_time
      if before_time + worst_time > timelimit
        debug("turn:#{turn} worst_time:#{worst_time}")
        break
      end
      solve_single()
      after_time = Time.utc.to_unix_ms
      worst_time = {worst_time, after_time - before_time}.max
      turn += 1
    end
    STDERR.puts("validate_score:#{score(@best_result.bits)}")
    return @best_result
  end

  def solve_single
    @ans.fill(-1)
    @dfs_count = 0
    @ok_count.fill(0)
    @ng_count.fill(0)
    pos = 0
    while pos < N
      if try_next(pos)
        while pos < N && @ok_count[pos] > 0
          pos += 1
        end
        if pos > @best_result.score
          @best_result.score = pos
          @best_result.bits[0, M] = @ans
          debug("best_score:#{pos}")
        end
        # if score(@ans) != pos
        #   debug("validate_score:#{score(@ans)} score:#{pos}")
        #   exit
        # end
      else
        break
      end
    end
  end

  def try_next(pos)
    shuffle(@search_order)
    3.times do |i|
      t = @terms[pos][@search_order[i]]
      if @ans[t.var] == -1
        @ans[t.var] = t.val
        @var_pos[t.var].each do |ps|
          p, v = ps
          if v == t.val
            @ok_count[p] += 1
          else
            @ng_count[p] += 1
          end
        end
        return true
      end
    end
    set = IndexSet.new(N)
    set.add(pos)
    used = Set(Int32).new
    @dfs_count = 0
    return try_dfs(pos, set, used)
  end

  def try_dfs(pos, set, used)
    # debug("try_dfs:#{pos} #{set} #{used}")
    if set.size == 0
      return true
    end
    if @dfs_count > 10000
      return false
    end
    cur = set.get_random
    # set.remove(cur)
    # if @ok_count[cur] > 0
    #   if try_dfs(pos, set, used)
    #     return true
    #   else
    #     set.add(cur)
    #     return false
    #   end
    # end
    shuffle(@search_order)
    @dfs_count += 1
    3.times do |i|
      t = @terms[cur][@search_order[i]]
      next if used.includes?(t.var)
      used << t.var
      old = @ans[t.var]
      @ans[t.var] = t.val
      @var_pos[t.var].each do |ps|
        p, v = ps
        if v == t.val
          @ok_count[p] += 1
          @ng_count[p] -= 1
          if p <= pos && @ok_count[p] == 1
            set.remove(p)
          end
        elsif old != -1
          @ok_count[p] -= 1
          @ng_count[p] += 1
          if p <= pos && @ok_count[p] == 0
            set.add(p)
          end
        end
        # debug("@ok_count[#{p}]=#{@ok_count[p]}")
        # debug("@ng_count[#{p}]=#{@ng_count[p]}")
      end
      if try_dfs(pos, set, used)
        return true
      end
      used.delete(t.var)
      @ans[t.var] = old
      @var_pos[t.var].each do |ps|
        p, v = ps
        if v == t.val
          @ok_count[p] -= 1
          @ng_count[p] += 1
          if p <= pos && @ok_count[p] == 0
            set.add(p)
          end
        elsif old != -1
          @ok_count[p] += 1
          @ng_count[p] -= 1
          if p <= pos && @ok_count[p] == 1
            set.remove(p)
          end
        end
      end
    end
    # set.add(cur)
    return false
  end

  def solve_single2
    @ans.fill(-1)
    @dfs_count = 0
    @ok_count.fill(0)
    @ng_count.fill(0)
    dfs(0)
  end

  def dfs(pos)
    if @pass[pos]
      dfs(pos + 1)
    end
    if pos > @best_result.score
      @best_result.score = pos
      @best_result.bits[0, M] = @ans
      debug("best_score:#{pos}")
    end
    if @dfs_count > 1000
      return
    end
    if @terms[pos].any? { |t| @ans[t.var] == t.val }
      dfs(pos + 1)
    else
      3.times do |i|
        t = @terms[pos][i]
        if @ans[t.var] != -1
          @eval[i] = -INF
          next
        end
        @eval[i] = RND.rand(5000000)
        @var_pos[t.var].each do |ps|
          p, v = ps
          next if p <= pos
          if v == t.val
            if @ok_count[p] == 0
              # @eval[i] += N - p
            end
          else
            if @ok_count[p] != 0
              # @eval[i] -= (N - p) * ((@ng_count[p] + 1) ** 2)
              if p <= @best_result.score && @ng_count[p] == 2
                @eval[i] -= INF * 2
              end
            end
          end
        end
      end
      # debug(@eval)
      cands = 3.times.select { |i| @eval[i] > -INF }.to_a
      return if cands.empty?
      @dfs_count += 1 if cands.size > 1
      cands.sort_by { |i| -@eval[i] }.each do |ti|
        t = @terms[pos][ti]
        @ans[t.var] = t.val
        @var_pos[t.var].each do |ps|
          p, v = ps
          if p > pos
            if v == t.val
              @ok_count[p] += 1
            else
              @ng_count[p] += 1
            end
          end
        end
        dfs(pos + 1)
        @var_pos[t.var].each do |ps|
          p, v = ps
          if p > pos
            if v == t.val
              @ok_count[p] -= 1
            else
              @ng_count[p] -= 1
            end
          end
        end
        @ans[t.var] = -1
      end
    end
  end

  def shuffle(a)
    (a.size - 1).times do |i|
      pos = RND.rand(a.size - i) + i
      a[i], a[pos] = a[pos], a[i]
    end
  end

  def score(ans)
    N.times do |i|
      if @terms[i].all? { |t| ans[t.var] != t.val }
        return i
      end
    end
    return N
  end
end

def main
  solver = Solver.new
  res = solver.solve(START_TIME + TL)
  puts res
  {% if flag?(:local) %}
    STDERR.puts("score:#{res.score}")
  {% end %}
end

main
0