Have regions

12/code.go

@@ -36,6 +36,63 @@ var directions [][]int = [][]int{
 	{0, -1}, {1, 0}, {0, 1}, {-1, 0},
 }
 
+func getPlots(current Point, matrix [][]byte, xMax, yMax int) []Point {
+	var plots []Point
+	for _, direction := range directions {
+		plot := Point{x: current.x + direction[0], y: current.y + direction[1]}
+		if plot.x >= 0 && plot.x < xMax &&
+			plot.y >= 0 && plot.y < yMax && matrix[plot.y][plot.x] == current.value {
+			plot.value = matrix[plot.y][plot.x]
+			plots = append(plots, plot)
+		}
+	}
+
+	return plots
+}
+
+func getRegion(start Point, matrix [][]byte, xMax, yMax int, checked map[string]bool) []Point {
+	plots := []Point{start}
+	var region []Point
+
+	for len(plots) > 0 {
+		current := plots[0]
+		plots = plots[1:]
+
+		if !checked[current.key()] {
+			region = append(region, current)
+		}
+
+		checked[current.key()] = true
+
+		newPlots := getPlots(current, matrix, xMax, yMax)
+		for _, plot := range newPlots {
+			if !checked[plot.key()] {
+				plots = append(plots, plot)
+			}
+		}
+	}
+
+	return region
+}
+
+func part1(matrix [][]byte) int {
+	xMax := len(matrix[0])
+	yMax := len(matrix)
+	checked := make(map[string]bool)
+
+	for y := range matrix {
+		for x := range matrix[y] {
+			current := Point{x: x, y: y, value: matrix[y][x]}
+			if !checked[current.key()] {
+				region := getRegion(current, matrix, xMax, yMax, checked)
+				fmt.Println(region)
+			}
+		}
+	}
+
+	return 0
+}
+
 func main() {
 	if len(os.Args) < 2 {
 		log.Fatal("You need to specify a file!")
@@ -48,5 +105,5 @@ func main() {
 	}
 
 	matrix := readInput(file)
-	fmt.Println(matrix)
+	fmt.Println(part1(matrix))
 }