力扣貪心 1326. 灌溉花園的最少水龍頭數目

在 x 軸上有一個一維的花園。花園長度爲 n，從點 0 開始，到點 n 結束。

花園裏總共有 n + 1 個水龍頭，分別位於 [0, 1, ..., n] 。

給你一個整數 n 和一個長度爲 n + 1 的整數數組 ranges ，其中 ranges[i] （下標從 0 開始）表示：如果打開點 i 處的水龍頭，可以灌溉的區域爲 [i -  ranges[i], i + ranges[i]] 。

請你返回可以灌溉整個花園的 最少水龍頭數目 。如果花園始終存在無法灌溉到的地方，請你返回 -1 。

示例 1：

輸入：n = 5, ranges = [3,4,1,1,0,0]
輸出：1
解釋：
點 0 處的水龍頭可以灌溉區間 [-3,3]
點 1 處的水龍頭可以灌溉區間 [-3,5]
點 2 處的水龍頭可以灌溉區間 [1,3]
點 3 處的水龍頭可以灌溉區間 [2,4]
點 4 處的水龍頭可以灌溉區間 [4,4]
點 5 處的水龍頭可以灌溉區間 [5,5]
只需要打開點 1 處的水龍頭即可灌溉整個花園 [0,5] 。
示例 2：

輸入：n = 3, ranges = [0,0,0,0]
輸出：-1
解釋：即使打開所有水龍頭，你也無法灌溉整個花園。
示例 3：

輸入：n = 7, ranges = [1,2,1,0,2,1,0,1]
輸出：3
示例 4：

輸入：n = 8, ranges = [4,0,0,0,0,0,0,0,4]
輸出：2
示例 5：

輸入：n = 8, ranges = [4,0,0,0,4,0,0,0,4]
輸出：1
 

提示：

1 <= n <= 10^4
ranges.length == n + 1
0 <= ranges[i] <= 100

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/minimum-number-of-taps-to-open-to-water-a-garden
著作權歸領釦網絡所有。商業轉載請聯繫官方授權，非商業轉載請註明出處。

Java:

class Solution {
    class Segment{
        int left, right;
        int num;
        Segment(int l, int r, int s){
            left = l;
            right = r;
            num = s;
        }
        public int getLeft(){
            return left;
        }
        public void setLeft(int left){
            this.left = left;
        }

        public int getRight(){
            return this.right;
        }
        public void setRight(int right){
            this.right = right;
        }

        public int getNum(){
            return this.num;
        }
        public void setNum(int num){
            this.num = num;
        }
    }
    public int minTaps(int n, int[] ranges) {
        //int dp[] = new dp[n+1];
        ArrayList<Segment> arrayList = new ArrayList<Segment>();
        for(int i = 0; i <= n; i++){
            int l = i - ranges[i] < 0 ? 0 : i - ranges[i];
            int r = i + ranges[i] > n ? n : i + ranges[i];
            arrayList.add(new Segment(l,r,1));
        }
        arrayList.sort(new Comparator<Segment>() {
            @Override
            public int compare(Segment o1, Segment o2) {
                if (o1.getLeft() == o2.getLeft())
                    return o1.getRight()-o2.getRight();
                return o1.getLeft()-o2.getLeft();
            }
        });
        // for(Segment s:arrayList){
        //     System.out.println(s.getLeft()+":"+s.getRight());
        // }
        // return 1;
        if(arrayList.get(0).getLeft() != 0)
            return -1;
        if(arrayList.get(0).getRight()==n)
            return 1;
        int f = 11000;
        for(int i = 1 ; i <= n; i++){
            Segment o1 = arrayList.get(i);
            int left = o1.getLeft();
            int right = o1.getRight();
            int num = o1.getNum();
            int s = -1;
            if(left == 0){
                if(right == n){
                    f = 1;
                }
                continue;
            }
            num = 11000;
            for(int j = 0; j < i; j++){
                Segment o2 = arrayList.get(j);
                if(o2.getLeft()==0 && o2.getRight() >= left && right > o2.getRight() && o2.getNum()+1 < num){
                    num = o2.getNum()+1;
                    s = j;
                }
            }
            if(s != -1){
                o1.setLeft(0);
                o1.setNum(num);
                if(right == n)
                    f = f > num ? num : f;
            }
        }
        return f == 11000 ? -1 : f;
    }
}

c++

class Solution {
public:
    int minTaps(int n, vector<int>& ranges) {
        vector<int> prev(n + 1);
        iota(prev.begin(), prev.end(), 0);

        for (int i = 0; i <= n; ++i) {
            int l = max(i - ranges[i], 0);
            int r = min(i + ranges[i], n);
            prev[r] = min(prev[r], l);
        }

        vector<int> dp(n + 1, INT_MAX);
        dp[0] = 0;
        for (int i = 1; i <= n; ++i) {
            for (int j = prev[i]; j < i; ++j) {
                if (dp[j] != INT_MAX) {
                    dp[i] = min(dp[i], dp[j] + 1);
                }
            }
        }

        return (dp[n] == INT_MAX ? -1 : dp[n]);
    }
};