简单的数学题

心怀凉月

于 2021-07-08 20:15:43 发布

阅读量205

点赞数

文章标签：数论

本文链接：https://blog.csdn.net/qq_46258139/article/details/118580836

版权

这篇博客讨论了一个算法问题，即如何在1到n的整数中选取k个数，使得选取的数之和等于给定的目标值s。文章指出，由于每个选取的数只能是1或0，因此可以将问题转化为在一定范围内选取连续整数的子集和。通过分析和证明，得出结论：当目标和位于特定边界内时，这样的数组存在；否则不存在。文章提供了相应的代码实现来验证这个结论。

摘要由CSDN通过智能技术生成

Description

给定一个正整数 $n$ ，表示 $1$ 到 $n$ 这 $n$ 个正整数，以及给出一个正整数 $s$ 和一个正整数 $k$ ，要求你造出一个数组 $a$ ， $a$ 数组应该满足 $a_i==0$ 或者 $a_i==1$ $\le i \le n）$ ，并且满足 $n=\sum\limits_{i=1}^n$ 和 $s=\sum\limits_{i=1}^na_i\times i$ ，请问是否存在这样的a数组。

Input

本题有多组数据。
第一行一个正整数 $T$ ，表示数据组数。
对于每组数据，一行 $3$ 个正整数 $n, k, s$

Output

对于每组数据：
输出一行，一个字符串， $Y e s$ 或 $N o$ ，表示是否能够创建这样的 $a$ 数组，满足题意

Sample Input

Sample Output

No
Yes
No

HINT

数据范围：

对于 $100\%$ 的数据， $\le T \le 10000， 1 \le n，k \le 10^9，1 \le s \le 10^{18}$

Solution

由于 $a_i$ 只能为 $1$ 或 $0$ ，因此存在 $k$ 组 $a_i=1$ ，同时 $a_i\times i$ 只能等于 $0$ 或 $i$ 。那么，问题就变成在 $1\sim n$ 中选 $k$ 个数使其和为 $s$ 。

这个和是有上下界的。最小是 $1+\cdots+k=\dfrac{(1+k)\times k}{2}$ ，最大是 $(n-k+1)+\cdots+n=\dfrac{(n - k + 1 + n)\times k}{2}$ 。

那我们就要证明这些和能取遍上界到下界的所有和，这是为什么呢？

证明：

$1+2+\cdots+k$ 这是和的下界，考虑对它进行 $+ 1$ 操作：先把最后的 $k\gets k+1$ ，直到 $k = n$ ，再来对第二个数 $k - 1$ 进行 $+ 1$ ，直到这个数 $= n - 1$ ，如果还不够，继续类推

那么就可以确定，上界与下界之间的和都可以构造出来。

那么程序只要判断是不是 $\dfrac{(1+k)\times k}{2}\le s\le\dfrac{(n - k + 1 + n)\times k}{2}$ 即可

Code

#include <bits/stdc++.h>
using namespace std;

long long T, n, k, s;

signed main()
{
    scanf("%lld", &T);
    while (T--)
    {
        scanf("%lld%lld%lld", &n, &k, &s);
        if (s > (n + n - k + 1) * k / 2 || s < (1 + k) * k / 2)
            printf("No\n");
        else
            printf("Yes\n");
    }
}