中心对称数 III

/*
 * 思路：
 * 1. 使用 Java Stream API 和一个 HashMap 来实现。
 * 2. Stream 中的 forEach 代替传统的 for 循环。
 * 3. 使用一个原子引用来存储结果。
 */
 
import java.util.HashMap;
import java.util.concurrent.atomic.AtomicReference;
import java.util.stream.IntStream;
 
public class TwoSumStream {
    public int[] twoSum(int[] nums, int target) {
        HashMap<Integer, Integer> map = new HashMap<>();
        AtomicReference<int[]> result = new AtomicReference<>(new int[0]);
        IntStream.range(0, nums.length).forEach(i -> {
            int complement = target - nums[i];
            if (map.containsKey(complement)) {
                result.set(new int[] { map.get(complement), i });
            }
            map.put(nums[i], i);
        });
        return result.get();
    }
}