27#ifndef OR_TOOLS_BOP_BOP_LS_H_
28#define OR_TOOLS_BOP_BOP_LS_H_
33#include "absl/container/flat_hash_map.h"
34#include "absl/container/flat_hash_set.h"
35#include "absl/random/random.h"
81 std::vector<sat::Literal>* propagated_literals);
103class LocalSearchAssignmentIterator;
122 bool ShouldBeRun(
const ProblemState& problem_state)
const override;
127 int64_t state_update_stamp_;
132 const int max_num_decisions_;
141 std::unique_ptr<LocalSearchAssignmentIterator> assignment_iterator_;
144 absl::BitGenRef random_;
159template <
typename IntType>
172 void ChangeState(IntType i,
bool should_be_inside);
176 int size()
const {
return size_; }
179 const std::vector<IntType>&
Superset()
const {
return stack_; }
192 std::vector<IntType> stack_;
193 std::vector<bool> in_stack_;
197 std::vector<int> saved_sizes_;
198 std::vector<int> saved_stack_sizes_;
203template <
typename IntType>
211 for (IntType i(0); i < size; ++i) {
212 hashes_[i] = absl::Uniform<uint64_t>(random_);
223 uint64_t
Hash(
const std::vector<IntType>& set)
const {
225 for (
const IntType i : set)
hash ^= hashes_[i];
231 uint64_t
Hash(IntType e)
const {
return hashes_[e]; }
237 absl::BitGenRef random_;
291 void Assign(
const std::vector<sat::Literal>& literals);
321 return infeasible_constraint_set_.
size();
326 return infeasible_constraint_set_.
Superset();
344 return constraint_lower_bounds_[constraint];
349 return constraint_upper_bounds_[constraint];
356 return constraint_values_[constraint];
370 void InitializeConstraintSetHasher();
377 DEFINE_INT_TYPE(ConstraintIndexWithDirection, int32_t);
378 ConstraintIndexWithDirection FromConstraintIndex(ConstraintIndex
index,
380 return ConstraintIndexWithDirection(2 *
index.value() + (up ? 1 : 0));
385 void MakeObjectiveConstraintInfeasible(
int delta);
389 struct ConstraintEntry {
390 ConstraintEntry(ConstraintIndex c, int64_t w) : constraint(c),
weight(w) {}
391 ConstraintIndex constraint;
401 BopSolution assignment_;
402 BopSolution reference_;
405 BacktrackableIntegerSet<ConstraintIndex> infeasible_constraint_set_;
410 std::vector<int> flipped_var_trail_backtrack_levels_;
411 std::vector<VariableIndex> flipped_var_trail_;
414 std::vector<sat::Literal> tmp_potential_repairs_;
415 NonOrderedSetHasher<ConstraintIndexWithDirection> constraint_set_hasher_;
416 absl::flat_hash_map<uint64_t, std::vector<sat::Literal>>
417 hash_to_potential_repairs_;
469 TermIndex init_term_index,
470 TermIndex start_term_index)
const;
474 bool RepairIsValid(ConstraintIndex ct_index, TermIndex term_index)
const;
491 void SortTermsOfEachConstraints(
int num_variables);
495 by_constraint_matrix_;
510 int max_num_decisions,
511 int max_num_broken_constraints,
512 absl::BitGenRef random,
519 use_potential_one_flip_repairs_ = v;
543 return better_solution_has_been_found_;
556 void UseCurrentStateAsReference();
559 static constexpr size_t kStoredMaxDecisions = 4;
567 SearchNode(ConstraintIndex c, TermIndex t) : constraint(c), term_index(t) {}
568 ConstraintIndex constraint;
569 TermIndex term_index;
573 void ApplyDecision(sat::Literal
literal);
584 bool NewStateIsInTranspositionTable(sat::Literal l);
587 void InsertInTranspositionTable();
591 void InitializeTranspositionTableKey(
592 std::array<int32_t, kStoredMaxDecisions>*
a);
597 bool EnqueueNextRepairingTermIfAny(ConstraintIndex ct_to_repair,
600 const int max_num_decisions_;
601 const int max_num_broken_constraints_;
602 bool better_solution_has_been_found_;
603 AssignmentAndConstraintFeasibilityMaintainer maintainer_;
604 SatWrapper*
const sat_wrapper_;
605 OneFlipConstraintRepairer repairer_;
606 std::vector<SearchNode> search_nodes_;
610 std::vector<sat::Literal> tmp_propagated_literals_;
625 bool use_transposition_table_;
626 absl::flat_hash_set<std::array<int32_t, kStoredMaxDecisions>>
627 transposition_table_;
629 bool use_potential_one_flip_repairs_;
635 int64_t num_skipped_nodes_;
639 int64_t num_improvements_;
640 int64_t num_improvements_by_one_flip_repairs_;
641 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)