Copy Books

Note:html

這道題頗有意思,用的是二分法。想法是不停的試錯。這個博客主把這個問題說的很清楚 https://xuezhashuati.blogspot.com/2017/03/lintcode-437-copy-books.html函數

其實二分法的方法能夠變化的點頗有限,最多變化的就是檢查條件。通常難的題都是判斷條件複雜或者須要單獨寫一個函數進行斷定。若是用了二分法,若是通常的思惟方式比較難以理解,能夠考慮試錯。spa

 

public class Solution {
    /**
     * @param pages: an array of integers
     * @param k: an integer
     * @return: an integer
     */
    public int copyBooks(int[] pages, int k) {
        // write your code here
        if (pages == null || pages.length == 0) {
            return 0;
        }
        
        int start = pages[0]; //at least it needs max{pages[i]}
        int end = 0; //at most it needs sum of pages
        for (int i = 0; i < pages.length; i++) {
            end += pages[i];
            if (start < pages[i]) {
                start = pages[i];
            }
        }
        
        while (start + 1 < end) {
            int mid = start + (end - start) / 2;
            if (countCopier(pages, mid) > k) {
                start = mid;
            } else {
                end = mid;
            }
        }
        //System.out.println(start + " " + end);
        
        if (countCopier(pages, start) <= k) {
            return start;
        }
        
        return end;
    }
    
    private int countCopier(int[] pages, int min) {
        int sum = pages[0];
        int copier = 1;
        for (int i = 1; i < pages.length; i++) {
            if (sum + pages[i] > min) {
                copier++;
                sum = 0;
            }
            sum += pages[i];
        }
        return copier;
    }
}
相關文章
相關標籤/搜索