范围和相等的最宽索引对

/*
题目：范围和相等的最宽索引对
 
题目思路：
1. 使用Stream API来实现相同的逻辑。
2. 使用一个哈希表来存储每个前缀和第一次出现的位置。
3. 使用IntStream进行遍历并计算前缀和。
4. 更新和计算最大宽度。
*/
 
import java.util.HashMap;
import java.util.stream.IntStream;
 
public class Solution {
    public int maxWidthRamp(int[] A) {
        HashMap<Integer, Integer> map = new HashMap<>();
        map.put(0, -1); // 初始化前缀和为0的位置为-1
        int[] prefixSum = {0};
        return IntStream.range(0, A.length)
            .map(i -> {
                prefixSum[0] += A[i];
                return i;
            })
            .map(i -> {
                if (map.containsKey(prefixSum[0])) {
                    return i - map.get(prefixSum[0]);
                } else {
                    map.put(prefixSum[0], i);
                    return 0;
                }
            })
            .max()
            .orElse(0);
    }
}