Leetcode 209. 长度最小的子数组( 尺取法 or 二分法)

题目链接

做法:由于数组内没有负数  O(n) :尺取法  O(nlog(n)): 二分法,如果有负数 需要 考虑multiset 一下 二分  求方案数 需要map记录。类似题博客:博客

尺取法:

class Solution {
public:
   int minSubArrayLen(int s, vector<int>& nums) {
	if(nums.size()==0) return 0;
	int l=0,r=0;
	int now=nums[0],ans=INT_MAX;
	//printf("r:%d %d\n",r,nums.size());
	int sz=nums.size();
	while(l<sz)
	{
		//printf("l:%d r:%d now:%d\n",l,r,now);
		if(now<s){
			if(r+1<nums.size())  now+=nums[++r];
			else break;
		}
		else{
			//printf("l:%d r:%d l:%d r:%d now:%d\n",l,r,l,r,now);
			ans=min(ans,r-l+1);
			now-=nums[l++];
		}
	}
	if(ans==INT_MAX) return 0;
	return ans;
}
};

二分法:

class Solution {
public:
   int minSubArrayLen(int s, vector<int>& nums) {
	if(nums.size() == 0) return 0;
	vector<int> sum;
	sum.resize(nums.size()+1);
	
	sum[1] = nums[0];
	for(int i = 1;i < nums.size(); ++i){
		sum[i + 1] = sum[i] + nums[i];
	}
	int ans = INT_MAX;
	for(int i = 0; i < nums.size(); ++i){
		int l = i , r = nums.size() - 1;
		
		//printf("l:%d r:%d\n",l,r);
		
		int res = INT_MAX;
		while(l <= r){
			int mid = l + r >> 1;
			if(sum[mid + 1] - sum[i] >= s) r = mid - 1 , res = mid;
			else l = mid + 1;
		} 
		//printf("i:%d res:%d\n",i,res);
		
		if(res!=INT_MAX)
		ans = min(ans, res - i + 1);
		
		
	}
	if(ans==INT_MAX) return 0;
	return ans;
}
};

猜你喜欢

转载自blog.csdn.net/qq_41286356/article/details/107000920