LeetCode OJ:Minimum Size Subarray Sum(最小子数组的和)

Given an array of n positive integers and a positive integer s, find the minimal length of a subarray of which the sum ≥ s. If there isn't one, return 0 instead.

For example, given the array [2,3,1,2,4,3] and s = 7,
the subarray [4,3] has the minimal length under the problem constraint.

给一个数组以及一个数字,求满足大于该数字的最小的连续的数组元素个数的最小值。

代码写的比较乱。具体的思想就是用两个指针,一个先向前走, 当相加之和大于s的时候,将另一个指针也向前走,并减去相应的数字,当小于的时候将元素的个数存入数组,代码如下:

 1 class Solution {
 2 public:
 3     int minSubArrayLen(int s, vector<int>& nums) {
 4         int sz = nums.size();
 5         vector<int> ret;
 6         if(sz == 0) return 0;
 7         int i = 0;
 8         int j = 0;
 9         int tmpSum = 0;
10         while(j < sz){
11             for( ; i < sz; ++i){
12                 tmpSum += nums[i];
13                 if(tmpSum >= s)
14                     break;
15             }
16             if(tmpSum < s) break; //i已经达到数组的末尾了
17             for( ; j <= i; ++j){
18                 tmpSum -= nums[j];
19                 if(tmpSum < s)
20                     break;
21             }
22             ret.push_back(i - j + 1);
23             i++, j++;
24         }
25         sz = ret.size();
26         if(sz == 0) return 0;
27         int min = ret[0];
28         for(int i = 1; i < sz; ++i){
29             if(min > ret[i])
30                 min = ret[i];
31         }
32         return min;
33     }
34 };

 java版本代码如下所示,对上面做了一些改进,其实完全用不到上下两个循环的,双指针一次搞定:

 1 public class Solution {
 2     public int minSubArrayLen(int s, int[] nums) {
 3         if(nums.length == 0)
 4             return 0;
 5         int subSum = nums[0];
 6         int ret = Integer.MAX_VALUE;
 7         int p1 = 1, p2 = 0;
 8         while(p2 < p1){
 9             if(subSum < s){  //达不到k,指针前移动或者移动到头直接返回
10                 if(p1 < nums.length){
11                     subSum += nums[p1];
12                     p1++;
13                 }else{
14                     if(ret == Integer.MAX_VALUE)
15                         return 0;
16                     return ret;
17                 }
18             }else{      //达到k,后指针向前移动并且考虑是否更新指针。
19                 ret = Math.min(ret, p1-p2);
20                 subSum -= nums[p2];
21                 p2++;
22             }
23         }
24         if(ret == Integer.MAX_VALUE) //如果没有找到合适的子数组的话,直接返回0
25             return 0;
26         return ret;
27     }
28 }

 

新修改的方法为:

 1 class Solution {
 2 public:
 3     int minSubArrayLen(int s, vector<int>& nums) {
 4         int min = INT_MAX;
 5         int i = 0,j = 0;
 6         int currSum = 0;
 7         int sz = nums.size();
 8         while(currSum < s && j < sz){
 9             currSum += nums[j++];
10         }
11         if(j == sz)
12             return 0;
13         while(j != sz && i <= j){
14             if(currSum >= s)
15                 min = min(min, currSum);
16             while(i < j && currSum >= s){
17                 currSum -= nums[i++];
18             }
19             while(j != sz && currSum < s){
20                 currSum += nums[++j];
21             }
22         }
23         return min;
24     }
25 };

 

posted @ 2015-10-20 17:32  eversliver  阅读(287)  评论(0编辑  收藏  举报