Leetcode Minimum Size Subarray Sum

题意:找出最短的连续数列,使其各项之和大于给定值。
思路:先判断是否存在这样的数列。枚举以每一个元素结尾的满足条件的最短数列,枚举的时候保持一个指针,指向满足条件数列开始的位置。
class Solution {
public:
    int minSubArrayLen(int s, vector<int>& nums) {
        vector <int> p(nums.size(), -1);
        int sum = 0;
        int st = 0;
        for(int i = 0; i < nums.size(); ++ i) {
            sum += nums[i];
        }
        if(sum < s) return 0;
        
        sum = 0;
        for(int i = 0; i < nums.size(); ++ i) {
            sum += nums[i];
            if(sum < s) continue;
            else {
                while(sum >= s) {
                    sum -= nums[st];
                    p[i] = st;
                    st ++;
                }
                //st --;
            }
        }
        int shortest = nums.size();
        for(int i = 0; i < nums.size(); ++ i) { //cout << p[i] << endl;
            if(p[i] != -1) {
                if(shortest > i - p[i]) shortest = i - p[i] + 1;
            }
        }
        
        return shortest;
    }
};

猜你喜欢

转载自blog.csdn.net/markpen/article/details/60762356