104 lines
1.7 KiB
Go
104 lines
1.7 KiB
Go
package twentytwo
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
)
|
|
|
|
type player struct {
|
|
deck []int
|
|
}
|
|
|
|
func (p *player) score() int {
|
|
score := 0
|
|
for i := 0; i < len(p.deck); i++ {
|
|
score += p.deck[i] * (len(p.deck) - i)
|
|
}
|
|
|
|
return score
|
|
}
|
|
|
|
type combat struct {
|
|
one, two player
|
|
}
|
|
|
|
func (c *combat) load(filename string) error {
|
|
file, err := os.Open(filename)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer file.Close()
|
|
|
|
one := false
|
|
two := false
|
|
scanner := bufio.NewScanner(file)
|
|
for scanner.Scan() {
|
|
if scanner.Text() == "Player 1:" {
|
|
one = true
|
|
continue
|
|
}
|
|
|
|
if scanner.Text() == "" {
|
|
one = false
|
|
two = false
|
|
continue
|
|
}
|
|
|
|
if scanner.Text() == "Player 2:" {
|
|
two = true
|
|
continue
|
|
}
|
|
|
|
card, err := strconv.Atoi(scanner.Text())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if one {
|
|
c.one.deck = append(c.one.deck, card)
|
|
} else if two {
|
|
c.two.deck = append(c.two.deck, card)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (c *combat) play() (round int) {
|
|
for {
|
|
round++
|
|
|
|
// play cards
|
|
played := []int{c.one.deck[0], c.two.deck[0]}
|
|
c.one.deck = c.one.deck[1:]
|
|
c.two.deck = c.two.deck[1:]
|
|
|
|
if played[0] > played[1] {
|
|
c.one.deck = append(c.one.deck, played[0], played[1])
|
|
} else if played[0] < played[1] {
|
|
c.two.deck = append(c.two.deck, played[1], played[0])
|
|
} else {
|
|
panic("TIES ARE LIKELY IN PART 2")
|
|
}
|
|
|
|
if len(c.one.deck) == 0 || len(c.two.deck) == 0 {
|
|
break
|
|
}
|
|
}
|
|
|
|
return round
|
|
}
|
|
|
|
// PartOne What is the winning player's score
|
|
func PartOne() string {
|
|
c := combat{}
|
|
if err := c.load("twentytwo/input.txt"); err != nil {
|
|
return err.Error()
|
|
}
|
|
|
|
rounds := c.play()
|
|
return fmt.Sprintf("The game lasted %d rounds with a score of %d - %d", rounds, c.one.score(), c.two.score())
|
|
}
|