codeforces 626E. Simple Skewness 三分

题目链接

给n个数， 让你去掉一些数， 使得剩下的数的平均值-中位数的差值最大。

先将数组排序， 然后枚举每一个数作为中位数的情况， 对于每个枚举的数， 三分它的左右区间长度找到一个平均值最大的情况， 平均值最大， 肯定是它左边的数是靠近他的那几个数， 右边的数是最右边的那几个数。 然后所有情况取最大值。

三分的写法lmid = (l*2+r)/2, rmid = (l+r*2+2)/3， 学到了。

并且求平均值最好不要除，比如说平均数-中位数， 那么写成   这几个数的和-中位数*长度。

 #include <iostream>
 #include <vector>
 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 #include <cmath>
 #include <map>
 #include <set>
 #include <string>
 #include <queue>
 #include <stack>
 #include <bitset>
 using namespace std;
 #define pb(x) push_back(x)
 #define ll long long
 #define mk(x, y) make_pair(x, y)
 #define lson l, m, rt<<1
 #define mem(a) memset(a, 0, sizeof(a))
 #define rson m+1, r, rt<<1|1
 #define mem1(a) memset(a, -1, sizeof(a))
 #define mem2(a) memset(a, 0x3f, sizeof(a))
 #define rep(i, n, a) for(int i = a; i<n; i++)
 #define fi first
 #define se second
 typedef pair<int, int> pll;
 const double PI = acos(-1.0);
 const double eps = 1e-;
 const int mod = 1e9+;
 const int inf = ;
 const int dir[][] = { {-, }, {, }, {, -}, {, } };
 const int maxn = 2e5+;
 ll a[maxn], sum[maxn];
 int n;
 ll cal(int len, int pos) {
     ll num = sum[pos]-sum[pos-len-]+sum[n]-sum[n-len];
     return num;
 }
 int main()
 {
     cin>>n;
     for(int i = ; i<=n; i++) {
         scanf("%I64d", &a[i]);
     }
     if(n ==  || n == ) {
         cout<<<<endl;
         cout<<a[]<<endl;
         return ;
     }
     sort(a+, a++n);
     ll maxx = ;
     int len = , mid = ;
     for(int i = ; i<=n; i++)
         sum[i] = sum[i-]+a[i];
     for(int i = ; i<=n-; i++) {
         int l = , r = min(i-, n-i);
         while(l<r) {
             int lmid = (*l+r)/;
             int rmid = (l+*r+)/;
             ll ans1 = cal(lmid, i);
             ll ans2 = cal(rmid, i);
             if(ans1*(*rmid+)<(*lmid+)*ans2) {
                 l = lmid+;
             } else {
                 r = rmid-;
             }
         }
         ll tmp = (sum[i]-sum[i-l-]+sum[n]-sum[n-l])-(*l+)*a[i];
         if(tmp*(*len+)>maxx*(*l+)) {
             maxx = tmp;
             len = l;
             mid = i;
         }
     }
     cout<<*len+<<endl;
     for(int i = mid-len; i<=mid; i++)
         printf("%d ", a[i]);
     for(int i = n-len+; i<=n; i++)
         printf("%d ", a[i]);
     return ;
 }