Simplex/nqueens_lp_generator.py at main · Wronskode/Simplex · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
for n in range(8, 51):
    filetxt = ""
    filetxt += "min: z;\n"
    for j in range(1, n+1):
        s = ""
        for i in range(1, n+1):
            s = s + "q_"+str(i)+"_"+str(j)+" + "
        s = s[0:len(s)-3]
        s = s + " = 1;\n"
        filetxt += s

    for i in range(1, n+1):
        s = ""
        for j in range(1, n+1):
            s = s + "q_"+str(i)+"_"+str(j)+" + "
        s = s[0:len(s)-3]
        s = s + " = 1;\n"
        filetxt += s

    for d in range(-(n-2), n-1):
        s = ""
        for i in range(1, n+1):
            j = i - d
            if 1 <= j <= n:
                s += f"q_{i}_{j} + "
        if s:
            s = s[:-3]
            s += " <= 1;\n"
            filetxt += s

    for d in range(3, 2*n+1):
        s = ""
        for i in range(1, n+1):
            j = d - i
            if 1 <= j <= n:
                s += f"q_{i}_{j} + "
        if s:
            s = s[:-3]
            s += " <= 1;\n"
            filetxt += s

    # for j in range(1, n+1):
    #     for i in range(1, n+1):
    #         filetxt += "q_"+str(i)+"_"+str(j)+" <= 1;\n"

    for j in range(1, n+1):
        for i in range(1, n+1):
            filetxt += "int q_"+str(i)+"_"+str(j)+";\n"

    with open("instances/"+str(n)+"queens.lp", "w") as f:
        f.write(filetxt)