leetcode 6450. k-avoiding 数组的最小总和

我很忙20102023-08-24 18:07

给你两个整数 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

C++

cpp 复制代码

class Solution {
public:
    int minimumSum(int n, int k) {
        unordered_map<int,int> neg;
        int res=0;
        int i=1;
        int count=0;
        while(i<=n) {
            if(neg.find(i)==neg.end()) {
                res+=i;
                count++;
                if(k-i!=i) {
                    neg[k-i]=1;
                }
            }
            i++;
        }
        while(count<n) {
            if(neg.find(i)==neg.end()) {
                count++;
                res+=i;
            }
            i++;
        }
        return res;
    }
};