【Lintcode】1457. Search Subarray

题目地址：

https://www.lintcode.com/problem/search-subarray/description

给定一个数组$A$，再给定一个数$k$，要求找到一个和为$k$的子数组。如果有多个，则优先选结束位置最小的，在结束位置相同的情况下，优先选开始位置最小的。返回长度即可。

思路是前缀和 + 哈希表。为了节省空间，我们直接在$A$上算前缀和（即用$A[0]+A[1]+...+A[i]$来赋值给$A[i]$，不另外开数组了），然后把每个前缀和及其右端点出现的最左边的位置记录进哈希表。此外，空数组代表的前缀和$0$的位置定义为$-1$也一并加入哈希表（接下来会看到为什么）。接下来遍历$A$（此时$A$已经变成原数组的前缀和了），对于$A[i]$，我们看$A[i]-k$在哈希表中对应的值是多少，假设是$j$，如果$j<i$，就说明原数组中$[j+1:i]$这一段的和就是$k$，其长度是$i-j$；之所以要存一个$(0,-1)$这个entry进哈希表，原因是如果$[0:i]$这一段的和是$k$，如果有$(0,-1)$这个entry存在，就不会漏解。代码如下：

import java.util.HashMap;
import java.util.Map;

public class Solution {
    /**
     * @param arr: The array
     * @param k:   the sum
     * @return: The length of the array
     */
    public int searchSubarray(int[] arr, int k) {
        // Write your code here
        for (int i = 1; i < arr.length; i++) {
            arr[i] += arr[i - 1];
        }
        
        Map<Integer, Integer> map = new HashMap<>();        
        for (int i = arr.length - 1; i >= 0; i--) {
            map.put(arr[i], i);
        }
        map.put(0, -1);
        
        for (int i = 0; i < arr.length; i++) {
            if (map.containsKey(arr[i] - k) && i > map.get(arr[i] - k)) {
            	// 一旦找到，就可以返回了，这已经满足右端点最小，左端点最小的条件
                return i - map.get(arr[i] - k);
            }
        }
        
        return -1;
    }
}

时空复杂度$O(l_A)$。