leetcode767:重构字符串C++解答

题目

给定一个字符串S，检查是否能重新排布其中的字母，使得两相邻的字符不同。

若可行，输出任意可行的结果。若不可行，返回空字符串。

示例 1:

输入: S = “aab”
输出: “aba”
示例 2:

输入: S = “aaab”
输出: “”
注意:

S 只包含小写字母并且长度在[1, 500]区间内。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/reorganize-string

题解

这道题是典型的贪心算法的题。重构字符串时，需要根据每个字母在字符串中出现的次数处理每个字母放置的位置。如果出现次数最多的字母可以在重新排布之后不相邻，则可以重新排布字母使得相邻的字母都不相同。
当 n是偶数时，有 n/2个偶数下标和 n/2个奇数下标，因此每个字母的出现次数都不能超过 n/2次，否则出现次数最多的字母一定会出现相邻。
当 n 是奇数时，由于共有 (n+1)/2 个偶数下标，因此每个字母的出现次数都不能超过 (n+1)/2 次，否则出现次数最多的字母一定会出现相邻。
由于当 n 是偶数时，在整数除法下满足 n/2 和 (n+1)/2 相等，因此可以合并 n 是偶数与 n 是奇数的情况：如果可以重新排布成相邻的字母都不相同的字符串，每个字母最多出现 (n+1)/2(次。
因此首先遍历字符串并统计每个字母的出现次数，如果存在一个字母的出现次数大于 (n+1)/2，则无法重新排布字母使得相邻的字母都不相同，返回空字符串。如果所有字母的出现次数都不超过 (n+1)/2，则考虑如何重新排布字母。

解法1：先计数再排列

class Solution {
public:
    string reorganizeString(string S) {
        if (S.length() < 2) { //长度太短，不用排
            return S;
        }
        //求计数
        vector<int> counts(26, 0); //26个元素，全部初始化为0
        int maxCount = 0;
        int length = S.length();
        for (int i = 0; i < length; i++) {
            char c = S[i];
            counts[c - 'a']++; //用偏移量c - 'a'来表示小写字母
            maxCount = max(maxCount, counts[c - 'a']); //判断最大计数值
        }
        if (maxCount > (length + 1) / 2) { //如果最多的超过了(length + 1) / 2，则不能重新排列
            return "";
        }
        //排列
        string reorganizeArray(length, ' '); //初始化字符串，全为空
        int evenIndex = 0, oddIndex = 1;
        int halfLength = length / 2;
        for (int i = 0; i < 26; i++) {
            char c = 'a' + i; //恢复字符
            while (counts[i] > 0 && counts[i] <= halfLength && oddIndex < length) {
                reorganizeArray[oddIndex] = c; //先填奇数位置
                counts[i]--;
                oddIndex += 2;
            }
            while (counts[i] > 0) { //再填偶数位置
                reorganizeArray[evenIndex] = c;
                counts[i]--;
                evenIndex += 2;
            }
        }
        return reorganizeArray;
    }
};