metalni

VI Prva Labaratoriska - 1

Jun 30th, 2021 (edited)
479
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 27.75 KB | None | 0 0
  1. import bisect
  2.  
  3. """
  4. Дефинирање на класа за структурата на проблемот кој ќе го решаваме со пребарување.
  5. Класата Problem е апстрактна класа од која правиме наследување за дефинирање на основните
  6. карактеристики на секој проблем што сакаме да го решиме
  7. """
  8.  
  9.  
  10. class Problem:
  11.     def __init__(self, initial, goal=None):
  12.         self.initial = initial
  13.         self.goal = goal
  14.  
  15.     def successor(self, state):
  16.         """За дадена состојба, врати речник од парови {акција : состојба}
  17.        достапни од оваа состојба. Ако има многу следбеници, употребете
  18.        итератор кој би ги генерирал следбениците еден по еден, наместо да
  19.        ги генерирате сите одеднаш.
  20.        :param state: дадена состојба
  21.        :return:  речник од парови {акција : состојба} достапни од оваа
  22.                  состојба
  23.        :rtype: dict
  24.        """
  25.         raise NotImplementedError
  26.  
  27.     def actions(self, state):
  28.         """За дадена состојба state, врати листа од сите акции што може да
  29.        се применат над таа состојба
  30.        :param state: дадена состојба
  31.        :return: листа на акции
  32.        :rtype: list
  33.        """
  34.         raise NotImplementedError
  35.  
  36.     def result(self, state, action):
  37.         """За дадена состојба state и акција action, врати ја состојбата
  38.        што се добива со примена на акцијата над состојбата
  39.        :param state: дадена состојба
  40.        :param action: дадена акција
  41.        :return: резултантна состојба
  42.        """
  43.         raise NotImplementedError
  44.  
  45.     def goal_test(self, state):
  46.         """Врати True ако state е целна состојба. Даденава имплементација
  47.        на методот директно ја споредува state со self.goal, како што е
  48.        специфицирана во конструкторот. Имплементирајте го овој метод ако
  49.        проверката со една целна состојба self.goal не е доволна.
  50.        :param state: дадена состојба
  51.        :return: дали дадената состојба е целна состојба
  52.        :rtype: bool
  53.        """
  54.         return state == self.goal
  55.  
  56.     def path_cost(self, c, state1, action, state2):
  57.         """Врати ја цената на решавачкиот пат кој пристигнува во состојбата
  58.        state2 од состојбата state1 преку акцијата action, претпоставувајќи
  59.        дека цената на патот до состојбата state1 е c. Ако проблемот е таков
  60.        што патот не е важен, оваа функција ќе ја разгледува само состојбата
  61.        state2. Ако патот е важен, ќе ја разгледува цената c и можеби и
  62.        state1 и action. Даденава имплементација му доделува цена 1 на секој
  63.        чекор од патот.
  64.        :param c: цена на патот до состојбата state1
  65.        :param state1: дадена моментална состојба
  66.        :param action: акција која треба да се изврши
  67.        :param state2: состојба во која треба да се стигне
  68.        :return: цена на патот по извршување на акцијата
  69.        :rtype: float
  70.        """
  71.         return c + 1
  72.  
  73.     def value(self):
  74.         """За проблеми на оптимизација, секоја состојба си има вредност.
  75.        Hill-climbing и сличните алгоритми се обидуваат да ја максимизираат
  76.        оваа вредност.
  77.        :return: вредност на состојба
  78.        :rtype: float
  79.        """
  80.         raise NotImplementedError
  81.  
  82.  
  83. """
  84. Дефинирање на класата за структурата на јазел од пребарување.
  85. Класата Node не се наследува
  86. """
  87.  
  88.  
  89. class Node:
  90.     def __init__(self, state, parent=None, action=None, path_cost=0):
  91.         """Креирај јазол од пребарувачкото дрво, добиен од parent со примена
  92.        на акцијата action
  93.        :param state: моментална состојба (current state)
  94.        :param parent: родителска состојба (parent state)
  95.        :param action: акција (action)
  96.        :param path_cost: цена на патот (path cost)
  97.        """
  98.         self.state = state
  99.         self.parent = parent
  100.         self.action = action
  101.         self.path_cost = path_cost
  102.         self.depth = 0  # search depth
  103.         if parent:
  104.             self.depth = parent.depth + 1
  105.  
  106.     def __repr__(self):
  107.         return "<Node %s>" % (self.state,)
  108.  
  109.     def __lt__(self, node):
  110.         return self.state < node.state
  111.  
  112.     def expand(self, problem):
  113.         """Излистај ги јазлите достапни во еден чекор од овој јазол.
  114.        :param problem: даден проблем
  115.        :return: листа на достапни јазли во еден чекор
  116.        :rtype: list(Node)
  117.        """
  118.  
  119.         return [self.child_node(problem, action)
  120.                 for action in problem.actions(self.state)]
  121.  
  122.     def child_node(self, problem, action):
  123.         """Дете јазел
  124.        :param problem: даден проблем
  125.        :param action: дадена акција
  126.        :return: достапен јазел според дадената акција
  127.        :rtype: Node
  128.        """
  129.         next_state = problem.result(self.state, action)
  130.         return Node(next_state, self, action,
  131.                     problem.path_cost(self.path_cost, self.state,
  132.                                       action, next_state))
  133.  
  134.     def solution(self):
  135.         """Врати ја секвенцата од акции за да се стигне од коренот до овој јазол.
  136.        :return: секвенцата од акции
  137.        :rtype: list
  138.        """
  139.         return [node.action for node in self.path()[1:]]
  140.  
  141.     def solve(self):
  142.         """Врати ја секвенцата од состојби за да се стигне од коренот до овој јазол.
  143.        :return: листа од состојби
  144.        :rtype: list
  145.        """
  146.         return [node.state for node in self.path()[0:]]
  147.  
  148.     def path(self):
  149.         """Врати ја листата од јазли што го формираат патот од коренот до овој јазол.
  150.        :return: листа од јазли од патот
  151.        :rtype: list(Node)
  152.        """
  153.         x, result = self, []
  154.         while x:
  155.             result.append(x)
  156.             x = x.parent
  157.         result.reverse()
  158.         return result
  159.  
  160.     """Сакаме редицата од јазли кај breadth_first_search или
  161.    astar_search да не содржи состојби - дупликати, па јазлите што
  162.    содржат иста состојба ги третираме како исти. [Проблем: ова може
  163.    да не биде пожелно во други ситуации.]"""
  164.  
  165.     def __eq__(self, other):
  166.         return isinstance(other, Node) and self.state == other.state
  167.  
  168.     def __hash__(self):
  169.         return hash(self.state)
  170.  
  171.  
  172. """
  173. Дефинирање на помошни структури за чување на листата на генерирани, но непроверени јазли
  174. """
  175.  
  176.  
  177. class Queue:
  178.     """Queue е апстрактна класа / интерфејс. Постојат 3 типа:
  179.        Stack(): Last In First Out Queue (стек).
  180.        FIFOQueue(): First In First Out Queue (редица).
  181.        PriorityQueue(order, f): Queue во сортиран редослед (подразбирливо,од најмалиот кон
  182.                                 најголемиот јазол).
  183.    """
  184.  
  185.     def __init__(self):
  186.         raise NotImplementedError
  187.  
  188.     def append(self, item):
  189.         """Додади го елементот item во редицата
  190.        :param item: даден елемент
  191.        :return: None
  192.        """
  193.         raise NotImplementedError
  194.  
  195.     def extend(self, items):
  196.         """Додади ги елементите items во редицата
  197.        :param items: дадени елементи
  198.        :return: None
  199.        """
  200.         raise NotImplementedError
  201.  
  202.     def pop(self):
  203.         """Врати го првиот елемент од редицата
  204.        :return: прв елемент
  205.        """
  206.         raise NotImplementedError
  207.  
  208.     def __len__(self):
  209.         """Врати го бројот на елементи во редицата
  210.        :return: број на елементи во редицата
  211.        :rtype: int
  212.        """
  213.         raise NotImplementedError
  214.  
  215.     def __contains__(self, item):
  216.         """Проверка дали редицата го содржи елементот item
  217.        :param item: даден елемент
  218.        :return: дали queue го содржи item
  219.        :rtype: bool
  220.        """
  221.         raise NotImplementedError
  222.  
  223.  
  224. class Stack(Queue):
  225.     """Last-In-First-Out Queue."""
  226.  
  227.     def __init__(self):
  228.         self.data = []
  229.  
  230.     def append(self, item):
  231.         self.data.append(item)
  232.  
  233.     def extend(self, items):
  234.         self.data.extend(items)
  235.  
  236.     def pop(self):
  237.         return self.data.pop()
  238.  
  239.     def __len__(self):
  240.         return len(self.data)
  241.  
  242.     def __contains__(self, item):
  243.         return item in self.data
  244.  
  245.  
  246. class FIFOQueue(Queue):
  247.     """First-In-First-Out Queue."""
  248.  
  249.     def __init__(self):
  250.         self.data = []
  251.  
  252.     def append(self, item):
  253.         self.data.append(item)
  254.  
  255.     def extend(self, items):
  256.         self.data.extend(items)
  257.  
  258.     def pop(self):
  259.         return self.data.pop(0)
  260.  
  261.     def __len__(self):
  262.         return len(self.data)
  263.  
  264.     def __contains__(self, item):
  265.         return item in self.data
  266.  
  267.  
  268. class PriorityQueue(Queue):
  269.     """Редица во која прво се враќа минималниот (или максималниот) елемент
  270.    (како што е определено со f и order). Оваа структура се користи кај
  271.    информирано пребарување"""
  272.     """"""
  273.  
  274.     def __init__(self, order=min, f=lambda x: x):
  275.         """
  276.        :param order: функција за подредување, ако order е min, се враќа елементот
  277.                      со минимална f(x); ако order е max, тогаш се враќа елементот
  278.                      со максимална f(x).
  279.        :param f: функција f(x)
  280.        """
  281.         assert order in [min, max]
  282.         self.data = []
  283.         self.order = order
  284.         self.f = f
  285.  
  286.     def append(self, item):
  287.         bisect.insort_right(self.data, (self.f(item), item))
  288.  
  289.     def extend(self, items):
  290.         for item in items:
  291.             bisect.insort_right(self.data, (self.f(item), item))
  292.  
  293.     def pop(self):
  294.         if self.order == min:
  295.             return self.data.pop(0)[1]
  296.         return self.data.pop()[1]
  297.  
  298.     def __len__(self):
  299.         return len(self.data)
  300.  
  301.     def __contains__(self, item):
  302.         return any(item == pair[1] for pair in self.data)
  303.  
  304.     def __getitem__(self, key):
  305.         for _, item in self.data:
  306.             if item == key:
  307.                 return item
  308.  
  309.     def __delitem__(self, key):
  310.         for i, (value, item) in enumerate(self.data):
  311.             if item == key:
  312.                 self.data.pop(i)
  313.  
  314.  
  315. import sys
  316.  
  317. """
  318. Неинформирано пребарување во рамки на дрво.
  319. Во рамки на дрвото не разрешуваме јамки.
  320. """
  321.  
  322.  
  323. def tree_search(problem, fringe):
  324.     """ Пребарувај низ следбениците на даден проблем за да најдеш цел.
  325.    :param problem: даден проблем
  326.    :type problem: Problem
  327.    :param fringe:  празна редица (queue)
  328.    :type fringe: FIFOQueue or Stack or PriorityQueue
  329.    :return: Node or None
  330.    :rtype: Node
  331.    """
  332.     fringe.append(Node(problem.initial))
  333.     while fringe:
  334.         node = fringe.pop()
  335.         print(node.state)
  336.         if problem.goal_test(node.state):
  337.             return node
  338.         fringe.extend(node.expand(problem))
  339.     return None
  340.  
  341.  
  342. def breadth_first_tree_search(problem):
  343.     """Експандирај го прво најплиткиот јазол во пребарувачкото дрво.
  344.    :param problem: даден проблем
  345.    :type problem: Problem
  346.    :return: Node or None
  347.    :rtype: Node
  348.    """
  349.     return tree_search(problem, FIFOQueue())
  350.  
  351.  
  352. def depth_first_tree_search(problem):
  353.     """Експандирај го прво најдлабокиот јазол во пребарувачкото дрво.
  354.    :param problem: даден проблем
  355.    :type problem: Problem
  356.    :return: Node or None
  357.    :rtype: Node
  358.    """
  359.     return tree_search(problem, Stack())
  360.  
  361.  
  362. """
  363. Неинформирано пребарување во рамки на граф
  364. Основната разлика е во тоа што овде не дозволуваме јамки,
  365. т.е. повторување на состојби
  366. """
  367.  
  368.  
  369. def graph_search(problem, fringe):
  370.     """Пребарувај низ следбениците на даден проблем за да најдеш цел.
  371.     Ако до дадена состојба стигнат два пата, употреби го најдобриот пат.
  372.    :param problem: даден проблем
  373.    :type problem: Problem
  374.    :param fringe:  празна редица (queue)
  375.    :type fringe: FIFOQueue or Stack or PriorityQueue
  376.    :return: Node or None
  377.    :rtype: Node
  378.    """
  379.     closed = set()
  380.     fringe.append(Node(problem.initial))
  381.     while fringe:
  382.         node = fringe.pop()
  383.         if problem.goal_test(node.state):
  384.             return node
  385.         if node.state not in closed:
  386.             closed.add(node.state)
  387.             fringe.extend(node.expand(problem))
  388.     return None
  389.  
  390.  
  391. def breadth_first_graph_search(problem):
  392.     """Експандирај го прво најплиткиот јазол во пребарувачкиот граф.
  393.    :param problem: даден проблем
  394.    :type problem: Problem
  395.    :return: Node or None
  396.    :rtype: Node
  397.    """
  398.     return graph_search(problem, FIFOQueue())
  399.  
  400.  
  401. def depth_first_graph_search(problem):
  402.     """Експандирај го прво најдлабокиот јазол во пребарувачкиот граф.
  403.    :param problem: даден проблем
  404.    :type problem: Problem
  405.    :return: Node or None
  406.    :rtype: Node
  407.    """
  408.     return graph_search(problem, Stack())
  409.  
  410.  
  411. def depth_limited_search(problem, limit=50):
  412.     """Експандирај го прво најдлабокиот јазол во пребарувачкиот граф
  413.    со ограничена длабочина.
  414.    :param problem: даден проблем
  415.    :type problem: Problem
  416.    :param limit: лимит за длабочината
  417.    :type limit: int
  418.    :return: Node or None
  419.    :rtype: Node
  420.    """
  421.  
  422.     def recursive_dls(node, problem, limit):
  423.         """Помошна функција за depth limited"""
  424.         cutoff_occurred = False
  425.         if problem.goal_test(node.state):
  426.             return node
  427.         elif node.depth == limit:
  428.             return 'cutoff'
  429.         else:
  430.             for successor in node.expand(problem):
  431.                 result = recursive_dls(successor, problem, limit)
  432.                 if result == 'cutoff':
  433.                     cutoff_occurred = True
  434.                 elif result is not None:
  435.                     return result
  436.         if cutoff_occurred:
  437.             return 'cutoff'
  438.         return None
  439.  
  440.     return recursive_dls(Node(problem.initial), problem, limit)
  441.  
  442.  
  443. def iterative_deepening_search(problem):
  444.     """Експандирај го прво најдлабокиот јазол во пребарувачкиот граф
  445.    со ограничена длабочина, со итеративно зголемување на длабочината.
  446.    :param problem: даден проблем
  447.    :type problem: Problem
  448.    :return: Node or None
  449.    :rtype: Node
  450.    """
  451.     for depth in range(sys.maxsize):
  452.         result = depth_limited_search(problem, depth)
  453.         if result is not 'cutoff':
  454.             return result
  455.  
  456.  
  457. def uniform_cost_search(problem):
  458.     """Експандирај го прво јазолот со најниска цена во пребарувачкиот граф.
  459.    :param problem: даден проблем
  460.    :type problem: Problem
  461.    :return: Node or None
  462.    :rtype: Node
  463.    """
  464.     return graph_search(problem, PriorityQueue(min, lambda a: a.path_cost))
  465.  
  466.  
  467. def up(x, y, obstacles, pos):
  468.     if y < 9 and (x, y + 1) not in obstacles and (x, y + 1) not in pos:
  469.         y += 1
  470.     return y
  471.  
  472.  
  473. def down(x, y, obstacles, pos):
  474.     if y > 0 and (x, y - 1) not in obstacles and (x, y - 1) not in pos:
  475.         y -= 1
  476.     return y
  477.  
  478.  
  479. def right(x, y, obstacles, pos):
  480.     if x < 9 and (x + 1, y) not in obstacles and (x + 1, y) not in pos:
  481.         x += 1
  482.     return x
  483.  
  484.  
  485. def left(x, y, obstacles, pos):
  486.     if x > 0 and (x - 1, y) not in obstacles and (x - 1, y) not in pos:
  487.         x -= 1
  488.     return x
  489.  
  490.  
  491. class Snake(Problem):
  492.  
  493.     def __init__(self, obstacles, initial, goal=None):
  494.         super().__init__(initial, goal)
  495.         self.obstacles = obstacles
  496.  
  497.     def successor(self, state):
  498.         successors = dict()
  499.         snake_pos = state[0]
  500.         head = snake_pos[-1]
  501.         head_x = head[0]
  502.         head_y = head[1]
  503.         greens = state[1]
  504.         ori = state[2]
  505.  
  506.         # Prodolzi pravo
  507.         if ori == 'dolu':
  508.             new_y = down(head_x, head_y, self.obstacles, snake_pos)
  509.             if head_y != new_y:
  510.                 tmp = list(snake_pos)
  511.                 if (head_x, new_y) in greens:
  512.                     tmp.append((head_x, new_y))
  513.                     successors['ProdolzhiPravo'] = (
  514.                         tuple(tmp), tuple([g for g in greens if g[0] != head_x or g[1] != new_y]), 'dolu')
  515.                 else:
  516.                     tmp.append((head_x, new_y))
  517.                     del tmp[0]
  518.                     successors['ProdolzhiPravo'] = (
  519.                         tuple(tmp), tuple([g for g in greens if g[0] != head_x or g[1] != new_y]), 'dolu')
  520.  
  521.         if ori == 'gore':
  522.             new_y = up(head_x, head_y, self.obstacles, snake_pos)
  523.             if head_y != new_y:
  524.                 tmp = list(snake_pos)
  525.                 if (head_x, new_y) in greens:
  526.                     tmp.append((head_x, new_y))
  527.                     successors['ProdolzhiPravo'] = (
  528.                         tuple(tmp), tuple([g for g in greens if g[0] != head_x or g[1] != new_y]), 'gore')
  529.                 else:
  530.                     tmp.append((head_x, new_y))
  531.                     del tmp[0]
  532.                     successors['ProdolzhiPravo'] = (
  533.                         tuple(tmp), tuple([g for g in greens if g[0] != head_x or g[1] != new_y]), 'gore')
  534.  
  535.         if ori == 'levo':
  536.             new_x = left(head_x, head_y, self.obstacles, snake_pos)
  537.             if head_x != new_x:
  538.                 tmp = list(snake_pos)
  539.                 if (new_x, head_y) in greens:
  540.                     tmp.append((new_x, head_y))
  541.                     successors['ProdolzhiPravo'] = (
  542.                         tuple(tmp), tuple([g for g in greens if g[0] != new_x or g[1] != head_y]), 'levo')
  543.                 else:
  544.                     tmp.append((new_x, head_y))
  545.                     del tmp[0]
  546.                     successors['ProdolzhiPravo'] = (
  547.                         tuple(tmp), tuple([g for g in greens if g[0] != new_x or g[1] != head_y]), 'levo')
  548.  
  549.         if ori == 'desno':
  550.             new_x = right(head_x, head_y, self.obstacles, snake_pos)
  551.             if head_x != new_x:
  552.                 tmp = list(snake_pos)
  553.                 if (new_x, head_y) in greens:
  554.                     tmp.append((new_x, head_y))
  555.                     successors['ProdolzhiPravo'] = (
  556.                         tuple(tmp), tuple([g for g in greens if g[0] != new_x or g[1] != head_y]), 'desno')
  557.                 else:
  558.                     tmp.append((new_x, head_y))
  559.                     del tmp[0]
  560.                     successors['ProdolzhiPravo'] = (
  561.                         tuple(tmp), tuple([g for g in greens if g[0] != new_x or g[1] != head_y]), 'desno')
  562.  
  563.         # Svrti desno
  564.         if ori == 'dolu':
  565.             new_x = left(head_x, head_y, self.obstacles, snake_pos)
  566.             if head_x != new_x:
  567.                 tmp = list(snake_pos)
  568.                 if (new_x, head_y) in greens:
  569.                     tmp.append((new_x, head_y))
  570.                     successors['SvrtiDesno'] = (
  571.                         tuple(tmp), tuple([g for g in greens if g[0] != new_x or g[1] != head_y]), 'levo')
  572.                 else:
  573.                     tmp.append((new_x, head_y))
  574.                     del tmp[0]
  575.                     successors['SvrtiDesno'] = (
  576.                         tuple(tmp), tuple([g for g in greens if g[0] != new_x or g[1] != head_y]), 'levo')
  577.         if ori == 'gore':
  578.             new_x = right(head_x, head_y, self.obstacles, snake_pos)
  579.             if head_x != new_x:
  580.                 tmp = list(snake_pos)
  581.                 if (new_x, head_y) in greens:
  582.                     tmp.append((new_x, head_y))
  583.                     successors['SvrtiDesno'] = (
  584.                         tuple(tmp), tuple([g for g in greens if g[0] != new_x or g[1] != head_y]), 'desno')
  585.                 else:
  586.                     tmp.append((new_x, head_y))
  587.                     del tmp[0]
  588.                     successors['SvrtiDesno'] = (
  589.                         tuple(tmp), tuple([g for g in greens if g[0] != new_x or g[1] != head_y]), 'desno')
  590.  
  591.         if ori == 'levo':
  592.             new_y = up(head_x, head_y, self.obstacles, snake_pos)
  593.             if head_y != new_y:
  594.                 tmp = list(snake_pos)
  595.                 if (head_x, new_y) in greens:
  596.                     tmp.append((head_x, new_y))
  597.                     successors['SvrtiDesno'] = (
  598.                         tuple(tmp), tuple([g for g in greens if g[0] != head_x or g[1] != new_y]), 'gore')
  599.                 else:
  600.                     tmp.append((head_x, new_y))
  601.                     del tmp[0]
  602.                     successors['SvrtiDesno'] = (
  603.                         tuple(tmp), tuple([g for g in greens if g[0] != head_x or g[1] != new_y]), 'gore')
  604.  
  605.         if ori == 'desno':
  606.             new_y = down(head_x, head_y, self.obstacles, snake_pos)
  607.             if head_y != new_y:
  608.                 tmp = list(snake_pos)
  609.                 if (head_x, new_y) in greens:
  610.                     tmp.append((head_x, new_y))
  611.                     successors['SvrtiDesno'] = (
  612.                         tuple(tmp), tuple([g for g in greens if g[0] != head_x or g[1] != new_y]), 'dolu')
  613.                 else:
  614.                     tmp.append((head_x, new_y))
  615.                     del tmp[0]
  616.                     successors['SvrtiDesno'] = (
  617.                         tuple(tmp), tuple([g for g in greens if g[0] != head_x or g[1] != new_y]), 'dolu')
  618.  
  619.         # Svrti levo
  620.         if ori == 'dolu':
  621.             new_x = right(head_x, head_y, self.obstacles, snake_pos)
  622.             if head_x != new_x:
  623.                 tmp = list(snake_pos)
  624.                 if (new_x, head_y) in greens:
  625.                     tmp.append((new_x, head_y))
  626.                     successors['SvrtiLevo'] = (
  627.                         tuple(tmp), tuple([g for g in greens if g[0] != new_x or g[1] != head_y]), 'desno')
  628.                 else:
  629.                     tmp.append((new_x, head_y))
  630.                     del tmp[0]
  631.                     successors['SvrtiLevo'] = (
  632.                         tuple(tmp), tuple([g for g in greens if g[0] != new_x or g[1] != head_y]), 'desno')
  633.  
  634.         if ori == 'gore':
  635.             new_x = left(head_x, head_y, self.obstacles, snake_pos)
  636.             if head_x != new_x:
  637.                 tmp = list(snake_pos)
  638.                 if (new_x, head_y) in greens:
  639.                     tmp.append((new_x, head_y))
  640.                     successors['SvrtiLevo'] = (
  641.                         tuple(tmp), tuple([g for g in greens if g[0] != new_x or g[1] != head_y]), 'levo')
  642.                 else:
  643.                     tmp.append((new_x, head_y))
  644.                     del tmp[0]
  645.                     successors['SvrtiLevo'] = (
  646.                         tuple(tmp), tuple([g for g in greens if g[0] != new_x or g[1] != head_y]), 'levo')
  647.  
  648.         if ori == 'levo':
  649.             new_y = down(head_x, head_y, self.obstacles, snake_pos)
  650.             if head_y != new_y:
  651.                 tmp = list(snake_pos)
  652.                 if (head_x, new_y) in greens:
  653.                     tmp.append((head_x, new_y))
  654.                     successors['SvrtiLevo'] = (
  655.                         tuple(tmp), tuple([g for g in greens if g[0] != head_x or g[1] != new_y]), 'dolu')
  656.                 else:
  657.                     tmp.append((head_x, new_y))
  658.                     del tmp[0]
  659.                     successors['SvrtiLevo'] = (
  660.                         tuple(tmp), tuple([g for g in greens if g[0] != head_x or g[1] != new_y]), 'dolu')
  661.         if ori == 'desno':
  662.             new_y = up(head_x, head_y, self.obstacles, snake_pos)
  663.             if head_y != new_y:
  664.                 tmp = list(snake_pos)
  665.                 if (head_x, new_y) in greens:
  666.                     tmp.append((head_x, new_y))
  667.                     successors['SvrtiLevo'] = (
  668.                         tuple(tmp), tuple([g for g in greens if g[0] != head_x or g[1] != new_y]), 'gore')
  669.                 else:
  670.                     tmp.append((head_x, new_y))
  671.                     del tmp[0]
  672.                     successors['SvrtiLevo'] = (
  673.                         tuple(tmp), tuple([g for g in greens if g[0] != head_x or g[1] != new_y]), 'gore')
  674.  
  675.         return successors
  676.  
  677.     def actions(self, state):
  678.         return self.successor(state).keys()
  679.  
  680.     def result(self, state, action):
  681.         return self.successor(state)[action]
  682.  
  683.     def goal_test(self, state):
  684.         return len(state[1]) == 0
  685.  
  686.  
  687. if __name__ == '__main__':
  688.     num_green = int(input())
  689.     green_apples = list()
  690.  
  691.     for j in range(num_green):
  692.         green_apples.append([int(i) for i in input().split(",")])
  693.  
  694.     for i in range(num_green):
  695.         green_apples[i] = tuple(green_apples[i])
  696.  
  697.     num_red = int(input())
  698.     red_apples = list()
  699.  
  700.     for j in range(num_red):
  701.         red_apples.append([int(i) for i in input().split(",")])
  702.  
  703.     for i in range(num_red):
  704.         red_apples[i] = tuple(red_apples[i])
  705.  
  706.     snake_position = ((0, 9), (0, 8), (0, 7))
  707.  
  708.     orientation = 'dolu'
  709.     snake = Snake(tuple(red_apples), (snake_position, tuple(green_apples), orientation))
  710.     res = breadth_first_graph_search(snake)
  711.     print(res.solution())
  712.  
Add Comment
Please, Sign In to add comment