PROBLEM_B： URAL 2025 B - Line Fighting

最新推荐文章于 2017-08-08 19:50:44 发布

原创最新推荐文章于 2017-08-08 19:50:44 发布 · 428 阅读

0 ·

CC 4.0 BY-SA版权

DSTA 专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一个具体的组合数学问题，并提供了一种通过编程实现的解决方案。该方案利用C语言进行编码，采用了一系列自定义宏来简化操作流程，通过计算组合数解决特定场景下的配对问题。

/*PROBLEM B: LINE FIGHTING "SOLVED" */
#include "cstdio"
#include "cstring"
#include "cctype"
#include "algorithm"
using namespace std;
#define Inc(i, a, b) for(int i = a; i < b; i++)
#define Dec(i, a, b) for(int i = a; i > b; i--)
#define Mem(a) memset(a, 0, sizeof(a))
#define Pii pair<int, int> 
#define Cin scanf
#define Put printf
#define CIN(a) scanf("%d", &a)
#define CII(a, b) scanf("%d%d", &a, &b)
#define CIS(a) scanf("%s", a)
#define PII(a, b) printf("%d %d\n", a, b)
#define ll long long
#define maxn 100005
int n, k;
ll C(int a, int b) {
	if(a < b) return 0;
	ll ret = 1;
	b = min(b, a - b);
	Inc(i, 0, b) ret *= (ll)(a - i), ret /= (ll)(i + 1);
	return ret;
} 
void solve() {
	int en = n, ek = k;
	ll mt = n / k;
	ll et = n % k;
	if(!et) Put("%lld\n", C(k, 2) * mt * mt);
	else Put("%lld\n", C(k - et, 2) * mt * mt + C(et, 2) * (mt + 1) * (mt + 1) + et * (k - et) * mt * (mt + 1));
}
int main() {
	int cas;
	CIN(cas);
	while(cas--) {
		CII(n, k);
		solve();
	}
}