TicketToWin/main.py at main · jhmccoy9/TicketToWin · 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
119
120
121
122
from connection import Connection
from color import Color
from color_graph import ColorGraph
from destination import Destination


if __name__ == "__main__":
    connections = [
        # gray connections
        Connection(Destination.SEATTLE, Destination.PORTLAND, Color.GRAY, 1),
        Connection(Destination.SEATTLE, Destination.VANCOUVER, Color.GRAY, 1),
        Connection(Destination.CALGARY, Destination.VANCOUVER, Color.GRAY, 3),
        Connection(Destination.CALGARY, Destination.SEATTLE, Color.GRAY, 4),
        Connection(Destination.CALGARY, Destination.HELENA, Color.GRAY, 4),
        Connection(Destination.LOS_ANGELES, Destination.LAS_VEGAS, Color.GRAY, 2),
        Connection(Destination.LOS_ANGELES, Destination.PHOENIX, Color.GRAY, 3),
        Connection(Destination.PHOENIX, Destination.SANTA_FE, Color.GRAY, 3),
        Connection(Destination.PHOENIX, Destination.EL_PASO, Color.GRAY, 3),
        Connection(Destination.SANTA_FE, Destination.DENVER, Color.GRAY, 2),
        Connection(Destination.SANTA_FE, Destination.EL_PASO, Color.GRAY, 2),
        Connection(Destination.SAULT_ST_MARIE, Destination.WINNIPEG, Color.GRAY, 6),
        Connection(Destination.DULUTH, Destination.OMAHA, Color.GRAY, 2),
        Connection(Destination.KANSAS_CITY, Destination.OMAHA, Color.GRAY, 1),
        Connection(Destination.KANSAS_CITY, Destination.OKLAHOMA_CITY, Color.GRAY, 2),
        Connection(Destination.DALLAS, Destination.OKLAHOMA_CITY, Color.GRAY, 2),
        Connection(Destination.DALLAS, Destination.HOUSTON, Color.GRAY, 1),
        Connection(Destination.NEW_ORLEANS, Destination.HOUSTON, Color.GRAY, 2),
        Connection(Destination.LITTLE_ROCK, Destination.OKLAHOMA_CITY, Color.GRAY, 2),
        Connection(Destination.LITTLE_ROCK, Destination.SAINT_LOUIS, Color.GRAY, 2),
        Connection(Destination.NASHVILLE, Destination.SAINT_LOUIS, Color.GRAY, 1),
        Connection(Destination.NASHVILLE, Destination.ATLANTA, Color.GRAY, 2),
        Connection(Destination.RALEIGH, Destination.ATLANTA, Color.GRAY, 2),
        Connection(Destination.RALEIGH, Destination.CHARLESTON, Color.GRAY, 2),
        Connection(Destination.RALEIGH, Destination.WASHINGTON, Color.GRAY, 2),
        Connection(Destination.RALEIGH, Destination.PITTSBURGH, Color.GRAY, 2),
        Connection(Destination.WASHINGTON, Destination.PITTSBURGH, Color.GRAY, 2),
        Connection(Destination.TORONTO, Destination.PITTSBURGH, Color.GRAY, 2),
        Connection(Destination.TORONTO, Destination.SAULT_ST_MARIE, Color.GRAY, 2),
        Connection(Destination.TORONTO, Destination.MONTREAL, Color.GRAY, 3),
        Connection(Destination.BOSTON, Destination.MONTREAL, Color.GRAY, 2),
        # red connections
        Connection(Destination.MIAMI, Destination.NEW_ORLEANS, Color.RED, 6),
        Connection(Destination.DENVER, Destination.OKLAHOMA_CITY, Color.RED, 4),
        Connection(Destination.SALT_LAKE_CITY, Destination.DENVER, Color.RED, 3),
        Connection(Destination.HELENA, Destination.OMAHA, Color.RED, 5),
        Connection(Destination.DALLAS, Destination.EL_PASO, Color.RED, 4),
        Connection(Destination.DULUTH, Destination.OMAHA, Color.RED, 3),
        Connection(Destination.BOSTON, Destination.NEW_YORK, Color.RED, 2),
        # blue connections
        Connection(Destination.PORTLAND, Destination.SALT_LAKE_CITY, Color.BLUE, 6),
        Connection(Destination.HELENA, Destination.WINNIPEG, Color.BLUE, 4),
        Connection(Destination.OMAHA, Destination.CHICAGO, Color.BLUE, 4),
        Connection(Destination.KANSAS_CITY, Destination.SAINT_LOUIS, Color.BLUE, 2),
        Connection(Destination.MONTREAL, Destination.NEW_YORK, Color.BLUE, 3),
        Connection(Destination.ATLANTA, Destination.MIAMI, Color.BLUE, 5),
        # orange connections
        Connection(
            Destination.SAN_FRANCISCO, Destination.SALT_LAKE_CITY, Color.ORANGE, 5
        ),
        Connection(Destination.LAS_VEGAS, Destination.SALT_LAKE_CITY, Color.ORANGE, 3),
        Connection(Destination.DENVER, Destination.KANSAS_CITY, Color.ORANGE, 4),
        Connection(Destination.HELENA, Destination.DULUTH, Color.ORANGE, 6),
        Connection(Destination.CHICAGO, Destination.PITTSBURGH, Color.ORANGE, 3),
        Connection(Destination.NEW_ORLEANS, Destination.ATLANTA, Color.ORANGE, 4),
        Connection(Destination.NEW_YORK, Destination.WASHINGTON, Color.ORANGE, 2),
        # green
        Connection(Destination.PORTLAND, Destination.SAN_FRANCISCO, Color.GREEN, 5),
        Connection(Destination.HELENA, Destination.DENVER, Color.GREEN, 4),
        Connection(Destination.EL_PASO, Destination.HOUSTON, Color.GREEN, 6),
        Connection(Destination.SAINT_LOUIS, Destination.CHICAGO, Color.GREEN, 2),
        Connection(Destination.SAINT_LOUIS, Destination.PITTSBURGH, Color.GREEN, 5),
        Connection(Destination.LITTLE_ROCK, Destination.NEW_ORLEANS, Color.GREEN, 3),
        Connection(Destination.PITTSBURGH, Destination.NEW_YORK, Color.GREEN, 2),
        # pink
        Connection(Destination.PORTLAND, Destination.SAN_FRANCISCO, Color.PINK, 5),
        Connection(Destination.HELENA, Destination.SALT_LAKE_CITY, Color.PINK, 3),
        Connection(Destination.DENVER, Destination.OMAHA, Color.PINK, 4),
        Connection(Destination.KANSAS_CITY, Destination.SAINT_LOUIS, Color.PINK, 2),
        Connection(Destination.DULUTH, Destination.TORONTO, Color.PINK, 6),
        Connection(Destination.MIAMI, Destination.CHARLESTON, Color.PINK, 4),
        Connection(Destination.SAN_FRANCISCO, Destination.LOS_ANGELES, Color.PINK, 3),
        # yellow
        Connection(Destination.SAN_FRANCISCO, Destination.LOS_ANGELES, Color.YELLOW, 3),
        Connection(Destination.SEATTLE, Destination.HELENA, Color.YELLOW, 6),
        Connection(Destination.SALT_LAKE_CITY, Destination.DENVER, Color.YELLOW, 3),
        Connection(Destination.EL_PASO, Destination.OKLAHOMA_CITY, Color.YELLOW, 5),
        Connection(Destination.NEW_ORLEANS, Destination.ATLANTA, Color.YELLOW, 4),
        Connection(Destination.SAINT_LOUIS, Destination.PITTSBURGH, Color.YELLOW, 4),
        Connection(Destination.NEW_YORK, Destination.BOSTON, Color.YELLOW, 2),
        # black
        Connection(Destination.LOS_ANGELES, Destination.EL_PASO, Color.BLACK, 6),
        Connection(Destination.DENVER, Destination.KANSAS_CITY, Color.BLACK, 4),
        Connection(Destination.WINNIPEG, Destination.DULUTH, Color.BLACK, 4),
        Connection(Destination.SAULT_ST_MARIE, Destination.MONTREAL, Color.BLACK, 5),
        Connection(Destination.CHICAGO, Destination.PITTSBURGH, Color.BLACK, 3),
        Connection(Destination.NEW_YORK, Destination.WASHINGTON, Color.BLACK, 2),
        Connection(Destination.NASHVILLE, Destination.RALEIGH, Color.BLACK, 3),
        # white
        Connection(Destination.CALGARY, Destination.WINNIPEG, Color.WHITE, 6),
        Connection(
            Destination.SAN_FRANCISCO, Destination.SALT_LAKE_CITY, Color.WHITE, 5
        ),
        Connection(Destination.PHOENIX, Destination.DENVER, Color.WHITE, 5),
        Connection(Destination.LITTLE_ROCK, Destination.NASHVILLE, Color.WHITE, 3),
        Connection(Destination.CHICAGO, Destination.SAINT_LOUIS, Color.WHITE, 2),
        Connection(Destination.CHICAGO, Destination.TORONTO, Color.WHITE, 4),
        Connection(Destination.PITTSBURGH, Destination.NEW_YORK, Color.WHITE, 2),
    ]

    graph = ColorGraph(connections)

    # example
    graph.generatePaths(
        [
            Destination.SAULT_ST_MARIE,
            Destination.OKLAHOMA_CITY,
            Destination.DALLAS,
            Destination.NEW_YORK,
            Destination.VANCOUVER,
            Destination.SANTA_FE,
        ],
    )