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
            
 

：