cs170-project1/main.py at main · anchitab/cs170-project1 · 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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
from ucs import *
from a_star import *

def main():
    mode = input('Welcome to 8-Puzzle Solver! Enter 1 to use an example puzzle or 2 to enter your own puzzle: \n')
    mode = mode.strip()
    puzzle = get_puzzle(mode)
    run_algorithm(puzzle)

def run_algorithm(puzzle):
    algo = input('Enter the algorithm you\'d like to run. Please enter (1) for '
    'Uniform Cost Search, (2) for A* with Misplaced Tile Heuristic, or (3) for '
    'A* with Manhattan heuristic: ')
    algo = int(algo.strip())
    if algo == 1:
        ucs(puzzle)
    elif algo == 2:
        a_star(puzzle, misplaced_heuristic)
    elif algo == 3:
        a_star(puzzle, manhattan_heuristic)
    else:
        print('Error')

def get_puzzle(mode):
    if mode == '1':
        difficulty = input('Enter difficulty (on a scale of 0-7) for your example puzzle: ')
        difficulty = int(difficulty.strip())
        if difficulty == 0:
            print('Depth 0 puzzle selected')
            return depth_0_test
        if difficulty == 1:
            print('Depth 2 puzzle selected')
            return depth_2_test
        elif difficulty == 2:
            print('Depth 4 puzzle selected')
            return depth_4_test
        elif difficulty == 3:
            print('Depth 8 puzzle selected')
            return depth_8_test
        elif difficulty == 4:
            print('Depth 12 puzzle selected')
            return depth_12_test
        elif difficulty == 5:
            print('Depth 16 puzzle selected')
            return depth_16_test
        elif difficulty == 6:
            print('Depth 20 puzzle selected')
            return depth_20_test
        elif difficulty == 7:
            print('Depth 24 puzzle selected')
            return depth_24_test
        else:
            print('Error')
            return None
    else:
        print('Please enter your valid 8-puzzle space separated using a 0 to represent blank space')
        rows = []
        rows.append(input('Enter your first row (space separated): '))
        rows.append(input('Enter your second row (space separated): '))
        rows.append(input('Enter your third row (space separated): '))

        for i in range(len(rows)):
            new_row = []
            for num in rows[i].strip().split():
                new_row.append(int(num))
            rows[i] = new_row
        return rows


depth_0_test = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 0]
]
depth_2_test = [
    [1, 2, 3],
    [4, 5, 6],
    [0, 7, 8]
]
depth_4_test = [
    [1, 2, 3],
    [5, 0, 6],
    [4, 7, 8]
]

depth_8_test = [
    [1, 3, 6],
    [5, 0, 2],
    [4, 7, 8]
]

depth_12_test = [
    [1, 3, 6],
    [5, 0, 7],
    [4, 8, 2]
]

depth_16_test = [
    [1, 6, 7],
    [5, 0, 3],
    [4, 8, 2]
]

depth_20_test = [
    [7, 1, 2],
    [4, 8, 5],
    [6, 3, 0]
]

depth_24_test = [
    [0, 7, 2],
    [4, 6, 1],
    [3, 5, 8]
]

if __name__ == '__main__':
    main()