Find Two Non-overlapping Sub-arrays Each With Target Sum

3527 단어 PrefixSumhashmap
Given an array of integers  arr  and an integer  target .
You have to find two non-overlapping sub-arrays of  arr  each with sum equal  target . There can be multiple answers so you have to find an answer where the sum of the lengths of the two sub-arrays is minimum.
Return the minimum sum of the lengths of the two required sub-arrays, or return -1 if you cannot find such two sub-arrays.
Example 1:
Input: arr = [3,2,2,4,3], target = 3
Output: 2
Explanation: Only two sub-arrays have sum = 3 ([3] and [3]). The sum of their lengths is 2.

Example 2:
Input: arr = [7,3,4,7], target = 7
Output: 2
Explanation: Although we have three non-overlapping sub-arrays of sum = 7 ([7], [3,4] and [7]), but we will choose the first and third sub-arrays as the sum of their lengths is 2.

Example 3:
Input: arr = [4,3,2,6,2,3,4], target = 6
Output: -1
Explanation: We have only one sub-array of sum = 6.

Example 4:
Input: arr = [5,5,4,4,5], target = 3
Output: -1
Explanation: We cannot find a sub-array of sum = 3.

Example 5:
Input: arr = [3,1,1,1,5,1,2,1], target = 3
Output: 3
Explanation: Note that sub-arrays [1,2] and [2,1] cannot be an answer because they overlap.

Constraints:
  • 1 <= arr.length <= 10^5
  • 1 <= arr[i] <= 1000
  • 1 <= target <= 10^8

  • 사고방식: 이 사고방식은 약간 Subarray Sum Equals K와 유사합니다. 그 문제는hashmap으로 prefixsum,frequency를 저장합니다.여기는subarray를 두 개만 찾으면prefixsum과 index를 저장합니다.sum-target 대표가 한 구간을 찾으면 현재 dp[i]를 업데이트하고 dp[i]는 현재 i까지 얻을 수 있는subarraysum와 target을 위한length를 대표합니다.하면, 만약, 만약...1 && dp[pre] != Integer.MAX_VALUE, 대표가 두 개의 Subarray 구간을 찾았습니다.res를 업데이트합니다.
    여기서 주의하는 것은prefixsum index는 반드시 0,-1을 해시맵에 저장해야 합니다. 앞의 X 요소가 target인 상황에 대응하기 위해서입니다. 그러면length i-pre, 마침 = 렌
    class Solution {
        public int minSumOfLengths(int[] arr, int target) {
            if(arr == null || arr.length == 0) {
                return 0;
            }
            //     
            HashMap hashmap = new HashMap();
            hashmap.put(0, -1); //  , target , len ,
            int n = arr.length;
            int[] dp = new int[n]; // dp , i , target length;
            int sum = 0;
            int res = Integer.MAX_VALUE;
            for(int i = 0; i < n; i++) {
                sum += arr[i];
                dp[i] = i > 0 ? dp[i - 1] : Integer.MAX_VALUE;
                
                if(hashmap.containsKey(sum - target)) {
                    int pre = hashmap.get(sum - target);
                    //  target ;
                    dp[i] = Math.min(dp[i], i - pre);
                    
                    // if we find 2 subarry;
                    // if pre equals -1 means we only get one sub-array whoes sum eqauls target
                    if(pre != -1 && dp[pre] != Integer.MAX_VALUE) {
                        res = Math.min(res, dp[pre] + i - pre);
                    }
                }
                hashmap.put(sum, i);
            }
            return res == Integer.MAX_VALUE ? -1: res;
        }
    }

    좋은 웹페이지 즐겨찾기