【JZOJ 杂题选讲】HDU5205

题目

思路

代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll n,m;
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%lld%Illd",&n,&m);
		ll ans = 0;
		if(m>=(n*(n-1))/2) ans = n*(n-1);
		else if (m >= n-1)
		{
			ans = 2*(n*n - n - m);
		}
		else
		{
			ans= 2*m*m+(n-m-1)*(m+1)*n*2+(n-m-1)*(n-m-2)*n;
		}
		printf("%lld\n",ans);
	}
}