code.go 605 B

12345678910111213141516171819202122232425262728293031323334353637
  1. package main
  2. import (
  3. "bytes"
  4. "fmt"
  5. "log"
  6. "os"
  7. )
  8. func getRanges(data []byte) [][]int {
  9. var ranges [][]int
  10. ids := bytes.Split(data, []byte(","))
  11. for _, id := range ids {
  12. var start, end int
  13. n, err := fmt.Sscanf(string(id), "%d-%d", &start, &end)
  14. if err != nil || n != 2 {
  15. log.Fatalf("Invalid range: %s", id)
  16. }
  17. ranges = append(ranges, []int{start, end})
  18. }
  19. return ranges
  20. }
  21. func main() {
  22. if len(os.Args) < 2 {
  23. log.Fatal("You need to specify a file!")
  24. }
  25. data, err := os.ReadFile(os.Args[1])
  26. if err != nil {
  27. log.Fatal(err)
  28. }
  29. ranges := getRanges(data)
  30. fmt.Println(ranges)
  31. }