LeetCode 1528. 重新排列字符串

给你一个字符串 s 和一个 长度相同 的整数数组 indices 。

请你重新排列字符串 s ,其中第 i 个字符需要移动到 indices[i] 指示的位置。

返回重新排列后的字符串。

输入:s = “codeleet”, indices = [4,5,6,7,0,2,1,3]
输出:“leetcode”
解释:“codeleet” 重新排列后变为 “leetcode” 。

法一:原地排序,时间复杂度为O(n),思路是遍历字符串s,每遍历到一个字符,查找该字符应该在的位置,之后交换该字符和它应该在的位置的字符,然后将indices中它应在的位置下标取负,代表该位置已经排好序了。之后记录下被交换的字符的下标,以查找该字符应在的下标,重复以上过程:

class Solution {
public:
    string restoreString(string s, vector<int>& indices) {
        int sz = s.size();
        for (int i = 0; i < sz; ++i) {
            int j = i;
            while (indices[j] > 0 && indices[j] != i) {
                swap(s[i], s[indices[j]]);
                indices[j] = -indices[j];    // 负数表示indices[j]处的元素已经是最终结果
                j = -indices[j];    // 记录当前i位置元素的原下标
            }
        }
        return s;
    }
};

法二:空间O(n)解法:

class Solution {
public:
    string restoreString(string s, vector<int>& indices) {
        string res;
        res.resize(s.size());
        for (int i = 0; i < s.size(); ++i) {
            res[indices[i]] = s[i];
        }
        return res;
    }
};

你可能感兴趣的:(LeetCode)