| ||||||||||
| Online Judge | Problem Set | Authors | Online Contests | User | ||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Web Board Home Page F.A.Qs Statistical Charts | Current Contest Past Contests Scheduled Contests Award Contest | |||||||||
Re:发一个最强的注释版的代码,保管一看就懂。In Reply To:发一个最强的注释版的代码,保管一看就懂。 Posted by:zoujing1978 at 2017-07-26 00:27:50 > /*
> 题意:给定N个数,两两做差取绝对值,再找出这些绝对值的中位数。也就是给n个数放在数组a中,求出C(n,2)个|a[i] - a[j]|数中的的中位数。
> 思路:对数组a排序后,估计一个两两做差的绝对值的集合的中位数mid,与a[i]的差大于mid(也就是某个数大于a[i] + mid)的那些数的个数如果小于C(n,2) / 2的话,说明mid太大了。以此为条件进行第一重二分搜索,第二重二分搜索是对X的搜索,直接用lower_bound实现。
> */
>
> #include <iostream>
> #include <fstream>
> #include <algorithm>
> #include <functional>
> #include <cstdio>
> using namespace std;
>
> // 计算组合数
> long long C(int m, int n)
> {
> int p = 1;
> int q = 1;
> for (int i = 1; i <= n; i++)
> {
> p *= (m - i + 1);
> q *= i;
> }
> return p / q;
> }
>
> // 判断mid作为中位数是否太大。执行此操作前数组a必须排序。
> bool Check(int* a, int n, int mid)
> {
> int sum = 0; // 与a[i]的差大于mid(也就是某个数大于a[i] + mid)的那些数的个数之和
> // 遍历每一个a[i],求大于或者等于mid+a[i]的个数
> for (int i = 0; i < n; i++)
> {
> // lower_bound(a + i + 1, a + n, mid + a[i])表示在a[i+1]到a[n-1]中二分查找大于或者等于mid+a[i]的第一个元素的位置
> sum += (a + n) - lower_bound(a + i + 1, a + n, mid + a[i]);
> }
> // 如果满足sum <= C(n, 2) / 2,说明大于或等于中位数的元素没有超过半数或者正好等于半数,分为这么4种可能:
> //1、大于或等于中位数mid(我们估计的)的这部分元素包含了这个中位数mid(我们估计的),个数正好等于半数,说明元素个数有偶数个,而后半截的第一个元素是这个中位数mid(我们估计的),显然应该是mid的前一个元素才是真正的中位数,说明mid估大了;
> //2、大于或等于中位数mid(我们估计的)的这部分元素不包含这个中位数mid(我们估计的),个数正好等于半数,说明元素个数有偶数个,而后半截的第一个元素大于mid,前半截最后一个元素小于mid,显然应该是mid的前一个元素才是真正的中位数,说明mid估大了;
> //3、大于或等于中位数mid(我们估计的)的这部分元素包含了这个中位数mid(我们估计的),个数小于半数,说明后半截的第一个元素是这个中位数mid(我们估计的),又因为后半截的元素个数不足半数,显然说明mid估大了;
> //4、大于或等于中位数mid(我们估计的)的这部分元素不包含了这个中位数mid(我们估计的),个数小于半数,说明后半截的第一个元素大于mid(我们估计的),前半截的最后一个元素小于mid,又因为后半截的元素个数不足半数,显然说明mid估大了;
> // 综合1、2、3、4,如果满足sum <= C(n, 2) / 2这种情况反正就是估大了
> return sum <= C(n, 2) / 2; // 返回true就是估大了,返回false就是估小了或者正好
> }
>
> int main()
> {
> int n = 0;
> //while (cin >> n)
> while(scanf("%d", &n)==1)
> {
> int* a = new int[n];
> for (int i = 0; i < n; i++)
> {
> //cin >> a[i];
> scanf("%d", &a[i]);
> }
> sort(a, a + n);
> long long lb = 0; // 两两做差取绝对值可能的最小值
> long long ub = a[n - 1] - a[0]; // 两两做差取绝对值可能的最大值
> // 为什么循环条件是ub - lb > 1?是因为ub、lb、mid都是整数,由于mid会向下取整,所以当ub-lb=1时,mid为lb,而此时真正的中位数要么为ub,要么为lb,也就是中位数mid要么估的正好,要么估低了,在循环体内,只会走else分支,下届lb被修正为mid,是同样的值,并没有改变,于是陷入死循环,因此循环条件不能写成ub - lb >= 1或者ub - lb >=0,只能写成ub - lb > 1。
> // 当循环条件为ub - lb > 1时,在循环快结束时,必定有ub-lb=2,最终mid的值lb+1,而此时真正的中位数要么为ub,要么为lb,要么为lb+1。
> // 如果真正的中位数为ub,说明mid估低了,在循环体内,走else分支,修正lb的值,在原先的值的基础上加1,得到了真实的中位数的值,随后循环结束;
> // 如果真正的中位数为lb,说明mid估高了,在循环体内,走if分支,修正ub的值,然后循环结束;
> // 如果真正的中位数为lb+1,说明mid估的正好,在循环体内,走else分支,修正lb的值,在原先的值的基础上加1,得到了真实的中位数的值,随后循环结束;
> // 综上,循环条件必须写成ub - lb > 1,循环结束后,lb就是确切的中位数的值
> while (ub - lb > 1)
> {
> int mid = (lb + ub) >> 1; // 尝试新的中位数,
> bool flag = Check(a, n, mid);
> // 如果中位数猜测的太大
> if (flag)
> {
> ub = mid; // 修改上界。
> }
> // 如果中位数猜测的太小,或者小于中位数的元素个数正好等于半数
> else
> {
> lb = mid; // 修改下界。
> }
> }
> //cout << lb << "\n";
> printf("%d\n", lb);
> delete[] a;
> }
> return 0;
> }
Followed by:
Post your reply here: |
All Rights Reserved 2003-2013 Ying Fuchen,Xu Pengcheng,Xie Di
Any problem, Please Contact Administrator