go - 按给定 slice 对 slice 字符串进行排序/排序

Question

如何按另一个字符串片给出的顺序对一个字符串片进行排序。如果输入片段中没有该字符串，则忽略它。

animalsInput := []string{"cat", "bird", "zebra", "fox"}
animalsOrder := []string{"bird", "lion", "fox"}

//desired output
//{"bird", "fox", "cat", "zebra"}

Answer 1

实现此目的的一种方法是通过基于订单数组编写rank映射

rank:=map[string]int{}
for i, x:=range animalsOrder {
   rank[x]=i
}

然后在 rank中使用 sort:

sort.Slice(animalsInput,func(i,j int) bool {
  irank, ok:=rank[animalsInput[i]]
  if !ok {
      irank=len(animalsInput)
  }
  jrank, ok:=rank[animalsInput[j]]
  if !ok {
      jrank=len(animalsInput)
  }
  return irank<jrank
})