LeetCode刷题记159-1326. 灌溉花园的最少水龙头数目

LeetCode刷题记159

1326. 灌溉花园的最少水龙头数目

题目

class Solution {
    public int minTaps(int n, int[] ranges) {
        if (n == 0) return 0;
        int[] sta = new int[n + 1];  // 记录有用的水龙头的下标
        int sta_id = 0;  // 有用的水龙头的最后一个
        int end = 0;
        for (int i = 0; i <= n; i ++) {
            if (i + ranges[i] <= end) {
                continue;  // 涵盖范围比之前小的跳过
            }
            if (sta_id > 0 && i - ranges[i] > sta[sta_id] + ranges[sta[sta_id]]) {
                continue;  // 不能接到上一个水龙头的跳过
            }
            while (sta_id > 0 && 
                i - ranges[i] <= (sta_id > 1 ? sta[sta_id - 1] + ranges[sta[sta_id - 1]] : 0)) {
                // 当前的水龙头可以覆盖上一个水龙头的有用范围
                sta_id --;  // 抛弃上一个水龙头
            }
            sta[++ sta_id] = i;  // 加入到有用的水龙头的数组里面
            end = i + ranges[i];  // 更新最远距离
        }
        while (sta_id > 1 && sta[sta_id - 1] + ranges[sta[sta_id - 1]] >= n) {
            sta_id --;  // 去掉多余的水龙头
        }
        return end >= n ? sta_id : -1;
    }
}

上面的贪心是O(N)的，下面来一个动态规划O(N*M)，M是数据range的范围，题目中是100。

class Solution {
    public int minTaps(int n, int[] ranges) {
        if (n == 0) return 0;
        int[] dp = new int[n + 1];
        Arrays.fill(dp, Integer.MAX_VALUE / 2);
        dp[0] = 0;
        for (int i = 0; i <= n; i ++) {
            int sta = Math.max(i - ranges[i], 0);
            int ed = Math.min(n, i + ranges[i]);
            for (int j = sta + 1; j <= ed; j ++) { 
                dp[j] = Math.min(dp[j], dp[sta] + 1);
            }
        }
        return dp[n] == Integer.MAX_VALUE / 2 ? -1 : dp[n];
    }
}