【2019牛客暑期多校训练营第一场 C题 Euclidean Distance】【拉格朗日乘子法】

题目链接：
https://ac.nowcoder.com/acm/contest/881/C
题意：
现给你一个n维空间中的点$a=(a_1,a_2,\cdot \cdot \cdot ,a_n)$，保证$a_i$为整数，现在求另外一个n维空间中的点$p$，需要满足条件：
①$p_1,p_2,\cdot \cdot \cdot ,p_n\in R$
②$p_1,p_2,\cdot \cdot \cdot ,p_n\ge 0$
③$\sum _{i=1}^n{p}_i=1$
④使得$\sum _{i=1}^n(\frac{a_i}{m}-p_i{)}^2$最小
最后输出这个式子的最小值
题解：
题目就是在满足一定约束条件下目标函数的最小值，如果没有那个不等式条件那就简单多了，用拉格朗日乘数法就解决了，但是这里多了一个条件②那样的不等式条件，所以用到了拉格朗日乘子法。
首先需要构造一个拉格朗日函数 $L(p_1,p_2,\cdot \cdot \cdot ,p_n,\lambda )=\sum _{i=1}^n(\frac{a_i}{m}-p_i{)}^2+2\lambda (\sum _{i=1}^n{p}_i-1)$
如果没有那个不等式条件，只需要对所有项求偏导等于0，即
$L_{p_1}=2(\frac{a_1}{m}-p_1)(-1)+2\lambda =0$
$L_{p_2}=2(\frac{a_2}{m}-p_2)(-1)+2\lambda =0$
$\cdot \cdot \cdot$
$L_{p_n}=2(\frac{a_n}{m}-p_n)(-1)+2\lambda =0$
$L_{\lambda }=\sum _{i=1}^n{p}_i-1=0$
但是这里有$p_1,p_2,\cdot \cdot \cdot ,p_n\ge 0$，需要一项项带进去检验$p_i\ge 0$，如果不满足就舍弃即$p_i=0$(取0是因为$p_i\ge 0$）
先对数组$a$从小到大排序
从上面的求偏导我们可以得到$p_i=\frac{a_i}{m}-\lambda$，如果当前已检查过符合条件的$p_i$的个数为$k$
那么此时$L_{\lambda }=\sum _{i=1}^k{p}_i-1=0$ 可以得到$\lambda =\frac{\sum _{i=1}^k(a_i)-m}{mk}$,带入$p_{k+1}=\frac{a_{k+1}}{m}-\lambda$
得到$mk{p}_{k+1}=k{a}_{k+1}-\sum _{i=1}^k{a}_i+m\ge 0$，如果符合条件继续判断下一项，如果不行那就停止（因为先排过序，之后结果只会更小）

现在假设最后的项数为$k$，则$p_{k+1},p_{k+2},\cdot \cdot \cdot ,p_n$均为0
则$\sum _{i=1}^n(\frac{a_i}{m}-p_i{)}^2$
$=\sum _{i=1}^k(\frac{a_i}{m}-p_i{)}^2+\sum _{i=k+1}^n\frac{a_i^2}{m^2}$
$=k{\lambda }^2+\frac{\sum _{i=k+1}^n{a}_i^2}{m^2}$
$=\frac{(\sum _{i=1}^k(a_i)-m{)}^2+k\sum _{i=k+1}^n{a}_i^2}{m^{2}k}$
最后上代码
代码：

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
#define pi 3.1415926
#define mp(x, y) make_pair(x, y)
#define vi vector<int>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> Pair;
const int MAX = 1e4 + 10;
const ll mod = 1e9 + 7;

ll n, m;
ll a[MAX], pre[MAX];

ll gcd(ll a, ll b) { return b ? gcd(b, a % b) : a; }
bool cmp(const ll &a, const ll &b) { return a > b; }

int main() {
#ifdef ACM_LOCAL
	freopen("input.txt", "r", stdin);
	freopen("output.txt", "w", stdout);
#endif
	ios::sync_with_stdio(0);
	while (~scanf("%lld%lld", &n, &m)) {
		for (int i = 1; i <= n; i++)
			scanf("%lld", &a[i]);
		sort(a + 1, a + 1 + n, cmp);
		//求前缀和
		pre[0] = -m;
		for (int i = 1; i <= n; i++)
			pre[i] = pre[i - 1] + a[i];
		//找满足的项数
		ll k = n;
		for (int i = 1; i < n; i++)
			if (i * a[i + 1] < pre[i]) {
				k = i;
				break;
			}
		ll p = pre[k] * pre[k], q = m * m * k;
		for (int i = k + 1; i <= n; i++)
			p += k * a[i] * a[i];
		ll gcdd = gcd(p, q);
		p /= gcdd; q /= gcdd;
		if (q == 1 || !p)printf("%lld\n", p);
		else printf("%lld/%lld\n", p, q);
		
	}
	return 0;
}