leetcode 2829. k-avoiding 数组的最小总和 中等
给你两个整数 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
分析:构造总和最小的数组,必然是从 1 开始的连续自然数序列。题目要求数组中不存在总和为 k的元素对,即不存在任意两个数之和等于 k 。这样保留1,就不保留k-1;保留2,就不保留k-2等等。这样一直到(k+1)/2向下取整,就保留了两个数之和等于k的前一半较小的数。之后从k开始,补足不够的数字。
int minimumSum(int n, int k) {
int ans=0,cnt=n,t=1;
int flag[200]={0};
for(int i=1;i<(k+1)/2;++i)
flag[k-i]=1;
while(cnt>0)
{
if(!flag[t])ans+=t,t++,cnt--;
else while(flag[t])t++;
}
return ans;
}