LeetCode 791. 自定义字符串排序

字符串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只包含小写字符。

class Solution:
    def customSortString(self, S: str, T: str) -> str:
        value_index_dict = {v:i for i,v in enumerate(S)}
        index_value_dict = {i:v for i,v in enumerate(S)}

        in_S_list = []
        not_in_S_list = []
        for i,v in enumerate(T):
            if v in value_index_dict:
                in_S_list.append(value_index_dict[v])
            else:
                not_in_S_list.append(v)
        in_S_list = sorted(in_S_list)
        in_S_list = [index_value_dict[i] for i in in_S_list]
        return ''.join(in_S_list+not_in_S_list)