27#ifndef OR_TOOLS_BOP_BOP_LS_H_
28#define OR_TOOLS_BOP_BOP_LS_H_
34#include "absl/container/flat_hash_map.h"
35#include "absl/container/flat_hash_set.h"
36#include "absl/random/random.h"
42#include "ortools/sat/boolean_problem.pb.h"
83 std::vector<sat::Literal>* propagated_literals);
105class LocalSearchAssignmentIterator;
124 bool ShouldBeRun(
const ProblemState& problem_state)
const override;
129 int64_t state_update_stamp_;
134 const int max_num_decisions_;
143 std::unique_ptr<LocalSearchAssignmentIterator> assignment_iterator_;
146 absl::BitGenRef random_;
161template <
typename IntType>
174 void ChangeState(IntType i,
bool should_be_inside);
178 int size()
const {
return size_; }
181 const std::vector<IntType>&
Superset()
const {
return stack_; }
194 std::vector<IntType> stack_;
195 std::vector<bool> in_stack_;
199 std::vector<int> saved_sizes_;
200 std::vector<int> saved_stack_sizes_;
205template <
typename IntType>
213 for (IntType i(0); i < size; ++i) {
214 hashes_[i] = absl::Uniform<uint64_t>(random_);
225 uint64_t
Hash(
const std::vector<IntType>& set)
const {
227 for (
const IntType i : set)
hash ^= hashes_[i];
233 uint64_t
Hash(IntType e)
const {
return hashes_[e]; }
239 absl::BitGenRef random_;
273 const sat::LinearBooleanProblem& problem, absl::BitGenRef random);
293 void Assign(
const std::vector<sat::Literal>& literals);
323 return infeasible_constraint_set_.
size();
328 return infeasible_constraint_set_.
Superset();
346 return constraint_lower_bounds_[constraint];
351 return constraint_upper_bounds_[constraint];
358 return constraint_values_[constraint];
372 void InitializeConstraintSetHasher();
379 DEFINE_STRONG_INDEX_TYPE(ConstraintIndexWithDirection);
380 ConstraintIndexWithDirection FromConstraintIndex(ConstraintIndex
index,
382 return ConstraintIndexWithDirection(2 *
index.value() + (up ? 1 : 0));
387 void MakeObjectiveConstraintInfeasible(
int delta);
391 struct ConstraintEntry {
392 ConstraintEntry(ConstraintIndex c, int64_t w) : constraint(c),
weight(w) {}
393 ConstraintIndex constraint;
403 BopSolution assignment_;
404 BopSolution reference_;
407 BacktrackableIntegerSet<ConstraintIndex> infeasible_constraint_set_;
412 std::vector<int> flipped_var_trail_backtrack_levels_;
413 std::vector<VariableIndex> flipped_var_trail_;
416 std::vector<sat::Literal> tmp_potential_repairs_;
417 NonOrderedSetHasher<ConstraintIndexWithDirection> constraint_set_hasher_;
418 absl::flat_hash_map<uint64_t, std::vector<sat::Literal>>
419 hash_to_potential_repairs_;
445 const sat::LinearBooleanProblem& problem,
471 TermIndex init_term_index,
472 TermIndex start_term_index)
const;
476 bool RepairIsValid(ConstraintIndex ct_index, TermIndex term_index)
const;
493 void SortTermsOfEachConstraints(
int num_variables);
497 by_constraint_matrix_;
512 int max_num_decisions,
513 int max_num_broken_constraints,
514 absl::BitGenRef random,
521 use_potential_one_flip_repairs_ = v;
545 return better_solution_has_been_found_;
558 void UseCurrentStateAsReference();
561 static constexpr size_t kStoredMaxDecisions = 4;
569 SearchNode(ConstraintIndex c, TermIndex t) : constraint(c), term_index(t) {}
570 ConstraintIndex constraint;
571 TermIndex term_index;
575 void ApplyDecision(sat::Literal
literal);
586 bool NewStateIsInTranspositionTable(sat::Literal l);
589 void InsertInTranspositionTable();
593 void InitializeTranspositionTableKey(
594 std::array<int32_t, kStoredMaxDecisions>*
a);
599 bool EnqueueNextRepairingTermIfAny(ConstraintIndex ct_to_repair,
602 const int max_num_decisions_;
603 const int max_num_broken_constraints_;
604 bool better_solution_has_been_found_;
605 AssignmentAndConstraintFeasibilityMaintainer maintainer_;
606 SatWrapper*
const sat_wrapper_;
607 OneFlipConstraintRepairer repairer_;
608 std::vector<SearchNode> search_nodes_;
612 std::vector<sat::Literal> tmp_propagated_literals_;
627 bool use_transposition_table_;
628 absl::flat_hash_set<std::array<int32_t, kStoredMaxDecisions>>
629 transposition_table_;
631 bool use_potential_one_flip_repairs_;
637 int64_t num_skipped_nodes_;
641 int64_t num_improvements_;
642 int64_t num_improvements_by_one_flip_repairs_;
643 int64_t num_inspected_one_flip_repairs_;
void resize(size_type new_size)
A simple class to enforce both an elapsed time limit and a deterministic time limit in the same threa...
int64_t ConstraintLowerBound(ConstraintIndex constraint) const
AssignmentAndConstraintFeasibilityMaintainer(const sat::LinearBooleanProblem &problem, absl::BitGenRef random)
void AddBacktrackingLevel()
void UseCurrentStateAsReference()
static const ConstraintIndex kObjectiveConstraint
int NumInfeasibleConstraints() const
std::string DebugString() const
void SetReferenceSolution(const BopSolution &reference_solution)
int64_t ConstraintUpperBound(ConstraintIndex constraint) const
size_t NumConstraints() const
bool ConstraintIsFeasible(ConstraintIndex constraint) const
int64_t ConstraintValue(ConstraintIndex constraint) const
const std::vector< ConstraintIndex > & PossiblyInfeasibleConstraints() const
const BopSolution & reference() const
bool Assignment(VariableIndex var) const
void Assign(const std::vector< sat::Literal > &literals)
const std::vector< sat::Literal > & PotentialOneFlipRepairs()
void ClearAndResize(IntType n)
void AddBacktrackingLevel()
BacktrackableIntegerSet()
void ChangeState(IntType i, bool should_be_inside)
const std::vector< IntType > & Superset() const
const std::string & name() const
bool Value(VariableIndex var) const
const BopSolution & LastReferenceAssignment() const
~LocalSearchAssignmentIterator()
void SynchronizeSatWrapper()
void UseTranspositionTable(bool v)
std::string DebugString() const
LocalSearchAssignmentIterator(const ProblemState &problem_state, int max_num_decisions, int max_num_broken_constraints, absl::BitGenRef random, SatWrapper *sat_wrapper)
void UsePotentialOneFlipRepairs(bool v)
void Synchronize(const ProblemState &problem_state)
double deterministic_time() const
bool BetterSolutionHasBeenFound() const
LocalSearchOptimizer(const std::string &name, int max_num_decisions, absl::BitGenRef random, sat::SatSolver *sat_propagator)
~LocalSearchOptimizer() override
bool IsInitialized() const
void Initialize(int size)
NonOrderedSetHasher(absl::BitGenRef random)
uint64_t Hash(IntType e) const
void IgnoreElement(IntType e)
uint64_t Hash(const std::vector< IntType > &set) const
sat::Literal GetFlip(ConstraintIndex ct_index, TermIndex term_index) const
static const TermIndex kInvalidTerm
ConstraintIndex ConstraintToRepair() const
bool RepairIsValid(ConstraintIndex ct_index, TermIndex term_index) const
TermIndex NextRepairingTerm(ConstraintIndex ct_index, TermIndex init_term_index, TermIndex start_term_index) const
static const TermIndex kInitTerm
OneFlipConstraintRepairer(const sat::LinearBooleanProblem &problem, const AssignmentAndConstraintFeasibilityMaintainer &maintainer, const sat::VariablesAssignment &sat_assignment)
static const ConstraintIndex kInvalidConstraint
SatWrapper(sat::SatSolver *sat_solver)
std::vector< sat::Literal > FullSatTrail() const
int ApplyDecision(sat::Literal decision_literal, std::vector< sat::Literal > *propagated_literals)
void ExtractLearnedInfo(LearnedInfo *info)
bool IsModelUnsat() const
const sat::VariablesAssignment & SatAssignment() const
double deterministic_time() const
const VariablesAssignment & Assignment() const
bool IsModelUnsat() const
ModelSharedTimeLimit * time_limit
Collection of objects used to extend the Constraint Solver library.
ConstraintTerm(VariableIndex v, int64_t w)