-
Notifications
You must be signed in to change notification settings - Fork 0
/
an_extra_task.py
46 lines (38 loc) · 1.87 KB
/
an_extra_task.py
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
# Импорт модуля os из стандартной библиотеки
# для взаимодействия с операционной системой.
import os
# Считывание переменной окружения REMOTE_JUDGE, чтобы определить,
# локально выполняется код или на удалённом сервере.
LOCAL = os.environ.get('REMOTE_JUDGE', 'false') != 'true'
# Если код выполняется на локальном компьютере, то...
if LOCAL:
# Класс, описывающий элементы связного списка:
class Node:
def __init__(self, value, next_item=None):
self.value = value
self.next_item = next_item
def solution(node, idx):
# Напишите код функции здесь.
# ヽ(´▽`)/
if idx == 0:
return node.next_item
current = node
for i in range(idx - 1):
current = current.next_item
current.next_item = current.next_item.next_item
return node
# Тестирующая функция для проверки решения.
# Не изменяйте её, она не требует вашего внимания.
def test():
node3 = Node("Задача 4: Обследовать грунт в радиусе 3 м", None)
node2 = Node("Задача 3: Измерить температуру атмосферы", node3)
node1 = Node("Задача 2: Пробурить скважину глубиной 0.5 м", node2)
node0 = Node("Задача 1: Фотосъёмка 360°", node1)
new_head = solution(node0, 1)
assert new_head is node0
assert new_head.next_item is node2
assert new_head.next_item.next_item is node3
assert new_head.next_item.next_item.next_item is None
# result is node0 -> node2 -> node3
if __name__ == '__main__':
test()