4877196_wa.cpp

来自「部分PKU上的源码」· C++ 代码 · 共 51 行

#include<iostream>
using namespace std;
int main()
{
	int i,n,m;
	int testnumber;
	bool find;
	cin>>testnumber;
	for(int count=0;count<testnumber;count++)
	{
		cin>>n>>m;
		int all=0,fn;
		int gi,gl;
		find=false;
		for(i=1;i<=n;i++)
		{
			all+=i;
			if(all>=m)
			{
				all-=i;
				fn=m-all;
				find=true;
				gi=1;gl=n+1-i;
				break;
			}
		}
		if(!find)
		{
		for(i=n-1;i>=1;i--)
		{
			all+=i;
			if(all>=m)
			{
				all-=i;
				fn=m-all;
				gi=n+1-i;
				gl=1;
				break;
			}
		}
		}
		for(i=1;i<fn;i++)
		{
			gi++;gl++;
		}
		__int64 re;
		re= gi*gi + 100000*gi + gl*gl - 100000 * gl + gi * gl;
		printf("%I64d\n",re);
	}
	return 0;
}