/** Solves the N Queens problem. See http://www.hakank.org/google_or_tools/nqueens2.py */
private static void solve(int n, int num, int print) {
Solver solver = new Solver("NQueens");
System.out.println("n: " + n);
//
// variables
//
IntVar[] q = solver.makeIntVarArray(n, 0, n - 1, "q");
//
// constraints
//
solver.addConstraint(solver.makeAllDifferent(q, true));
IntVar b = solver.makeIntVar(1, 1, "b");
IntVar[] q1 = new IntVar[n];
IntVar[] q2 = new IntVar[n];
for (int i = 0; i < n; i++) {
for (int j = 0; j < i; j++) {
// // q[i]+i != q[j]+j
solver.addConstraint(
solver.makeNonEquality(solver.makeSum(q[i], i).Var(), solver.makeSum(q[j], j).Var()));
// q[i]-i != q[j]-j
solver.addConstraint(
solver.makeNonEquality(solver.makeSum(q[i], -i).Var(), solver.makeSum(q[j], -j).Var()));
}
}
//
// Solve
//
DecisionBuilder db =
solver.makePhase(q, solver.CHOOSE_MIN_SIZE_LOWEST_MAX, solver.ASSIGN_CENTER_VALUE);
solver.newSearch(db);
int c = 0;
while (solver.nextSolution()) {
if (print != 0) {
for (int i = 0; i < n; i++) {
System.out.print(q[i].value() + " ");
}
System.out.println();
}
c++;
if (num > 0 && c >= num) {
break;
}
}
solver.endSearch();
// Statistics
System.out.println();
System.out.println("Solutions: " + solver.solutions());
System.out.println("Failures: " + solver.failures());
System.out.println("Branches: " + solver.branches());
System.out.println("Wall time: " + solver.wall_time() + "ms");
}
/** Solves the Coins Grid problm. See http://www.hakank.org/google_or_tools/coins_grid.py */
private static void solve() {
Solver solver = new Solver("CoinsGrid");
//
// data
//
int n = 31;
int c = 14;
//
// variables
//
IntVar[][] x = new IntVar[n][n];
IntVar[] x_flat = new IntVar[n * n];
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
x[i][j] = solver.makeIntVar(0, 1, "x[" + i + "," + j + "]");
x_flat[i * n + j] = x[i][j];
}
}
//
// constraints
//
// sum row/columns == c
for (int i = 0; i < n; i++) {
IntVar[] row = new IntVar[n];
IntVar[] col = new IntVar[n];
for (int j = 0; j < n; j++) {
row[j] = x[i][j];
col[j] = x[j][i];
}
solver.addConstraint(solver.makeSumEquality(row, c));
solver.addConstraint(solver.makeSumEquality(col, c));
}
System.out.println("here1");
// quadratic horizonal distance
IntVar[] obj_tmp = new IntVar[n * n];
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
obj_tmp[i * n + j] = solver.makeProd(x[i][j], (i - j) * (i - j)).Var();
}
}
IntVar obj_var = solver.makeSum(obj_tmp).Var();
//
// objective
//
OptimizeVar obj = solver.makeMinimize(obj_var, 1);
//
// search
//
DecisionBuilder db =
solver.makePhase(x_flat, solver.CHOOSE_FIRST_UNBOUND, solver.ASSIGN_MAX_VALUE);
solver.newSearch(db, obj);
//
// output
//
while (solver.nextSolution()) {
System.out.println("obj_var: " + obj_var.value());
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
System.out.print(x[i][j].value() + " ");
}
System.out.println();
}
System.out.println();
}
solver.endSearch();
// Statistics
System.out.println();
System.out.println("Solutions: " + solver.solutions());
System.out.println("Failures: " + solver.failures());
System.out.println("Branches: " + solver.branches());
System.out.println("Wall time: " + solver.wall_time() + "ms");
}