+ // Ok, that was the important stuff. Now some fun stuff.
+
+ // Calculate total cost of fighters and shields
+ if *drones > 0 || *batteries > 0 {
+ fmt.Println()
+ }
+ if *drones > 0 {
+ final_state[BuyFighters] = 0
+ alt_best := FindBestState(data, dims, table, final_state)
+ cost := table[alt_best].value - table[best].value
+ fmt.Println("\rDrones were", float64(cost)/float64(*drones), "each")
+ final_state[BuyFighters] = 1
+ }
+ if *batteries > 0 {
+ final_state[BuyShields] = 0
+ alt_best := FindBestState(data, dims, table, final_state)
+ cost := table[alt_best].value - table[best].value
+ fmt.Println("\rBatteries were", float64(cost)/float64(*batteries), "each")
+ final_state[BuyShields] = 1
+ }
+