Rikka with Graph

原题链接：点击查看原题

题意：

给你n个点，你可以在这n个点之间任意连<=m条线，使∑ni=1∑nj=1dist(i,j).的值最小。

其中dist(i,j)表示如过i和j之间有通路，则dist(i,j)的值为i和j之间经过的边数。如a--->b---c

（前提是没有a--->c，如过有的话，就取 1 而不是 2 ，因为要使值最小)那么dist(a,c)等于2
因为经过了两条边，如过a和b之间没有通路，那么dist(a,b)就等于n。

解题思路：

假设这些点相互都相连，即n*(n-1)/2=m，则答案为：n*(n-1)，

假设这些点是联通的，即：m>=n-1,我们可以根据上种情况接着思考，如果每有一个不相互连通，则就要加2，

即：n*（n-1）+（n*(n-1)/2-m）*2.

假设有一部分相连，即：m<n-1,

q表示没相连的点，p为相连的点。则：p=m+1，q=n-m-1。

对于没有相连的：q*(q-1)*n

对于相连的：(p-1)*(p-1)*2;//由(1+(m-1)*2)*m+m化简得到

对于没相连的点与相连的：q*p*n；

把上边的相加就是最终答案。

原代码：

#include <iostream>
using namespace std;
int main()
{
   int t;
   cin>>t;
   while(t--)
   {
       long long n,m;
       cin>>n>>m;
       if(n*(n-1)/2<=m){
            cout<<n*(n-1)<<endl;
            continue;
       }
       if(m>=n-1)
        cout<<n*(n-1)+(n*(n-1)/2-m)*2<<endl;
       else
       {
           long long p=m+1;
           long long q=n-m-1;
           long long ans=q*p*n*2+q*(q-1)*n+2*(p-1)*(p-1);
           cout<<ans<<endl;
       }
   }
   return 0;
}