[VI] Rooks

from constraint import *

def not_attacking(rook1,rook2):
    if rook1==rook2:
        return False
    return True

if __name__ == '__main__':

    problem = Problem()

    rooks= []
    domain= []
    for i in range(8):
        rooks.append(int(i))
        domain.append(int(i))

    problem.addVariables(rooks,domain)

    for i in range (8):
        for j in range(i+1,8):
            if i<j:
                problem.addConstraint(not_attacking,(rooks[i],rooks[j]))

    print(problem.getSolution())