sorting - 排序时忽略 nil 元素

Question

为了对这样的结构进行排序，我实现了三种方法:

type Event struct {
    timeEnd interface{}
    size float64
}

func (s ByTime) Len() int {
    return len(s)
}
func (s ByTime) Swap(i, j int) {
    s[i], s[j] = s[j], s[i]
}

有时不是所有Events.timeEnd已经初始化，我想把这样的 Event s 到 slice 的后面并按 size 对它们进行排序字段。

func (s ByTime) Less(i, j int) bool {
    if s[i].timeEnd == nil || s[j].timeEnd == nil{
        return s[i].size < s[j].size
    }else {
        return s[i].timeEnd.(float64) < s[j].timeEnd.(float64)
    }
}

 a, b, c, d := Event{timeEnd:12.}, Event{timeEnd:nil, size:-10}, Event{timeEnd:56.}, Event{timeEnd:nil, size:2}
 queue := []*Event{}
 queue = append(queue, &a, &b, &c, &d) 

但结果我有:

[<nil> 12 56 <nil>] 

期待[12 56 <nil> <nil>]如何正确实现？

这里的例子在https://play.golang.org/p/FWU98npGtbU

Answer 1

在 Less 中，您不想比较 size，除非 both timeEnd 是 nil 然后你需要单独处理“只有一个 timeEnd 是 nil”的情况。所以你有四种情况要考虑，而不是两种:

func (s ByTime) Less(i, j int) bool {
    if s[i].timeEnd == nil && s[j].timeEnd == nil {
        return s[i].size < s[j].size
    } else if s[i].timeEnd == nil {
        return false
    } else if s[j].timeEnd == nil {
        return true
    } else {
        return s[i].timeEnd.(float64) < s[j].timeEnd.(float64)
    }
}