最小差值平方和

/*
 * 思路：
 * 1. 计算初始差值平方和。
 * 2. 使用 Java Stream 将 nums1 和 nums2 对应位置元素的差值的绝对值存储在一个列表中。
 * 3. 根据 k1 和 k2 的总和调整最大的差值，减少差值平方和。
 * 4. 最后计算调整后的差值平方和。
 */

import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.IntStream;

public class Solution {
    public long minSumSquareDiff(int[] nums1, int[] nums2, int k1, int k2) {
        int n = nums1.length;
        List<Integer> diffs = IntStream.range(0, n)
                .map(i -> Math.abs(nums1[i] - nums2[i]))
                .boxed()
                .sorted((a, b) -> b - a)
                .collect(Collectors.toList());

        long initialSum = diffs.stream().mapToLong(diff -> (long) diff * diff).sum();
        int totalK = k1 + k2;
        
        for (int i = 0; i < n && totalK > 0; i++) {
            if (diffs.get(i) > 0) {
                int newDiff = Math.max(0, diffs.get(i) - 1);
                initialSum -= (long) diffs.get(i) * diffs.get(i);
                initialSum += (long) newDiff * newDiff;
                diffs.set(i, newDiff);
                totalK--;
            }
        }
        
        return initialSum;
    }
}