问题 - 478B - 代码强制 (codeforces.com)
有n个人,要分成m个队伍,队伍中的其中两人会成为朋友,一对朋友对两个人都是一对的作用效果,也就是这里不考虑重复的效果,一个人和其中队伍剩余的人成为一对朋友,假设一个队伍中有5 个人
朋友对的计算为1+2+3+4 == n*(n-1)/2
#include <iostream>
#include <bits/stdc++.h>
#include <cstring>
#include <vector>
#include <queue>
#include <algorithm>
#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<vector<LL>> VVL;
typedef vector<LL> VL;
typedef vector<int> VI;
typedef pair<int,int> PII;
typedef pair<double,double> PDD;
typedef pair<double,int> PDI;
typedef pair<char,int> PCI;
typedef pair<string,int> PSI;
typedef pair<int,string> PIS;
typedef pair<LL,LL> 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 <<endl;
}
signed main()
{
// freopen("1.txt","w",stdout);
ios
int t=1;
// cin>>t;
int now = 1;
while(t -- )
{
// cout<<"Case ";
// cout<<"Scenario #";
// cout<< now ++ <<": ";
// cout<< now ++ <<": \n";
solve();
}
return 0;
}