always-cautious/ortools-clone
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

simplify MakeIsMemberCt if first argument is a product

Browse Source
This commit is contained in:
lperron@google.com
2014-05-25 02:13:22 +00:00
parent 08269e4ab0
commit a3bef694ab
1 changed files with 15 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/constraint_solver/expr_cst.cc
View File

@@ -1222,7 +1222,21 @@ class IsMemberCt : public Constraint {
template <class T>
Constraint* BuildIsMemberCt(Solver* const solver, IntExpr* const expr,
const std::vector<T>& values, IntVar* const boolvar) {
const std::vector<T>& values, IntVar* const boolvar) {
// Simplify and filter if expr is a filter.
IntExpr* sub = nullptr;
int64 coef = 1;
if (solver->IsProduct(expr, &sub, &coef) && coef != 0 && coef != 1) {
std::vector<int64> new_values;
new_values.reserve(values.size());
for (const int64 value : values) {
if (value % coef == 0) {
new_values.push_back(value / coef);
}
}
return BuildIsMemberCt(solver, sub, new_values, boolvar);
}
std::set<T> set_of_values(values.begin(), values.end());
std::vector<int64> filtered_values;
bool all_values = false;