Medium

Minimum Size Subarray in Infinite ArrayC++

Full explanation · Time O(n) · Space O(1)

// Time:  O(n)
// Space: O(1)

// two pointers, sliding window
class Solution {
public:
    int minSizeSubarray(vector<int>& nums, int target) {
        static const int INF = numeric_limits<int>::max();

        const int64_t total = accumulate(cbegin(nums), cend(nums), 0ll);
        const int64_t q = target / total;
        target %= total;
        if (!target) {
            return q * size(nums);
        }
        int result = INF;
        int curr = 0;
        for (int right = 0, left = 0; right < (size(nums) - 1) + (size(nums) - 1); ++right) {
            curr += nums[right % size(nums)];
            while (curr > target) {
                curr -= nums[left++ % size(nums)];
            }
            if (curr == target) {
                result = min(result, right - left + 1);
            }
        }
        return result != INF ? result + q * size(nums) : -1;
    }
};


// Time:  O(n)
// Space: O(n)
// prefix sum, hash table
class Solution2 {
public:
    int minSizeSubarray(vector<int>& nums, int target) {
        static const int INF = numeric_limits<int>::max();

        const int64_t total = accumulate(cbegin(nums), cend(nums), 0ll);
        const int64_t q = target / total;
        target %= total;
        if (!target) {
            return q * size(nums);
        }
        int result = INF;
        unordered_map<int64_t, int> lookup = {{0, -1}};
        int64_t prefix = 0;
        for (int right = 0; right < (size(nums) - 1) + (size(nums) - 1); ++right) {
            prefix += nums[right % size(nums)];
            if (lookup.count(prefix - target)) {
                result = min(result, right - lookup[prefix - target]);
            }
            lookup[prefix] = right;
        }
        return result != INF ? result + q * size(nums) : -1;
    }
};