2019牛客暑期多校训练营(第一场)C
题解: 拉格朗日乘子法,首先引入拉格朗日乘子得出公式
f
(
x
)
=
∑
i
=
1
n
(
p
i
−
a
i
)
2
+
2
∗
λ
(
∑
i
=
0
n
p
i
−
1
)
f(x)=\sum_{i=1}^{n}(p_i-a_i)^2+2*\lambda(\sum_{i=0}^{n}p_i-1)
f(x)=i=1∑n(pi−ai)2+2∗λ(i=0∑npi−1)
这个应该看的懂,然后引入对偶变成成
max
λ
f
(
λ
)
\max_\lambda f(\lambda)
λmaxf(λ)其中
f
(
λ
)
=
min
p
i
≥
0
∑
i
=
1
n
(
p
i
−
a
i
)
2
+
2
∗
λ
(
∑
i
=
0
n
p
i
−
1
)
f(\lambda)=\min_{p_i\geq0}\sum_{i=1}^{n}(p_i-a_i)^2+2*\lambda(\sum_{i=0}^{n}p_i-1)
f(λ)=pi≥0mini=1∑n(pi−ai)2+2∗λ(i=0∑npi−1)
然后化成叉姐给的题解里面的公式就行了
f
(
λ
)
=
min
p
i
≥
0
∑
i
=
1
n
(
p
i
−
(
a
i
−
λ
)
)
2
+
λ
(
∑
i
=
0
n
(
a
i
2
−
(
a
i
−
λ
)
2
)
−
2
λ
f(\lambda)=\min_{p_i\geq0}\sum_{i=1}^{n}(p_i-(a_i-\lambda))^2+\lambda(\sum_{i=0}^{n}(a_i^2-(a_i-\lambda)^2)-2\lambda
f(λ)=pi≥0mini=1∑n(pi−(ai−λ))2+λ(i=0∑n(ai2−(ai−λ)2)−2λ
再然后,我就不会了qaq
后来看了一下别的大佬的博客,突然感觉可以直接理解一下,
假设所有的
p
i
=
0
p_i=0
pi=0,
f
(
x
)
f(x)
f(x)就等于
a
i
2
a_i^2
ai2的和
题目要求的是 在
q
i
≥
0
,
∑
i
=
0
n
q
i
=
1
q_i\geq0,\sum_{i=0}^{n}q_i=1
qi≥0,∑i=0nqi=1条件下求
f
(
x
)
=
∑
i
=
1
n
(
p
i
−
a
i
)
2
f(x)=\sum_{i=1}^{n}(p_i-a_i)^2
f(x)=∑i=1n(pi−ai)2。不就相当于分配
p
i
p_i
pi的值,去让
(
p
i
−
a
i
)
2
(p_i-a_i)^2
(pi−ai)2更小么。根据二次函数的性质,都是自变量
x
x
x越大因变量
y
y
y化越快。所以先分配给最大肯定更优啊,直接贪心下去啊。最后肯定是变成
#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<queue>
#include<map>
#include<set>
#include<stack>
#include<bitset>
using namespace std;
typedef long long LL;
typedef unsigned long long uLL;
typedef pair<int,int> P;
#define VNAME(value) (#value)
#define bug printf("*********\n");
#define debug(x) cout<<"["<<VNAME(x)<<" = "<<x<<"]"<<endl;
#define mid (l+r)/2
#define chl 2*k+1
#define chr 2*k+2
#define lson l,mid,chl
#define rson mid+1,r,chr
#define pb push_back
#define mem(a,b) memset(a,b,sizeof(a));
const long long mod=1e9+7;
const int maxn=1e6+5;
const int INF=0x7fffffff;
const LL inf=0x3f3f3f3f;
const double eps=1e-8;
void f() {
#ifndef ONLINE_JUDGE
freopen("dat.in", "r", stdin);
#endif // ONLIN_JUDGE
}
LL a[maxn];
int n;
LL m;
int main() {
while(~scanf("%d%lld",&n,&m)) {
for(int i=1; i<=n; i++) {
scanf("%lld",&a[i]);
}
sort(a+1,a+n+1,greater<int>());
LL r=m; //p[i]的总分配价值是 1 也就是 m/m
LL pos=1; // pos 标记能够分配p[i] 到第 pos 个
while(pos<n) {
if(r<(a[pos]-a[pos+1])*pos)break;
r-=(a[pos]-a[pos+1])*pos;
pos++;
}
// 最终前pos个的值 都会是a[pos]-r/pos,将他扩大pos 倍, 然后再乘以 pos 个
LL ans=(a[pos]*pos-r)*(a[pos]*pos-r)*pos; //因为最后的值可能是 1/pos,所以把分子分母同时乘以pos个
LL b=m*m*pos*pos; // 因为求的是距离的平方 分母 就是 m*m*pos*pos
for(int i=pos+1; i<=n; i++) { //分配不到的 a[i],直接加上
ans+=a[i]*a[i]*pos*pos;
}
LL g=__gcd(ans,b);
if(ans%b==0)printf("%lld\n",ans/b);
else printf("%lld/%lld\n",ans/g,b/g);
}
return 0;
}