Add solution for Challenge 3 by Johrespi

challenge-3/submissions/Johrespi/solution-template.go

@@ -0,0 +1,85 @@
+package main
+
+import "fmt"
+
+type Employee struct {
+	ID     int
+	Name   string
+	Age    int
+	Salary float64
+}
+
+type Manager struct {
+	Employees []Employee
+}
+
+// AddEmployee adds a new employee to the manager's list.
+func (m *Manager) AddEmployee(e Employee) {
+	// TODO: Implement this method
+	m.Employees = append(m.Employees, e)
+}
+
+// RemoveEmployee removes an employee by ID from the manager's list.
+func (m *Manager) RemoveEmployee(id int) {
+	// TODO: Implement this method
+	if id < 0 {
+	    return
+	}
+
+	idx := -1
+	for i, e := range m.Employees{
+	    if e.ID == id{
+	        idx = i
+	        break
+	    }
+	}
+
+	if idx == -1{
+	    return
+	}
+
+	m.Employees = append(m.Employees[:idx], m.Employees[idx + 1:]...)
+
+}
+
+// GetAverageSalary calculates the average salary of all employees.
+func (m *Manager) GetAverageSalary() float64 {
+	// TODO: Implement this method
+	if len(m.Employees) == 0 {
+	    return 0.0
+	}
+
+	employees := len(m.Employees)
+	var total float64
+	for _ , e := range m.Employees {
+	    total += e.Salary
+	}
+
+	return total / float64(employees)
+}
+
+// FindEmployeeByID finds and returns an employee by their ID.
+func (m *Manager) FindEmployeeByID(id int) *Employee {
+	// TODO: Implement this method
+	for i , e := range m.Employees{
+	    if e.ID == id{
+            return &m.Employees[i]
+	    }
+	}
+
+	return nil
+}
+
+func main() {
+	manager := Manager{}
+	manager.AddEmployee(Employee{ID: 1, Name: "Alice", Age: 30, Salary: 70000})
+	manager.AddEmployee(Employee{ID: 2, Name: "Bob", Age: 25, Salary: 65000})
+	manager.RemoveEmployee(1)
+	averageSalary := manager.GetAverageSalary()
+	employee := manager.FindEmployeeByID(2)
+
+	fmt.Printf("Average Salary: %f\n", averageSalary)
+	if employee != nil {
+		fmt.Printf("Employee found: %+v\n", *employee)
+	}
+}