import choco.Problem;
import choco.integer.*;
import choco.Solver;
import choco.Solution;
import choco.search.*;

public class Magic {

    public static void main(String[] args) throws Exception {
	//int n = Integer.parseInt(args[0]);
	int n = 4;
	Problem pb = new Problem();

	IntDomainVar S[][]   = new IntDomainVar[n][n]; // the square
	IntDomainVar T[][]   = new IntDomainVar[n][n]; // the square
	IntDomainVar v[]     = new IntDomainVar[n*n];
	IntDomainVar A[]     = new IntDomainVar[n];
	int count = 0;
	int k = n*(n*n + 1)/2;


	for (int i=0;i<n;i++)
	    for (int j=0;j<n;j++){
		S[i][j] = pb.makeEnumIntVar("SQ[" + i + "," + j + "]",1,n*n);
		T[j][i] = S[i][j];
		v[count] = S[i][j];
		count++;
	    }

	pb.post(pb.allDifferent(v));
	for (int i=0;i<n;i++)
	    pb.post(pb.eq(k,pb.sum(S[i])));
	for (int i=0;i<n;i++)
	    pb.post(pb.eq(k,pb.sum(T[i])));
	for (int i=0;i<n;i++) A[i] = S[i][i];
	pb.post(pb.eq(k,pb.sum(A)));
	for (int i=0;i<n;i++) A[i] = S[i][n-i-1];
	pb.post(pb.eq(k,pb.sum(A)));

	S[3][1].setVal(15);
	S[3][2].setVal(14);


       	Solver s = pb.getSolver();       
	pb.solve(false);
	// get 1st solution

	for (int i=0;i<n;i++){
	    for (int j=0;j<n;j++)
		System.out.print(S[i][j].getVal() + " ");
	    System.out.println();
	}
	// display solution


	AbstractGlobalSearchLimit GS = (AbstractGlobalSearchLimit)s.getSearchSolver().limits.get(1);
	int nds = GS.getNbTot();
	System.out.println("nodes: " + nds);
	// stats
    }


}