TP 5/6 : Refactoring d'un composant logiciel (2/3 et 3/3)

private Solver makeQueens(int n) {
    var solver = new Solver();
    var queens = new Variable[n];
    for (int i = 0; i < n; i++) {
        queens[i] = solver.newVar("R" + i).domain(1, n);
        for (int j = 0; j < i; j++) {
            solver.addConstraint(queens[i], "<>", queens[j]);
            solver.addConstraint(queens[i], "+", i, "<>", queens[j], "+", j);
            solver.addConstraint(queens[i], "-", i, "<>", queens[j], "-", j);
        }
    }
    solver.reduceAndCheckIntervalsStrategy();
    return solver;
}

public Solver buildAssignmentProblem(int[][] costs) {
    var solver = new Solver();
    int nbTasks = costs.length;
    int nbAgents = costs[0].length;
    Variable[][] matrix = new Variable[nbTasks][nbAgents];
    
    // create nbTasks * nbAgents variables 0..1
    for (int t = 0; t < nbTasks; t++) {
        for (int a = 0; a < nbAgents; a++) {
            matrix[t][a] = solver.newVar("T" + t + "A" + a).domain(0, 1);
        }
    }
    
    // each task have one agent
    var zero = solver.newConstant(0);
    for (int t = 0; t < nbTasks; t++) {
        var sum = zero;
        for (int a = 0; a < nbAgents; a++) {
            sum = solver.expression2(sum, "+", matrix[t][a]);
        }
        solver.addConstraint(sum, "=", 1);
    }
    
    // each agent have at most one task
    for (int a = 0; a < nbAgents; a++) {
        var sum = zero;
        for (int t = 0; t < nbTasks; t++) {
            sum = solver.expression2(sum, "+", matrix[t][a]);
        }
        solver.addConstraint(sum, "<=", 1);
    }
    
    // compute cost
    var cost = zero;
    for (int t = 0; t < nbTasks; t++) {
        for (int a = 0; a < nbAgents; a++) {
            cost = solver.expression2(cost, "+", matrix[t][a], "*", costs[t][a]);
        }
    }
    var varCost = solver.newVar("cost");
    solver.addConstraint(cost, "=", varCost);
    solver.setVerbose(false);
    // Code a adapter pour minimiser cost
    ....
    ....
    return solver;
}