leetcode 791. 自定义字符串排序

题解

找出每个字母在S中的索引位置i,遍历T,与arr中相应位置的字符串合并

示例代码(go)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
func customSortString(S string, T string) string {
arr := make([]string, 26)
indexMap := make(map[rune]int)
for i, v := range S {
indexMap[v] = i
}
for _, v := range T {
if index, ok := indexMap[v]; ok {
arr[index] += string(v)
} else {
arr[25] += string(v)
}
}
return strings.Join(arr, "")
}