Python的Leetcode791解法

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/custom-sort-string
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

from collections import Counter


class Solution:
    def customSortString(self, S: str, T: str) -> str:

        # 一条语句来建立一个字典，方便快捷
        dict2 = dict(Counter(T))
        res = ""

        for i in S:
            if i in dict2:
                res += i*dict2[i]
                dict2.pop(i)

        # S中没有出现的字符，位置无所谓的了
        for j in dict2:
            res += j*dict2[j]
        return res


if __name__ == '__main__':
    S = "cba"
    T = "abcd"
    s2 = Solution()
    print(s2.customSortString(S, T))