[算法题]重排字符串

题目链接: 重排字符串

首先找出出现次数最多的字符和其出现次数, 如果该字符的出现次数大于 (字符串的长度 + 1) / 2, 那么该字符串必然不能重排, 如果该字符的出现次数小于等于 (字符串的长度 + 1) / 2, 是可以重排的, 重排方式如下图所示:

先以间隔的方式将出现次数最多的字符排完, 再依次排其他字符, 题解代码:

#include <iostream>
using namespace std;

int main()
{
    int n;
    string str;
    cin >> n >> str;
    
    int cnt[26] = {0}; //用于存储每个字母的出现次数
    char maxCh; //出现次数最多的字母
    int maxCnt = 0; //出现次数最多的字母的出现次数
    
    //统计str中每个字母的出现次数
    for(int i = 0; i < n; ++i)
    {
        //更新字母和出现次数
        if(++cnt[str[i] - 'a'] > maxCnt)
        {
            maxCh = str[i];
            maxCnt = cnt[str[i] - 'a'];
        }
    }

    char res[n]; //存放重排后的字符串
    //判断str是否可以重排
    if(maxCnt > (n + 1) / 2)
    {
        cout << "no" << endl;
    }
    else
    {
        cout << "yes" << endl;
        int i = 0;
        //先排maxCh
        for( ; i < n && maxCnt--; i += 2)
        {
            res[i] = maxCh;
        }
        //再排其他字母
        for(int j = 0; j < 26; ++j)
        {
            while(cnt[j] && j + 'a' != maxCh)
            {
                if(i >= n)
                {
                    i = 1;
                }
                res[i] = j + 'a';
                i += 2;
                cnt[j]--;
            }
        }
        //输出重排后的结果
        for(auto& ch : res)
        {
            cout << ch;
        }
        cout << endl;
    }
    return 0;
}