package com.google.ortools.samples;
import com.google.ortools.constraintsolver.ConstraintSolverParameters;
import com.google.ortools.constraintsolver.Decision;
import com.google.ortools.constraintsolver.DecisionBuilder;
import com.google.ortools.constraintsolver.JavaDecisionBuilder;
import com.google.ortools.constraintsolver.IntVar;
import com.google.ortools.constraintsolver.Solver;
import java.util.logging.Logger;
/**
* Sample showing how to model using the constraint programming solver.
*
*/
public class RabbitsPheasants {
private static Logger logger = Logger.getLogger(RabbitsPheasants.class.getName());
static {
System.loadLibrary("jniortools");
}
public static class MyDecisionBuilder extends JavaDecisionBuilder {
IntVar var1;
IntVar var2;
public MyDecisionBuilder(IntVar v1, IntVar v2) {
var1 = v1;
var2 = v2;
}
@Override
public Decision next(Solver solver) throws Solver.FailException {
if (!var1.bound()) {
return solver.makeAssignVariableValue(var1, var1.min());
}
if (!var2.bound()) {
return solver.makeAssignVariableValue(var2, var2.min());
}
return null;
}
@Override
public String toString() {
return "MyDecisionBuilder";
}
}
/**
* Solves the rabbits + pheasants problem. We are seing 20 heads
* and 56 legs. How many rabbits and how many pheasants are we thus
* seeing?
*/
private static void solve(boolean traceSearch) {
ConstraintSolverParameters parameters =
ConstraintSolverParameters.newBuilder()
.mergeFrom(Solver.defaultSolverParameters())
.setTraceSearch(traceSearch)
.build();
Solver solver = new Solver("RabbitsPheasants", parameters);
IntVar rabbits = solver.makeIntVar(0, 100, "rabbits");
IntVar pheasants = solver.makeIntVar(0, 100, "pheasants");
solver.addConstraint(solver.makeEquality(solver.makeSum(rabbits, pheasants), 20));
solver.addConstraint(
solver.makeEquality(
solver.makeSum(solver.makeProd(rabbits, 4), solver.makeProd(pheasants, 2)), 56));
DecisionBuilder db = new MyDecisionBuilder(rabbits, pheasants);
solver.newSearch(db);
solver.nextSolution();
solver.endSearch();
}
public static void main(String[] args) throws Exception {
boolean traceSearch = args.length > 0 && args[1].equals("--trace");
RabbitsPheasants.solve(traceSearch);
}
}