浏览代码

Let's find start

Piotr Czajkowski 1 周之前
父节点
当前提交
6adc950f33
共有 1 个文件被更改,包括 14 次插入4 次删除
  1. 14 4
      07/code.go

+ 14 - 4
07/code.go

@@ -7,9 +7,10 @@ import (
 	"os"
 	"os"
 )
 )
 
 
-func readInput(file *os.File) []string {
+func readInput(file *os.File) ([]string, int) {
 	scanner := bufio.NewScanner(file)
 	scanner := bufio.NewScanner(file)
 	var maze []string
 	var maze []string
+	start := -1
 
 
 	for scanner.Scan() {
 	for scanner.Scan() {
 		line := scanner.Text()
 		line := scanner.Text()
@@ -17,10 +18,19 @@ func readInput(file *os.File) []string {
 			continue
 			continue
 		}
 		}
 
 
+		if start < 0 {
+			for i := range line {
+				if line[i] == 'S' {
+					start = i
+					break
+				}
+			}
+		}
+
 		maze = append(maze, line)
 		maze = append(maze, line)
 	}
 	}
 
 
-	return maze
+	return maze, start
 }
 }
 
 
 func main() {
 func main() {
@@ -34,6 +44,6 @@ func main() {
 		log.Fatalf("Failed to open %s!\n", filePath)
 		log.Fatalf("Failed to open %s!\n", filePath)
 	}
 	}
 
 
-	maze := readInput(file)
-	fmt.Println(maze)
+	maze, start := readInput(file)
+	fmt.Println(maze, start)
 }
 }