CodeForces - 626E Simple Skewness

$\text{Update on 2022.7.10}$：

一份更清晰的口胡。

首先可以证明选择偶数个是不优的，为了简化证明过程，可以对当前选取的集合 $S$ 做一个操作：设此集合排序后最中间两个数为 $l,r,l⩽r$，将集合中所有数减去 $l$，记此时集合元素之和为 $s$，可以发现平均数减去中位数的值（下文称其为集合的权值）仍然不变，为 $s/|S|-(0+r)/2$.

另外也可以发现，$s$ 一定大于零，否则集合的权值一定 $⩽0$，那么这个集合一定不优于只选一个元素的集合（选择奇数个）。否则我们去掉 $r$，此时集合的权值变成 $(s-r)/(|S|-1)-0$。不妨假设去掉 $r$ 的集合权值更大，这等价于
$$2s>|S|\cdot r\cdot (3-|S|)$$
由于 $|S|⩾3$（$⩽2$ 的情况可以单独讨论），所以右式恒 $⩽0$，而左式恒大于零，所以原假设成立。

于是可以枚举中位数 $x$，最大化选取集合的平均数，且 $x$ 两边的数字个数必须相等。容易想到二分 $x$ 右边选取的数字个数，如何选取？实际上就是第一次选取序列最大值和小于等于 $x$ 的最大值、第二次选取序列次大值和小于等于 $x$ 的次大值，依此类推……

每次加入一个数对，可以用它们的平均值刻画对已选集合的影响，如果平均值大于已选集合的平均值，加入就能增加集合的平均值，反之只会更劣。这个位置就是需要二分到的点。

复杂度 $\mathcal{O}(n\log n)$.

Solution

今天没有奇奇怪怪的版块了。。。

首先，我们可以证明选偶数个一定是不优的。

设 $left,right$ 分别为选择数列的最中间的两个数， $sum$ 为不包含 $left,right$ 的选择数列的和，$len$ 为选择数列的长度。

我们将数列向左平移，使 $left$ 到达 $0$。

那么此时答案就是，

$$ans1=\frac{sum}{len}-\frac{right}{2}$$

我们去掉 $right$，

$$ans2=\frac{sum-right}{len-1}-0$$

由于 $sum>0$ （小于 $0$ 平均数就小于 $0$，比中位数 $0$ 还要小，肯定是不优的），相减很容易发现 $ans1>ans2$。

好的我们接下来就只用考虑奇数个的情况。

可以枚举中位数，我们就需要最大化平均数，而且保证中位数左边选择的数和右边选择的数个数相等。

那么左右边的数从大到小选就成。我们三分中位数向外扩展的长度。

为什么要三分？

在这之前我们知道加入 $x,y$ 进入数列，相当于加两个 $x,y$ 的平均数。（对数列的平均数而言）

先开始数列只有 $1$，加入的数的平均数可能大于 $a[i]$，数列的平均数增大。到后来，加入的数的平均数变小，当减小至比数列的平均数更小，数列的平均数就会增大。

仔细思考，其实这道题是有平台的（不知道平台可以在评论区问本蒟蒻）。

但是如果出现平台，函数值已经开始下降了。因为在平台之后加入的数的平均数一定小于平台时加入的数的平均数。

卡精度，可以化除为乘或者像我一样用分数表示。（可以不用约分，约分的代码我跑了 $55s$ 多 $\mathtt{Q}\mathtt{w}\mathtt{Q}$）

Code

#include <cstdio>
#include <iostream>
#include <algorithm>
#define make make_pair
using namespace std;
typedef long long ll;
typedef pair <ll, ll> Pair;

const int N = 2e5 + 5;

int n, a[N], o, R;
ll pre[N];

int read() {
	int x = 0, f = 1; char s;
	while((s = getchar()) > '9' || s < '0') if(s == '-') f = -1;
	while(s >= '0' && s <= '9') x = (x << 1) + (x << 3) + (s ^ 48), s = getchar();
	return x * f;
}

Pair f(const int x, const int i) {return make(pre[i] - pre[i - x - 1] + pre[n] - pre[n - x], x * 2 + 1);}

bool cmp(const Pair x, const Pair y) {return 1ll * x.first * y.second - 1ll * x.second * y.first > 0;}

/*ll gcd(ll x, ll y) {
	if(! y) return x;
	return gcd(y, x % y);
}

Pair Do(const Pair x) {
	ll g = gcd(x.first, x.second);
	return make(x.first / g, x.second / g);
}*/

Pair dec(const Pair x, const Pair y) {
	Pair t = make(1ll * x.first * y.second - 1ll * x.second * y.first, 1ll * x.second * y.second);
	return t;
}

int main() {
    int l, r, left, right, len; Pair ans1, ans2, ans, res = make(-1, 1);
    n = read();
    for(int i = 1; i <= n; ++ i) a[i] = read();
    sort(a + 1, a + n + 1);
    for(int i = 1; i <= n; ++ i) pre[i] = pre[i - 1] + a[i];
    for(int i = 1; i <= n; ++ i) {
        l = 0, r = min(i - 1, n - i) + 1;
        ans = make(-1, 1);
        while(l < r) {
            left = l + r >> 1, right = left + r >> 1;
            ans1 = f(left, i), ans2 = f(right, i); //ans1 = Do(ans1); ans2 = Do(ans2);
            if(cmp(ans2, ans1)) {
            	l = left;
            	if(cmp(ans2, ans)) ans = ans2, len = right;
			}
            else {
            	r = right;
            	if(cmp(ans1, ans)) ans = ans1, len = left;
			}
        }
        ans = dec(ans, make(a[i], 1));
        if(cmp(ans, res)) o = i, R = len, res = ans;
    }
    printf("%d\n", 2 * R + 1);
    for(int i = o - R; i <= o; ++ i) printf("%d ", a[i]);
	for(int i = n; i > n - R; -- i) printf("%d ", a[i]); puts("");
    return 0;
}