B. Random Teams(cf构造题)

问题 - 478B - 代码强制 (codeforces.com)

有n个人，要分成m个队伍，队伍中的其中两人会成为朋友，一对朋友对两个人都是一对的作用效果，也就是这里不考虑重复的效果，一个人和其中队伍剩余的人成为一对朋友，假设一个队伍中有5 个人
朋友对的计算为1+2+3+4 == n*（n-1）/2

#include 
#include 
#include 
#include 
#include 
#include 
#define x first
#define y second
#define pb emplace_back
#define fu(i,a,b) for(int i=a;i<=b;i ++ )
#define fd(i,a,b) for(int i=a;i>=b;i -- )
#define endl '\n'
#define ms(x,y) memset(x,y,sizeof x)
#define ios ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
using namespace std;

typedef long long LL;
typedef vector> VVL;
typedef vector VL;
typedef vector VI;
typedef pair PII;
typedef pair PDD;
typedef pair PDI;
typedef pair PCI;
typedef pair PSI;
typedef pair PIS;
typedef pair PLL;
typedef __int128 i128;
typedef unsigned long long ULL;
const int N = 1e6+ 10,M = N * 4,INF = 0x3f3f3f3f ;
const int mod = 1e9; 
const double eps = 1e-8;


LL cal(LL x)
{
    return x * (x-1) /2 ;
}


inline void solve()
{
    LL n ,m ; cin >> n >> m; 
    
    if(n < m)
    {
        cout << 0 << " " << 0 << endl;
        return;
    }
        
    LL k = n% m; // 有k组的人数比其他组多1 
    LL ans1=0,ans2=0;
    ans1 += k * cal(n /m + 1  )  + (m- k) * cal(n / m) ;
    ans2 += cal(n-m+1);// 一组的人数是n-m+1, 其他组都是1个人 
        
    cout << ans1 <<" " << ans2 <>t;
    int now = 1;
    while(t -- )
    {
//      cout<<"Case "; 
//      cout<<"Scenario #"; 
//      cout<< now ++ <<": ";
//      cout<< now ++ <<": \n";
        solve();
    }


    return 0;
}