LeetCode第172场周赛: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:

扫描二维码关注公众号,回复: 9038340 查看本文章

输入: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

思路:按照每个水龙头能够覆盖的最右端位置进行排序,然后从前往后遍历,能省则省,其中一个需要处理的地方是覆盖范围的左端点小于0则按照0处理,右端点大于n按照n处理。

class Solution {
	
	class node implements Comparable<node>{
		int val,id;
		public node(int val,int id) {
			this.val=val;
			this.id=id;
		}
		@Override
		public int compareTo(node o) {
			// TODO 自动生成的方法存根
			if(val==o.val)
				return id-o.id;
			return val-o.val;
		}
	}
	
    public int minTaps(int n, int[] ranges) {
        
    	Stack<node> q=new Stack<>();
    	node[] arr=new node[n+1];
    	for(int i=0;i<=n;i++)
    		arr[i]=new node(Math.min(n, i+ranges[i]),i);
    	Arrays.parallelSort(arr);
    	for(int i=0;i<=n;i++) {
    		int num=Math.max(0, arr[i].id-ranges[arr[i].id]);
    		while(!q.isEmpty()) {
    			node tmp=q.pop();
    			if(!q.isEmpty() && q.peek().val>=num || 
    					Math.max(0, tmp.id-ranges[tmp.id])>=num)
    				continue;
    			else {
    				q.add(tmp);
    				break;
    			}
    		}
    		if(q.isEmpty() || q.peek().val<arr[i].val) q.add(arr[i]);
    	}

    	if(q.peek().val<n) return -1;
    	int ans=q.size();
    	node x=q.pop();
    	while(!q.isEmpty()) {
    		System.out.println(x.id);
    		node y=q.pop();
    		if(x.id-ranges[x.id]>y.val) return -1;
    		x=y;
    	}
    	if(x.id-ranges[x.id]>0) return -1;
    	return ans;
    }
}
发布了987 篇原创文章 · 获赞 175 · 访问量 21万+

猜你喜欢

转载自blog.csdn.net/haut_ykc/article/details/104050330