Leetcode 791. Custom Sort String

Posted SnailTyan

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Leetcode 791. Custom Sort String相关的知识,希望对你有一定的参考价值。

文章作者:Tyan
博客:noahsnail.com  |  CSDN  |  简书

1. Description

Custom Sort String

2. Solution

**解析:**Version 1,Python3 3.6之后字典是有序的,因此先根据order构建一个有序字典,遍历str,将在order中的字符放到字典中,不存在的放到最终结果的前面,最后遍历字典,合并字符串。Version 2利用python自带的排序。

  • Version 1
class Solution:
    def customSortString(self, order: str, str: str) -> str:
        dictionary = {ch: [] for ch in order}
        ans = ''
        for ch in str:
            if ch in dictionary:
                dictionary[ch].append(ch)
            else:
                ans += ch
        for val in dictionary.values():
            ans += ''.join(val)
        return ans
  • Version 2
class Solution:
    def customSortString(self, order: str, str: str) -> str:
        mapping = {}
        for index, val in enumerate(order):
            mapping[val] = index
        s1 = ''
        s2 = ''
        for ch in str:
            if ch in mapping:
                s1 += ch
            else:
                s2 += ch
        return ''.join(sorted(s1, key=lambda ch: mapping[ch])) + s2

Reference

  1. https://leetcode.com/problems/custom-sort-string/

以上是关于Leetcode 791. Custom Sort String的主要内容,如果未能解决你的问题,请参考以下文章

791. Custom Sort String - LeetCode

[leetcode]791. Custom Sort String自定义排序字符串

Leetcode 791. Custom Sort String

LWC 73: 791. Custom Sort String

791. Custom Sort String

791. Custom Sort String