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”也是有效的输出。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/custom-sort-string
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

---

• 字典dic用于统计每一个s中的字母在order中的下标，因为字典的键不能重复，所以对于重复出现在s中的字符不会被多次保存在字典dic中
• 因此又引入了一个字典count用于统计s中每一个字母出现的次数
• 借助字典dic按照value值从小到大排序就可以得到s中字母按order顺序排列
• 接下来就按照排好序的key，依次取出放入ans中，如果count[key]>1,就需要将key*count[key]放入ans中

---

代码

class Solution:
    def customSortString(self, order: str, s: str) -> str:
        n=len(s)
        ans=""
        dic={}
        count={}
        for x in s:
            dic[x]=order.find(x)
            count[x]=count.get(x,0)+1
        a=sorted(dic.items(),key=lambda x:x[1])
        a=dict(a)
        for key in a.keys():
            ans+=(key*count[key])
        return ans