题目:
给定一个字符串S
,检查是否能重新排布其中的字母,使得两相邻的字符不同。
若可行,输出任意可行的结果。若不可行,返回空字符串。
示例 1:
输入: S = "aab" 输出: "aba"
示例 2:
输入: S = "aaab" 输出: ""
注意:
S
只包含小写字母并且长度在[1, 500]
区间内。
分析:
写一个NewChar类,里面包含字母的出现频数,和字母本身。用优先队列PriorityQueue来存储一个一个的NewChar,并自己写一个比较器,通过字母的频数降序排列,即构建一个大顶堆。之后两两输出,输出前两个大的,然后将它们两个对应的count频率-1,再次放入,继续输出……
这样输出是为了总能有一个字母可以把频率最多的字母隔开,优先队列是为了维持储存NewChar的集合总是可以降序输出。
代码:
class Solution {
public String reorganizeString(String S) {
int[] counts = new int[26];
for (int i = 0; i < S.length(); i++) {
counts[S.charAt(i)-'a'] ++;
}//各个字母对应出现的频率
PriorityQueue<NewChar> pq = new PriorityQueue<>(new Comparator<NewChar>() {
@Override
public int compare(NewChar o1, NewChar o2) {
return o2.count-o1.count;
}
});
for (int i = 0; i < 26; i++) {
if (counts[i] > 0 && counts[i] <= (S.length()+1)/2){
pq.add(new NewChar(counts[i],(char)(i+'a')));
}else if (counts[i] > (S.length()+1)/2){
return "";
}
}
String str = "";
while (pq.size() > 1){
NewChar c1 = pq.poll();
NewChar c2 = pq.poll();
if (str.length()==0 || c1.letter != str.charAt(str.length()-1)){
str += c1.letter;
str += c2.letter;
}
if (--c1.count > 0) pq.add(c1);
if (--c2.count > 0) pq.add(c2);
}
if (pq.size()>0)
str += pq.poll().letter;
return str;
}
static class NewChar{
int count;
char letter;
NewChar(int count, char letter){
this.count = count;
this.letter = letter;
}
}
}
注意:
创建PriorityQueue的时候一定要写一个比较器Comparator,因为NewChar是自己写的一个类,不写比较器的话程序自己不知道该如何排序,从而会报错
cannot be cast to java.lang.Comparable at java.util.PriorityQueue.siftUpCom