LeetCode 791 自定义字符串排序 (Python)
题目描述如下:
字符串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
只包含小写字符。
思路:
- 创建list a用以依次存放字符串S中字母的AsicII码的值-97
- 创建长度为26的数组存放字符串T中每个字母出现的频率
- 按照list a中的字母顺序将T中的字母依次加入result中,若某字母出现两次,则加入两次
- 将T中有但S中没出现的字母再加入result中
代码如下:
from numpy import *
class Solution(object):
def customSortString(self, S, T):
"""
:type S: str
:type T: str
:rtype: str
"""
a=[]
b=zeros(26)
result=''
for i in range(0,len(S)):
a.append(ord(S[i])-97)
for i in range(0,len(T)):
index=ord(T[i])-97
b[index]=b[index]+1
for i in range(0,len(a)):
index=a[i]
while b[index]>0:
result=result+chr(a[i]+97)
b[index]=b[index]-1
for i in range(0,26):
while b[i]>0:
result=result+chr(i+97)
b[i]=b[i]-1
return result
: