打表找规律,随便找找就有了,实在不行看题解,写的很清楚
sqrt出现了误差,判断一下就就好了。
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=1e5+5;
typedef long long ll;
ll n,k;
inline ll Sqrt(ll n)
{
ll x=sqrt(n);
if (x*x==n)return x;
if (x*x>n)
{
while (x*x>n)x--;
return x;
}
else
{
while ((x+1)*(x+1)<n)x++;
return x;
}
}
int main()
{
int cas;
scanf("%d",&cas);
while (cas--)
{
scanf("%lld%lld",&n,&k);
if (!n)printf("0\n");
else
{
ll x=Sqrt(n);
if (k==1)
{
if (n==1)
{
printf("1\n");
continue;
}
if (x*x==n)printf("%lld\n",x*2-1);
else printf("%lld\n",x*2);
}
else
{
if (n==1||n==3)printf("-1\n");
else if (n==2)printf("1\n");
else if (n==5)printf("3\n");
else
{
ll tmp=Sqrt(n-1);
if (x*x==n||tmp*tmp==n-1)printf("%lld\n",x*2-2);
else printf("%lld\n",x*2-1);
}
}
}
}
}