Piotr Czajkowski 1 hete
szülő
commit
727ebb771e
1 módosított fájl, 62 hozzáadás és 7 törlés
  1. 62 7
      10/code.go

+ 62 - 7
10/code.go

@@ -7,13 +7,9 @@ import (
 	"os"
 )
 
-var directions [][]int = [][]int{
-	{0, -1}, {1, 0}, {0, 1}, {-1, 0},
-}
-
 type Point struct {
-	x, y      int
-	direction int
+	x, y  int
+	value byte
 }
 
 func (p *Point) key() string {
@@ -36,6 +32,65 @@ func readInput(file *os.File) [][]byte {
 	return matrix
 }
 
+var directions [][]int = [][]int{
+	{0, -1}, {1, 0}, {0, 1}, {-1, 0},
+}
+
+func getMoves(reindeer Point, matrix [][]byte, xMax, yMax int) []Point {
+	var moves []Point
+	for _, direction := range directions {
+		move := Point{x: reindeer.x + direction[0], y: reindeer.y + direction[1]}
+		if move.x >= 0 && move.x < xMax &&
+			move.y >= 0 && move.y < yMax && matrix[move.y][move.x] > reindeer.value {
+			move.value = matrix[move.y][move.x]
+			moves = append(moves, move)
+		}
+	}
+
+	return moves
+}
+
+func hike(reindeer Point, matrix [][]byte, xMax, yMax int) int {
+	var nines int
+	visited := make(map[string]bool)
+
+	moves := []Point{reindeer}
+	for len(moves) > 0 {
+		current := moves[0]
+		moves = moves[1:]
+		if matrix[current.y][current.x] == '9' && !visited[current.key()] {
+			nines++
+		}
+
+		newMoves := getMoves(current, matrix, xMax, yMax)
+		for _, newMove := range newMoves {
+			if !visited[newMove.key()] {
+				moves = append(moves, newMove)
+				visited[newMove.key()] = true
+			}
+		}
+	}
+
+	return nines
+}
+
+func part1(matrix [][]byte) int {
+	var result int
+	xMax := len(matrix[0])
+	yMax := len(matrix)
+
+	for y := range matrix {
+		for x := range matrix[y] {
+			if matrix[y][x] == '0' {
+				reindeer := Point{x: x, y: y, value: '0'}
+				result += hike(reindeer, matrix, xMax, yMax)
+			}
+		}
+	}
+
+	return result
+}
+
 func main() {
 	if len(os.Args) < 2 {
 		log.Fatal("You need to specify a file!")
@@ -48,5 +103,5 @@ func main() {
 	}
 
 	matrix := readInput(file)
-	fmt.Println(matrix)
+	fmt.Println("Part1:", part1(matrix))
 }