478B. Random Teams

链接:

https://codeforces.com/problemset/problem/478/B

题意

n个选手组成m队，每队至少1人，成为队友比赛后，互相之间会成为朋友，问你比赛之后组成最少和最多的好友对数各是多少？
1 ≤ m ≤ n ≤ 1e9

Example

input
5 1

output
10 10

input
3 2

output
1 1

input
6 3

output
3 6

解析

这题需要找到两个最值，当假设m队里前面所有的队里都是一个人，剩下p人都在最后一队里，此时的对数就是Cp2（组合运算），即为最大值；最小值就是将n个人均分到每个队伍里。
具体如何均分：

#include <iostream>

using namespace std;

typedef long long ll;

int main()
{
	ll n, m;
	cin >> n >> m;

	if (m == 1) cout << n * (n - 1) / 2 << ' ' << n * (n - 1) / 2;
	else
	{
		ll k_max, k_min;
		ll num = n / m;
		ll t = n % m;
		ll w = m - t;

		ll p = n - (m - 1);
		k_max = p * (p - 1) / 2;
		k_min = t * (num + 1) * num / 2 + w * num * (num - 1) / 2;

		cout << k_min << ' ' << k_max;
	}

	return 0;
}