计算机Leetcode, again

Leetcode - Rearrange String k Di

2016-10-10  本文已影响44人  Richardo92

My code:

public class Solution {
    public String rearrangeString(String str, int k) {
        if (str == null) {
            return "";
        }
        else if (k <= 1) {
            return str;
        }
        
        HashMap<Character, Integer> map = new HashMap<Character, Integer>();
        for (int i = 0; i < str.length(); i++) {
            char c = str.charAt(i);
            if (!map.containsKey(c)) {
                map.put(c, 1);
            }
            else {
                map.put(c, map.get(c) + 1);
            }
        }
        
        PriorityQueue<Character> pq = new PriorityQueue<Character>(10, new Comparator<Character>() {
            public int compare(Character c1, Character c2) {
                if (map.get(c1).compareTo(map.get(c2)) != 0) {
                    return map.get(c2) - map.get(c1);
                }
                else {
                    return c1.compareTo(c2);
                }
            }    
        });
        
        for (Character c : map.keySet()) {
            pq.offer(c);
        }
        
        StringBuilder sb = new StringBuilder();
        int len = str.length();
        while (!pq.isEmpty()) {
            int cnt = Math.min(k, len);
            List<Character> temp = new ArrayList<Character>();
            for (int i = 0; i < cnt; i++) {
                if (pq.isEmpty()) {
                    return "";
                }
                char curr = pq.poll();
                sb.append(curr);
                map.put(curr, map.get(curr) - 1);
                len--;
                if (map.get(curr) > 0) {
                    temp.add(curr);
                }
            }
            for (Character c : temp) {
                pq.offer(c);
            }
        }
        
        return sb.toString();
    }
}

reference:
http://www.programcreek.com/2014/08/leetcode-rearrange-string-k-distance-apart-java/

这道题目直接看的答案,用的
PriorityQueue + HashMap

pq 的优先级是,出现频率越大的字母,优先级越高。

然后我把所有字母插入pq,相同字母只插入一次。
第一次,按照距离至少是k来插入。
所以 cnt = Math.min(k, len);

然后把除了出现这次外,还会出现的字母,再次插入到queue中
以此类推。

Anyway, Good luck, Richardo! -- 10/09/2016

上一篇下一篇

猜你喜欢

热点阅读