Leetcode 791. Custom Sort String
Posted SnailTyan
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Leetcode 791. Custom Sort String相关的知识,希望对你有一定的参考价值。
文章作者:Tyan
博客:noahsnail.com | CSDN | 简书
1. Description
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
以上是关于Leetcode 791. Custom Sort String的主要内容,如果未能解决你的问题,请参考以下文章
791. Custom Sort String - LeetCode
[leetcode]791. Custom Sort String自定义排序字符串
Leetcode 791. Custom Sort String