LeetCode每日一题——791. 自定义字符串排序

LeetCode每日一题系列

题目：791. 自定义字符串排序
难度：普通

---

---

题目

给定两个字符串 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)