leetcode 791. 自定义字符串排序

题目描述

字符串S和 T 只包含小写字符。在S中,所有字符只会出现一次。
S 已经根据某种规则进行了排序。我们要根据S中的字符顺序对T进行排序。更具体地说,如果S中x在y之前出现,那么返回的字符串中x也应出现在y之前。
返回任意一种符合条件的字符串T。

示例:
输入:
S = “cba”
T = “abcd”
输出: “cbad”
解释:
S中出现了字符 “a”, “b”, “c”, 所以 “a”, “b”, “c” 的顺序应该是 “c”, “b”, “a”.
由于 “d” 没有在S中出现, 它可以放在T的任意位置. “dcba”, “cdba”, “cbda” 都是合法的输出。

注意:

  1. S的最大长度为26,其中没有重复的字符。
  2. T的最大长度为200。
  3. S和T只包含小写字符。

题解

找出每个字母在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, "")
}