1327A Sum of Odd Integers（思維）

1327A Sum of Odd Integers（思維）

題面

題意：有 $T$ 個詢問，每個詢問給出兩個整數 $n$ 和 $k$，問 $n$ 是否能夠用 $k​$ 個不同正奇數之和。

範圍：$1 \le T \le 10^5~,~1 \le n, k \le 10^7​$

分析：已知 $k$ 個不同正奇數可以表示的最小數字是 $1+3+5+...+2k-1 = k^2$， 之後可以改變其中任意一個數字就可以得到 $k^2+2,k^2+4,...k^2+2m$，因此我們只需要判斷 $n \ge k^2$ 且 $n$ 和 $k​$ 的奇偶性相同即可。

Code：

#include <bits/stdc++.h>
#define int long long
#define double long double
using namespace std;

// const int MAXN =
const int INF = 0x3f3f3f3f;
const int MOD = 1e9 + 7;
const double eps = 1e-9;
const double PI = acos(-1.0);

int n;

inline int read()
{
    int s = 0, w = 1;
    char ch = getchar();
    while (ch < '0' || ch > '9')
    {
        if (ch == '-')
            w = -1;
        ch = getchar();
    }
    while (ch >= '0' && ch <= '9')
        s = s * 10 + ch - '0', ch = getchar();
    return s * w;
}

signed main()
{
    int T = read();
    while (T--)
    {
        int a, b;
        cin >> a >> b;
        if (a >= b * b && a % 2 == b % 2)
        {
            cout << "YES" << endl;
        }
        else
        {
            cout << "NO" << endl;
        }
    }
    return 0;
}

【END】感謝觀看