Combining Functions to Solve Complex Problems in Go

Welcome back, Go enthusiast! In our last lesson, we successfully navigated the intricacies of variable scope in Go. Today, we're elevating our skills. We'll learn how to combine multiple functions to tackle more complex problems. Picture it as constructing a multi-star engineer's toolkit; each tool is a function, and employing the right combination can assist you in crafting an architectural masterpiece.

Imagine embarking on a European trip; you need to compute the cost of your visits to various countries based on a budget. It seems daunting, doesn't it? But fear not; with Go functions, we can dissect this problem into smaller, more manageable tasks.

Before diving into function creation, let's consider some sample data to work with:

package main

import "fmt"

var travelBudget = 5000
var countryCosts = map[string]int{"France": 1200, "Italy": 1500, "Spain": 800, "Germany": 900, "Greece": 1100}

Here, we have a travel budget of 5000, and the costs of visiting France, Italy, Spain, Germany, and Greece are outlined in our map, countryCosts. These values will serve as the basis for crafting and demonstrating our functions.

First, let's focus on selecting which countries we can visit without exceeding our budget using the chooseCountries function:

func chooseCountries(budget int, costs map[string]int) []string {
    totalCost := 0
    chosenCountries := []string{}
    for country, cost := range costs {
        if totalCost+cost > budget {
            break
        }
        totalCost += cost
        chosenCountries = append(chosenCountries, country)
    }
    return chosenCountries
}

func main() {
    chosenCountries := chooseCountries(travelBudget, countryCosts)
    fmt.Println(chosenCountries) // Prints [France Italy Spain Germany]
}

This function iterates through each country and its associated cost, adding countries to our visit list until we reach our budget limit. This direct approach ensures that our travel experience is always within budgetary constraints.

Next, we will calculate the total cost of visiting the selected countries with the calculateCost function:

func calculateCost(countries []string, costs map[string]int) int {
    totalCost := 0
    for _, country := range countries {
        totalCost += costs[country]
    }
    return totalCost
}

func main() {
    chosenCountries := chooseCountries(travelBudget, countryCosts)
    totalCost := calculateCost(chosenCountries, countryCosts)
    fmt.Println(totalCost) // Prints 4400
}

After selecting our countries, this function calculates the total expenditure. By iterating through the list of chosen countries, it sums up their associated costs from our map, countryCosts, providing us with a clear tally of our planned expenses.