week 2
import time
# Employee class
class Employee:
def __init__(self, emp_id, name, salary):
self.emp_id = emp_id
self.name = name
self.salary = salary
# Employee ADT class
class EmployeeADT:
def __init__(self):
self.employees = []
def add_employee(self, employee):
self.employees.append(employee)
def remove_employee(self, emp_id):
self.employees = [emp for emp in self.employees if emp.emp_id != emp_id]
def display_employees(self):
start = time.time()
for emp in self.employees: # O(n)
print(emp.emp_id, emp.name, emp.salary)
end = time.time()
print(f"Display employees (Time: {end - start:.6f}s)")
# Example usage ADT
print("-----Using Employee ADT-----")
adt = EmployeeADT()
adt.add_employee(Employee(1, "Alice", 50000))
adt.add_employee(Employee(2, "Bob", 60000))
adt.display_employees()
adt.remove_employee(1)
adt.display_employees()
Comments
Post a Comment