go - 合并带连字符的字符串范围之外的区间

Question

我有一个这样的字符串： ports := []string{"1", "2-7", "12-1200", "10-500"}

我想用这个做一个整数集，输出应该是：

[]intSet{ 1, 2-7, 10-1200 }

哪里intSet是某种整数集，我可以轻松地从中删除和添加元素。

更新 1

intSet是一个列表sets。所以，2-7也是一个集合。

更新 2

这里最大的集合被合并。

例如

• “1”-> 1
• “2-7”-> 2-7
• "12-1200" & "10-500" => "10..12.....500....1200" -> 10-1200

因为它是一个set所以它包含一个独特的范围，一个覆盖整个集合的范围。

标签： godata-structures

Answer 1

package main

import (
    "fmt"
    "log"
    "strconv"
    "strings"
)

type intSet struct {
    start int
    end   int
}

func (s intSet) String() string {
    if s.start == s.end {
        return fmt.Sprintf("%d", s.start)
    }
    return fmt.Sprintf("%d-%d", s.start, s.end)
}

func (s intSet) in(i int) bool {
    return s.start <= i && i <= s.end
}

func (s *intSet) union(set intSet) {
    if set.start < s.start {
        s.start = set.start
    }

    if set.end > s.end {
        s.end = set.end
    }
}

func insert(set intSet, is []intSet) bool {
    for i, s := range is {
        if s.in(set.start) || s.in(set.end) {
            is[i].union(set)
            return true
        }
        //updated here with thankful to @mh-cbon
        if set.in(s.start) || set.in(s.end) {
            is[i].union(set)
            return true
        }
    }
    return false
}

func main() {
    var set []intSet
    ports := []string{"1", "2-7", "12-1200", "10-500", "9-5500"}
    for _, port := range ports {
        s := strings.Split(port, `-`)
        if len(s) < 1 || len(s) > 2 {
            log.Fatalln(`set cannot have multiple values or no value`)
        }
        start, err := strconv.Atoi(s[0])
        if err != nil {
            log.Fatalln(err)
        }
        end := start
        if len(s) == 2 {
            end, err = strconv.Atoi(s[1])
            if err != nil {
                log.Fatalln(err)
            }
        }

        temSet := intSet{
            start: start,
            end:   end,
        }
        if !insert(temSet, set) {
            set = append(set, temSet)
        }
    }

    fmt.Println(set) //[1 2-7 9-5500]
}

跑到这里