Leetcode2829. k-avoiding 数组的最小总和

最新推荐文章于 2025-04-03 15:29:45 发布

万水千山.

最新推荐文章于 2025-04-03 15:29:45 发布

阅读量298

点赞数 6

文章标签：算法数据结构

本文链接：https://blog.youkuaiyun.com/L2003__/article/details/146531282

版权

给你两个整数 n 和 k 。

对于一个由不同正整数组成的数组，如果其中不存在任何求和等于 k 的不同元素对，则称其为 k-avoiding 数组。

返回长度为 n 的 k-avoiding 数组的可能的最小总和。

示例 1：

输入：n = 5, k = 4
输出：18
解释：设若 k-avoiding 数组为 [1,2,4,5,6] ，其元素总和为 18 。
可以证明不存在总和小于 18 的 k-avoiding 数组。

示例 2：

输入：n = 2, k = 6
输出：3
解释：可以构造数组 [1,2] ，其元素总和为 3 。
可以证明不存在总和小于 3 的 k-avoiding 数组。

提示：

1 <= n, k <= 50

题目要求：数组中不能有重复元素，且任意元素相加不能等于k，且数组和最小

方法一：自己想的不太优雅的方法，用栈记录是否存在k-i

代码：

class Solution {
    public int minimumSum(int n, int k) {
        int ans = 0;
        int i = 1;//从一开始添加使得数组元素和最小
        Set<Integer> t = new HashSet<>();

        //加入数组元素个数
        while(n > 0){
            //添加进set
            t.add(i);
            //i < k使得判断更快，因为i>k的部分不要判断
            if(i < k && 2*i!=k && t.contains(k-i)){
                i++;
                continue;
            }
            ans += i;
            i++;
            n--;
        }
        return ans;
    }
}

方法2：优雅出自灵茶山艾府

代码：

class Solution {
    public int minimumSum(int n, int k) {
        int m = Math.min(k / 2, n);
        return (m * (m + 1) + (k * 2 + n - m - 1) * (n - m)) / 2;
    }
}

作者：灵茶山艾府
链接：https://leetcode.cn/problems/determine-the-minimum-sum-of-a-k-avoiding-array/solutions/2396408/o1-gong-shi-pythonjavacgo-by-endlesschen-cztk/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。