跳转至

LeetCode: 791. 自定义字符串排序

1、题目描述

字符串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" 都是合法的输出。

注意:

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

2、解题思路

  • 将所有S中出现,T中也出现的字符取出来排序,其他字符放到后面
class Solution:
    def customSortString(self, S: str, T: str) -> str:
        set_s = set(S)
        inter = []
        not_inter = []
        for c in T:
            if c in set_s:
                inter.append(c)
            else:
                not_inter.append(c)

        return "".join(sorted(inter, key=S.index)) + "".join(not_inter)
  • 建立字符的映射关系
class Solution:
    def customSortString(self, S: str, T: str) -> str:
        import string
        mapping = {key: val for val, key in enumerate(string.ascii_lowercase)}
        for index, c in enumerate(S):
            mapping[c] = index

        return "".join(sorted(T, key=lambda x: mapping[x]))