思路:
1.因为需要cost
最小,所以所有的暖气需要是sum/c
(设为a
)个部分或者sum/c+1
(设为b
)个部分;
2.我们设前者x
个,后者y
个,可以得到
{
a
x
+
b
y
=
s
u
m
x
+
y
=
c
\begin{cases} ax+by=sum\\ x+y=c \end{cases}
{ax+by=sumx+y=c
解出x、y
然后算总开销即可;
代码:
#define IOS ios::sync_with_stdio(false);cin.tie(0)
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
int main(){
IOS;
int n;
cin>>n;
while(n--){
LL c,sum;
cin>>c>>sum;
LL a=sum/c,b=a+1;
LL x=b*c-sum;
LL y=c-x;
cout<<x*a*a+y*b*b<<'\n';
}
return 0;
}