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 =  * (a - );
        }
        else if (a == b - ) {
            ans =  * (a - );
        }
        else {
            LL c = sqrt(a * b);
            if (c * c == a * b) c--;
            if(c * (c + ) >= a * b) {
                ans =  * c - ;
            }
            else {
                ans =  * c - ;
            }
        }
        printf("%lld\n",ans);
    }
    return ;
}