AtCoder Regular Contest 094 D Worst Case

Worst Case

思路：

使 a <= b

当 a == b 时 或者 a == b - 1 时，答案显然为 2 * (a - 1)

否则找到最大的 c ，使得 c * c < a * b

如果 c * (c + 1) >= a * b ,那么可以构造 c - 1 + c 个 数对 ，减去 一个包含 a 的 ，则答案为 2 * c - 2

 如果 c * (c + 1) < a * b ,那么可以构造 c + c 个 数对 ，减去 一个包含 a 的 ，则答案为 2 * c - 1

代码：

#include<bits/stdc++.h>
using namespace std;
#define LL long long 
#define pb push_back
#define mem(a, b) memset(a, b, sizeof(a))
 
int main() {
    int q;
    LL a, b, ans;
    scanf("%d", &q);
    while (q--) {
        scanf("%lld%lld", &a, &b);
        if (a > b) swap(a, b);
        if (a == b) {
            ans = 2 * (a - 1);
        }
        else if (a == b - 1) {
            ans = 2 * (a - 1);
        }
        else {
            LL c = sqrt(a * b);
            if (c * c == a * b) c--;
            if(c * (c + 1) >= a * b) {
                ans = 2 * c - 2;
            }
            else {
                ans = 2 * c - 1;
            }
        }
        printf("%lld\n",ans);
    }
    return 0;
}