| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768 |
- package main
- import (
- "bufio"
- "fmt"
- "log"
- "os"
- "strconv"
- "strings"
- )
- func readInput(file *os.File) ([][]int, []string) {
- scanner := bufio.NewScanner(file)
- var numbers [][]int
- var symbols []string
- for scanner.Scan() {
- line := scanner.Text()
- if line == "" {
- continue
- }
- if line[0] == '+' || line[0] == '*' {
- parts := strings.Split(line, " ")
- for _, part := range parts {
- if part == "" || part == " " {
- continue
- }
- symbols = append(symbols, strings.Trim(part, " "))
- }
- } else {
- var lineNumbers []int
- parts := strings.Split(line, " ")
- for _, part := range parts {
- if part == "" || part == " " {
- continue
- }
- i, err := strconv.Atoi(strings.Trim(part, " "))
- if err != nil {
- log.Fatalf("Bad input: %s", part)
- }
- lineNumbers = append(lineNumbers, i)
- }
- numbers = append(numbers, lineNumbers)
- }
- }
- return numbers, symbols
- }
- func main() {
- if len(os.Args) < 2 {
- log.Fatal("You need to specify a file!")
- }
- filePath := os.Args[1]
- file, err := os.Open(filePath)
- if err != nil {
- log.Fatalf("Failed to open %s!\n", filePath)
- }
- numbers, symbols := readInput(file)
- fmt.Println(numbers, symbols)
- }
|