67 lines
1.1 KiB
Go
67 lines
1.1 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"strings"
|
|
)
|
|
|
|
const high = rune('#')
|
|
|
|
type sForest struct {
|
|
data [][]bool
|
|
height int
|
|
width int
|
|
}
|
|
|
|
func (f sForest) at(y, x int) bool {
|
|
if y >= f.height {
|
|
return false
|
|
}
|
|
return f.data[y][x%f.width]
|
|
}
|
|
|
|
func (f sForest) traverse(y, x int) int {
|
|
trees := 0
|
|
for dy := 0; dy <= f.height; dy++ {
|
|
tree := f.at(dy*y, dy*x)
|
|
if tree {
|
|
trees++
|
|
}
|
|
}
|
|
return trees
|
|
}
|
|
|
|
func main() {
|
|
data, _ := ioutil.ReadFile("./solutions/3/data.txt")
|
|
|
|
rows := strings.Split(strings.TrimSpace(string(data)), "\n")
|
|
height, width := len(rows), len(rows[0])
|
|
d := make([][]bool, height)
|
|
|
|
for y, row := range rows {
|
|
d[y] = make([]bool, width)
|
|
for x, char := range []rune(row) {
|
|
d[y][x] = char == high
|
|
}
|
|
}
|
|
|
|
forest := sForest{
|
|
data: d,
|
|
height: height,
|
|
width: width,
|
|
}
|
|
|
|
fmt.Println("Simple: ", forest.traverse(1, 3))
|
|
|
|
trees11 := forest.traverse(1, 1)
|
|
trees13 := forest.traverse(1, 3)
|
|
trees15 := forest.traverse(1, 5)
|
|
trees17 := forest.traverse(1, 7)
|
|
trees21 := forest.traverse(2, 1)
|
|
|
|
fmt.Println(trees11, trees13, trees15, trees17, trees21)
|
|
fmt.Println(trees11 * trees13 * trees15 * trees17 * trees21)
|
|
|
|
}
|