4878479_wa.cpp

部分PKU上的源码

#include<iostream>
#include<math.h>
using namespace std;
	__int64 n,m,ch;
__int64 get(__int64 a)
{
	int j;
	__int64 all=0;
	for(j=1;j<=n;j++)
	{
		__int64 t=(__int64)(sqrt((double)((ch+j)*(ch+j)-4*(j*j-ch*j-a)))-(ch+j))/2;
		if(t>=n) all+=n;
		else if(t>=0) all+=t;
	}
	return all;
}
int main()
{
	int test;
	ch=100000;
	cin>>test;
	for(int count=0;count<test;count++)
	{
		scanf("%I64d%I64d",&n,&m);
		__int64 min,max;
		max=ch*ch;
		min=0-max;
		while(min<max-1)
		{
			__int64 mid=(min+max)/2;
			__int64 t=get(mid);
			if(t>=m) max=mid;
			else min=mid;
		}
		printf("%I64d",max);
	}
	return 0;
}