Add score counting

main.go

@@ -75,14 +75,19 @@ type Car struct {
 	Arrival int
 	X       int
 	Y       int
+
+	score int
 }
 
 func (c *Car) Update(r *Ride) {
 	c.moveTo(r.a, r.b)
-	if c.Arrival < r.s {
+	if c.Arrival <= r.s {
 		c.Arrival = r.s
+		// count bonus points for being on time
+		c.score += B
 	}
 	c.moveTo(r.x, r.y)
+	c.score += r.length()
 	c.Rides = append(c.Rides, r.ID)
 }
 
@@ -194,13 +199,16 @@ func solve() {
 	for assign() {
 	}
 
+	totalScore := 0
 	for _, c := range Cars {
 		fmt.Fprintf(output, "%d", len(c.Rides))
 		for _, ri := range c.Rides {
 			fmt.Fprintf(output, " %d", ri)
 		}
 		fmt.Fprintf(output, "\n")
+		totalScore += c.score
 	}
+	fmt.Printf("%d\n", totalScore)
 }
 
 func main() {