标签:algo 估计 sort can else 输出 一半 中位数 平均值
2017-08-02 23:12:52
writer:pprp
题目大意:给你n个数,从n个数中选取几个数,使平均数和中位数的差值最大,将选取的个数还有选取的数字找出;
算法分析:先枚举,再三分
枚举中位数,可以证明中位数一定是一个,而不是两个组成的。
三分主要用于类似于二次函数的曲线中,有极大或者极小值
代码及分析如下:
#include <iostream> #include <algorithm> #include <stdio.h> #include <cstdlib> using namespace std; typedef long long ll; const ll maxn = 200050; const ll INF = 0x3f3f3f3f; ll a[maxn]; ll sum[maxn];//sum数组是记录了从1到i的总和 int main() { ll n; cin >> n; for(ll i = 1; i <= n ; i++) { scanf("%lld",&a[i]); } sort(a+1,a+1+n); sum[0] = 0; //完成sum的赋值 for(ll i = 1; i <= n; i++) { sum[i] = sum[i - 1] + a[i]; } //对n = 1和 n = 2的情况进行特判 if(n == 1 || n == 2) { printf("1\n"); printf("%lld\n",a[1]); return 0; } ll l, r; //represent left and right ll m1, m2; //三分之一点处的长度,和三分之二处的长度 ll s1, s2; //分别代表m1/m2时的总和(不包括枚举的点) ll mark_sum = 0, mark_len = 0, mark_mid = 1;//作为标记 ll ssum = 0, len = 0, mid = 1 ; //初始化 //头和尾不可能所以从 2到n-1 for(ll i = 2 ; i <= n - 1 ; i++) { l = 1; r = min(n-i,i-1);//看看左右两边哪个更短,r是长度不是角标 for(ll j = 1; j <= 100; j++)//估计了一个数:100 { m1 = (2 * l + r) / 3; //三分之一点到右端的距离 m2 = (l + 2 * r + 2) / 3;//为了向上取整才加上2,三分之二点到右端的距离 s1 = sum[i] - sum[i - m1 - 1] + sum[n] - sum[n - m1]; s2 = sum[i] - sum[i - m2 - 1] + sum[n] - sum[n - m2]; if(s1 * (2 * m2 + 1) < s2 * (2 * m1 + 1)) //2 * m + 1是总个数,这个比较的是平均值得大小 { l = m1 + 1; //三分转移,从右往左看,从r处往左侧理解 } else { r = m2 - 1; //三分转移 } } ssum = sum[i] - sum[i - l - 1] + sum[n] - sum[n - l] - (2 * l + 1) * a[i]; len = l;//记录长度为len mid = i;//记录下来枚举的点的坐标 if(ssum*(2 * mark_len + 1) > mark_sum * (2 * len + 1))//取最大值 { mark_sum = ssum; mark_len = len; mark_mid = mid; } } //输出个数 cout << mark_len * 2 + 1 << endl; //为了格式 ll flag = 1; for(ll i = mark_mid - mark_len ; i <= mark_mid ; i++) //输出前一半 { if(flag) { flag = 0; printf("%lld",a[i]); } else { printf(" %lld",a[i]); } } //输出后一半 for(ll i = n - mark_len + 1; i <= n ; i++) printf(" %lld",a[i]); printf("\n"); return 0; }
codeforces 262E Simple Skewness - 三分
标签:algo 估计 sort can else 输出 一半 中位数 平均值
原文地址:http://www.cnblogs.com/ilovelianghui/p/7277376.html