CodeForces 478 B.Random Teams（组合数学）

Description

将$n$个人分成$m$组，每组至少一人，组内人都成了好朋友，问所有方案中，朋友对数最少和最多分别是多少

Input

两个正整数$n,m(1\le m\le n\le 10^9)$

Output

输出最少朋友对数和最多朋友对数

Sample Input

3 2

Sample Output

1 1

Solution

由$C_x^2+C_y^2<C_{x+y}^2$知，最多时就是$m-1$个人一人一组，剩下人一组，此时答案为$C_{n-m+1}^2$，最少时就是人数尽量均分，有$n\%m$组有$\lceil\frac{n}{m}\rceil$个人，有$m-n\%m$组有$\lfloor\frac{n}{m}\rfloor$，以此计算出最少对数

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=100001;
ll C(int n)
{
    return (ll)n*(n-1)/2;
}
int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m)) 
    {
        ll k1=(ll)(n%m)*C(n/m+1)+(ll)(m-n%m)*C(n/m);
        ll k2=C(n-m+1);
        printf("%I64d %I64d\n",k1,k2);
    }
    return 0;
}