使前缀和数组非负

/*
 * 思路：
 * 使用Java Stream API来实现上述逻辑。我们可以使用一个AtomicInteger来追踪前缀和，
 * 并在遍历过程中根据前缀和的值调整每个元素。
 */
import java.util.concurrent.atomic.AtomicInteger;
import java.util.stream.IntStream;

public class Solution {
    public int[] makePrefSumNonNegative(int[] nums) {
        AtomicInteger prefixSum = new AtomicInteger(0);
        return IntStream.of(nums)
            .map(num -> {
                int currentSum = prefixSum.addAndGet(num);
                while (currentSum < 0) {
                    num++;
                    currentSum = prefixSum.incrementAndGet();
                }
                return num;
            })
            .toArray();
    }
}