leetcode_767 重构字符串

leetcode_767 重构字符串

题目

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

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

示例 1:

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

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

思路

一：借助PriorityQueue
代码：

class Solution {
    public String reorganizeString(String S) {
        if(S.length()<2||S==null) return S;
        char[] chars=S.toCharArray();
        int[] nums=new int[26];
        int maxcount=0;
       //记录字符串中每个字母出现的次数
       for(char x:chars){
            nums[x-'a']++;
            maxcount=Math.max(maxcount,nums[x-'a']);
       }
       //如果最大的字母数大于数组的一半，那么他肯定就不能重组
       if(maxcount>(S.length()+1)/2) return "";
         PriorityQueue<Character> queue = new PriorityQueue<Character>(new Comparator<Character>() {
            public int compare(Character letter1, Character letter2) {
                return nums[letter2 - 'a'] - nums[letter1 - 'a'];
            }
        });
        //对26个字母进行遍历但是不对数组chars进行遍历是为了queue中有重复的字符
        for(int i=0;i<26;i++){
            if(nums[i]>0)
                queue.offer((char)(i+'a'));
        }
        StringBuffer res=new StringBuffer();
        while(queue.size()>1){
            char a=queue.poll();
            char b=queue.poll();
            res.append(a);
            res.append(b);

            nums[a-'a']--;
            nums[b-'a']--;

            if(nums[a-'a']>0)
                queue.offer(a);
            if(nums[b-'a']>0)
                queue.offer(b);          
        }
        if (queue.size() > 0) {
            res.append(queue.poll());
        }
    return res.toString();

    }
}

二：不借助PriotyQueue(),就是每一次都找最大值，然后每一次都加入最大值的数字和其余任意一个数字
代码：

class Solution {
    public String reorganizeString(String S) {
        if(S.length()<2||S==null) return S;
        char[] chars=S.toCharArray();
        int[] nums=new int[26];
        int maxcount=0;
       //记录字符串中每个字母出现的次数
       for(char x:chars){
            nums[x-'a']++;
            maxcount=Math.max(maxcount,nums[x-'a']);
       }
       if(maxcount>(S.length()+1)/2) return "";
       StringBuffer res=new StringBuffer();
       while(res.length()!=S.length()){
           int max=0;
           //找出当前的具有最大字母数的字母
           for(int i=0;i<26;i++){
               if(nums[i]>nums[max])
                    max=i;
           }
           //将这个字母加入
           res.append((char)(max+'a'));
           nums[max]--;
           //再放入一个不同的字母
           for(char x:chars){
               if(nums[x-'a']>0&&x!=(char)(max+'a')){
                res.append(x);
                nums[x-'a']--;
                break;
               }
           }
       }
       return res.toString();
    }
}

总结

1.PriorityQueue可以按照自己所定义的东西进行排序

PriorityQueue queue = new PriorityQueue(new Comparator() {
public int compare(Character letter1, Character letter2) {
return nums[letter2 - ‘a’] - nums[letter1 - ‘a’];
}
});

peek（） 返回队头元素但是不出队
poll（） 出队头元素
offer（）入队元素
isEmpty（）判断是否为空