字符串S和 T 只包含小写字符。在S中,所有字符只会出现一次。
S 已经根据某种规则进行了排序。我们要根据S中的字符顺序对T进行排序。更具体地说,如果S中x在y之前出现,那么返回的字符串中x也应出现在y之前。
返回任意一种符合条件的字符串T。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/custom-sort-string
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
python版本:
class Solution(object):
def customSortString(self, S, T):
return ‘’.join(sorted(list(T), key=lambda x:S.find(x)))
C++版本:利用sort第三个参数
class Comparator{
public:
Comparator(unordered_map<char, int>& m){
this->m = &m;
}
bool operator()(const char a, const char b) const{
int posA = (*m)[a];
int posB = (*m)[b];
return posA < posB;
}
private:
unordered_map<char, int>* m;
};
class Solution {
public:
string customSortString(string S, string T) {
unordered_map<char, int> m;
for(int i = 0; i<S.length(); ++i){
m[S[i]] = i;
}
sort(T.begin(), T.end(), Comparator(m));
return T;
}
};