summaryrefslogtreecommitdiffstats
path: root/06-puzzle-challenge/allIntervalSeriesSolver.py
blob: 9f973fdd553fbab9863c9cc512e842d8145ced6a (plain)
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
#!/usr/bin/python
# -*- coding: utf-8 -*-

def solveIt(n):
    # Modify this code to run your puzzle solving algorithm
    
    # define the domains of all the variables (0..n-1)
    domains = [range(0,n)]*n
    
    # start a trivial depth first search for a solution
    sol = tryall([],domains)
    
    # prepare the solution in the specified output format
    # if no solution is found, put 0s
    outputData = str(n) + '\n'
    if sol == None:
        print 'no solution found.'
        outputData += ' '.join(map(str, [0]*n))+'\n'
    else: 
        outputData += ' '.join(map(str, sol))+'\n'
    
    return outputData


# this is a depth first search of all assignments
def tryall(assignment, domains):
    # base-case: if the domains list is empty, all values are assigned
    # check if it is a solution, return None if it is not
    if len(domains) == 0:
        if checkIt(assignment):
            return assignment
        else:
            return None
    
    # recursive-case: try each value in the next domain
    # if we find a solution return it. otherwise, try the next value
    else:
        for v in domains[0]:
            sol = tryall(assignment[:]+[v],domains[1:])
            if sol != None:
                return sol


# checks if an assignment is feasible
def checkIt(sol):
    n = len(sol)
    
    items = set(sol)
    if len(items) != n:
        return False
    
    deltas = set([abs(sol[i]-sol[i+1]) for i in range(0,n-1)])
    if len(deltas) != n-1:
        return False
    
    return True


import sys

if __name__ == "__main__":
    if len(sys.argv) > 1:
        try:
            n = int(sys.argv[1].strip())
        except:
            print sys.argv[1].strip(), 'is not an integer'
        print 'Solving Size:', n
        print(solveIt(n))

    else:
        print('This test requires an instance size.  Please select the size of problem to solve. (i.e. python allIntervalSeriesSolver.py 5)')