자료구조 검사 받고 싶습니다.(앞으로 3~4개 정도 더 올릴 예정입니다.)

Question

파이썬으로 자료구조 'Heap'을 구현해봤는데 혹시 부족하거나 틀린 부분 있으면 알려주시면 좋겠습니다.(GPT)는 못 믿겠어서 OKKY에 있는 개발자 고수분들에게 피드백을 받고 싶어서요...제가 아래에 있는 Heap Class를 만들었을 때 한 가지 제약을 걸어 두긴 했습니다.(똑같은 값을 가진 노드는 없다.)이 부분을 감안해서 봐주셨으면 좋겠습니다. 감사합니다. 앞으로 3개~4개 정도 더 올릴 예정입니다.class Heap: def __init__(self,data): self.list = [] self.list.append(None) self.list.append(data) def insert(self,data): self.list.append(data) inserted_idx = len(self.list) -1 # 딱히 self를 붙일 필요는 없다. while self.move_up(inserted_idx): # 클래스 내의 메서드에서 다른 메서드를 사용할 경우 self를 붙여야 함 parent_idx = inserted_idx // 2 # 아래 메서드의 속성을 가져와서 사용할 수 없으므로 초기화를 해야 한다. self.list[parent_idx] , self.list[inserted_idx] = self.list[inserted_idx] , self.list[parent_idx] inserted_idx = parent_idx # return을 할 필요가 없다. def move_up(self,inserted_idx): parent_idx = inserted_idx // 2 # 첫 번째 메서드에서 속성을 사용함 그래서 따로 초기화를 할 필요가 없다. if self.list[parent_idx] < self.list[inserted_idx]: return True else: return False def pop(self,data): if data not in self.list: a = print(f"{data}가 없습니다.") return a temp = self.list[-1] self.list[1] = self.list[-1] del temp delete_idx = 1 while self.move_down(delete_idx): left_child = delete_idx * 2 right_child = delete_idx * 2 + 1 # 왼쪽 자식만 있는 경우 if len(self.list) -1 < right_child: self.list[delete_idx], self.list[left_child] = self.list[left_child] , self.list[delete_idx] delete_idx = left_child # 오른쪽 자식만 있는 경우 elif len(self.list) -1 >= right_child: if self.list[left_child] < self.list[right_child]: self.list[right_child], self.list[delete_idx] = self.list[delete_idx], self.list[right_child] delete_idx = right_child else: self.list[left_child] , self.list[delete_idx] = self.list[delete_idx], self.list[left_child] delete_idx = left_child def move_down(self,delete_idx): left_child = delete_idx * 2 right_child = delete_idx * 2 +1 # 노드가 한 개 밖에 없을 때 if len(self.list) - 1 == 1: return None # 더 이상 노드가 없을 때 if len(self.list) -1 < left_child: return None # 왼쪽 자식 노드 밖에 없을 때 elif len(self.list) - 1 < right_child: if self.list[delete_idx] < self.list[left_child]: return True else: return False # 노드가 둘 다 있을 때 elif len(self.list) -1 >= right_child: if self.list[left_child] < self.list[right_child]: if self.list[right_child] > self.list[delete_idx]: return True else: return False else: if self.list[left_child] > self.list[delete_idx]: return True else: return False

kenu,전독시 · Answer

이 코드 테스트하는 코드는 있나요?import unittest class TestHeap(unittest.TestCase): def setUp(self): self.heap &#61; Heap(10) # Assuming 10 is the initial data for the heap def test_insert(self): self.heap.insert(5) self.heap.insert(15) self.heap.insert(20) self.assertEqual(self.heap.list[1], 10) # Check if the initial data is still at the root self.assertEqual(self.heap.list[2], 5) # Check if the inserted data is in the correct position self.assertEqual(self.heap.list[3], 15) self.assertEqual(self.heap.list[4], 20) def test_pop(self): self.heap.insert(5) self.heap.insert(15) self.heap.insert(20) self.heap.pop(15) self.assertNotIn(15, self.heap.list) # Check if the popped data is removed self.assertEqual(self.heap.list[1], 10) # Check if the heap is reorganized correctly self.assertEqual(self.heap.list[2], 5) self.assertEqual(self.heap.list[3], 20) def test_move_up(self): self.heap.insert(5) self.heap.insert(15) self.heap.insert(20) self.assertTrue(self.heap.move_up(3)) # Assuming 20 is moved up self.assertEqual(self.heap.list[1], 20) self.assertEqual(self.heap.list[2], 10) self.assertEqual(self.heap.list[3], 5) def test_move_down(self): self.heap.insert(5) self.heap.insert(15) self.heap.insert(20) self.assertTrue(self.heap.move_down(1)) # Assuming 10 is moved down self.assertEqual(self.heap.list[1], 5) self.assertEqual(self.heap.list[2], 10) self.assertEqual(self.heap.list[3], 20) if __name__ &#61;&#61; '__main__': unittest.main() from: https://www.phind.com/search?cache&#61;tjht0lfeu5o7ija12bq9r0tc이런 거요.

aadddsf · Answer

윗분 말씀대로 테스트를 하는게 낫겠어요아니면 코드를 보완해서 OJ에 제출하시던지

자료구조 검사 받고 싶습니다.(앞으로 3~4개 정도 더 올릴 예정입니다.)

2개의 답변

Q&A 목록