diff --git a/utils/utils.go b/utils/utils.go index 5c77b06..0d4ce34 100644 --- a/utils/utils.go +++ b/utils/utils.go @@ -46,6 +46,24 @@ func ParseIntArray(s string, sep string) []int { return result } +func ParseInt64Array(s string, sep string) []int64 { + result := []int64{} + + var vals []string + if sep == " " { + vals = strings.Fields(strings.TrimSpace(s)) + } else { + vals = strings.Split(strings.TrimSpace(s), sep) + } + + for _, val := range vals { + n, _ := strconv.ParseInt(val, 10, 64) + result = append(result, n) + } + + return result +} + func IntPow(x int, y int) int { result := 1