题目
给定两个字符串 order 和 s 。order 的所有单词都是 唯一 的,并且以前按照一些自定义的顺序排序。
对 s 的字符进行置换,使其与排序的 order 相匹配。更具体地说,如果在 order 中的字符 x 出现字符 y 之前,那么在排列后的字符串中, x 也应该出现在 y 之前。
返回 满足这个性质的 s 的任意排列 。
示例
示例 1:
输入: order = “cba”, s = “abcd”
输出: “cbad”
解释: “a”、“b”、“c”是按顺序出现的,所以“a”、“b”、“c”的顺序应该是“c”、“b”、“a”。因为“d”不是按顺序出现的,所以它可以在返回的字符串中的任何位置。“dcba”、“cdba”、“cbda”也是有效的输出。
示例 2:
输入: order = “cbafg”, s = “abcd”
输出: “cbad”
提示:
1 <= order.length <= 26
1 <= s.length <= 200
order 和 s 由小写英文字母组成
order 中的所有字符都 不同
思路
定义答案数组ans,先遍历order数组,判断order中的元素i是否在s中
- 如果在s中,计算s中i的数量,在ans中添加等数量的i,删除s中等数量的i
- 如果不在s中,直接跳过即可
以上操作可以将s中在order中的元素按照顺序排列好,最后只需要再加上s中剩下的元素即可
题解
class Solution: def customSortString(self, order: str, s: str) -> str: ans = "" s = list(s) for i in range(len(order)): # order元素在s中 if order[i] in s: # 添加等数量的元素 ans += order[i] * s.count(order[i]) # 删除s中等数量的元素 for j in range(s.count(order[i])): s.remove(order[i]) # 最后返回ans+s剩下的元素 return ans + ''.join(s)