LeetCode 767. 重构字符串(思维+构造)

题目链接
Description

Solution
先将S串中出现的不同字母计数，设出现次数最多的字母出现了x次
发现当x - 1 > S.size() - x 时，不存在答案；其他情况下答案一定存在
因为出现次数越多的字母越难以满足不处在相邻的位置上，而我们现在统计到的出现次数最多的字母出现了x次，需要至少x-1个字母来填补间隔。

考虑特殊构造：我们开辟一个列数为x的矩阵，按照字母出现次数顺序依次一行行填入

因为是按照字母出现次数从大到小的顺序填字母，所以必然不可能存在同一种字母在列中相邻(1)

最后按照列从小到大的顺序依次将每列的字母按照行从小到大的顺序放入答案中

因为(1)的性质存在，所以答案字符串中一定不存在相邻两个字母相同的情况

时间复杂度$O(N)$

Code

class Solution {
public:
    struct node{
        int ch,num;
        inline bool operator <(const node&it) const{
            return num > it.num;
        }
    }str[26];
    string reorganizeString(string S) {
        int num[26], a[505];
        memset(num,0,sizeof(num));
        memset(a,0,sizeof(a));
        int n = S.size();
        for(int i = 0;i < 26;++i) str[i].ch = i, str[i].num = 0;
        for(int i = 0;i < n;++i) str[S[i]-'a'].num++;
        sort(str,str+26);
        string res = "";
        //不存在答案的情况
        if((n-str[0].num) < str[0].num-1) return res;

        int m = str[0].num;
        vector<int>vec[m];
        int op = 0;
        for(int i = 0;i < 26;++i) {
            if(str[i].num == 0) break;
            for(int j = 0;j < str[i].num;++j) {
                a[op++] = str[i].ch;
            }
        }op = 0;
        while(true) {
            bool flag = false;
            for(int i = 0;i < m;++i) {
                if(op == n){flag = true;break;}
                vec[i].push_back(a[op++]);
            }if(flag) break;
        }
        for(int i = 0;i < m;++i) {
            int sz = vec[i].size();for(int j = 0;j < sz;++j) res += ('a' + vec[i][j]);
        }return res;
    }
};