Leetcode - Russian Doll Envelopes

weixin_33751566發表於2016-10-13

My code:

public class Solution {
    public int maxEnvelopes(int[][] envelopes) {
        if (envelopes == null || envelopes.length == 0 || envelopes[0].length == 0) {
            return 0;
        }

        Arrays.sort(envelopes, new Comparator<int[]>() {
            public int compare(int[] a, int[] b) {
                if (a[0] != b[0]) {
                    return a[0] - b[0];
                }
                else {
                    return b[1] - a[1];
                }
            }    
        });
        
        List<Integer> rights = new ArrayList<Integer>();
        for (int i = 0; i < envelopes.length; i++) {
            if (rights.size() == 0 || rights.get(rights.size() - 1) < envelopes[i][1]) {
                rights.add(envelopes[i][1]);
            }
            else {
                int begin = 0;
                int end = rights.size() - 1;
                while (begin <= end) {
                    int mid = begin + (end - begin) / 2;
                    if (rights.get(mid) < envelopes[i][1]) {
                        begin = mid + 1;
                    }
                    else if (rights.get(mid) > envelopes[i][1]) {
                        end = mid - 1;
                    }
                    else {
                        begin = mid;
                        break;
                    }
                }
                rights.set(begin, envelopes[i][1]);
            }
        }
        
        return rights.size();
    }
}

reference:
http://www.programcreek.com/2016/08/leetcode-russian-doll-envelopes-java/

這道題目沒能做出來。看了解法,還是很精彩的。
先按照width大小排序,升序。
再按照height大小排序,降序。

我們可以保證,下一個envelope,他的width一定是當前最大。
但是他的height,並不確定,可能很小,可能很大。
我們需要不斷更新這個右側。
如圖:

161212-c5cf92ecc54c2f79.jpg
FullSizeRender 3.jpg
161212-7840a3436856cad5.jpg

具體自己理解吧。

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

相關文章