提高模拟赛Day8T1求中位数-CSDN博客

本文链接：https://blog.csdn.net/weixin_46304837/article/details/121202939

本文介绍了一种更高效的方法来解决竞赛问题：利用双指针技巧求解给定数组两两差值后中位数的查找。通过对比正解的双指针策略，展示了如何避免排序并降低时间复杂度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

提高模拟赛Day8T1求中位数

文章目录

题目
思路
代码

题目

$n$ 个数 $a_{1\ldots n}$ ,两两做差(大减小),得到数组 $b$ ,求 $b$ 的中位数.

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-Y8gsHi49-1636338364992)(YbtOJ提高模拟赛Day8T1求中位数.assets/image-20211108101734381.png)]

思路

对原数组排序,二分枚举中位数 $m i d$ ,求有多少个做差数组中比 $m i d$ 大/小的数分别由多少个,时间复杂度是 $O(n\log^2n)$ ,可以卡过去.

正解是双指针?

代码

//#pragma GCC optimize(2)

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>

#define int long long
using namespace std;
int read() {
    int re = 0;
    char c = getchar();
    bool negt = false;
    while(c < '0' || c > '9')negt |= (c == '-') , c = getchar();
    while(c >= '0' && c <= '9')re = (re << 1) + (re << 3) + c - '0' , c = getchar();
    return negt ? -re : re;
}

const int N = 2e5 + 10;
int abs_(int a) {
    return a < 0 ? -a : a;
}

int n , m;
int a[N];

vector <int> ans;
int check(int val) { //return 0(too small) 1(too big) 2(find answer)
    int sm = 0 , bg = 0;
    int v , l , r;
    for(int i = 1 ; i <= n ; ++i) {
        v = a[i] + val;
        l = lower_bound(a + 1 , a + n + 1 , v) - a , r = upper_bound(a + 1 , a + n + 1 , v) - a;
        if(a[l] >= v || l == n + 1)--l;
        sm += max(0ll , l - i) ;
        bg += n - r + 1;
    }
    int equ = m - sm - bg;
    if(sm >= (m + 1) / 2)return 1;
    if(sm + equ < (m + 1) / 2)return 0;
    if(equ == 0)return 0;
    ans.push_back(val);
    return 2;
}
int check2(int val) { //return 0(too small) 1(too big) 2(find answer)
    int sm = 0 , bg = 0;
    int v , l , r;
    for(int i = 1 ; i <= n ; ++i) {
        v = a[i] + val;
        l = lower_bound(a + 1 , a + n + 1 , v) - a , r = upper_bound(a + 1 , a + n + 1 , v) - a;
        if(a[l] >= v || l == n + 1)--l;
        sm += max(0ll , l - i) ;
        bg += n - r + 1;
    }
    int equ = m - sm - bg;
    if(sm >= m / 2 + 1)return 1;
    if(sm + equ < m / 2 + 1)return 0;
    if(equ == 0)return 0;
    ans.push_back(val);
    return 2;
}
signed main() {
	// freopen("mid.in" , "r" , stdin);
	// freopen("mid.out" , "w" , stdout);
	
    n = read() , m = n * (n - 1) / 2;
    for(int i = 1 ; i <= n ; i++)
        a[i] = read();
    sort(a + 1 , a + n + 1);
    
    
    int l = 0 , r = 1ll << 31;
    while(l <= r) {
        int mid = (l + r) / 2;
        int tmp = check(mid);
        if(tmp == 2)break;
        if(tmp == 0)l = mid + 1;
        else r = mid - 1;
    }
    if((m & 1) == 0) {//有两个中位数
    	l = 0 , r = 1ll << 31;
	    while(l <= r) {
	        int mid = (l + r) / 2;
	        int tmp = check2(mid);
	        if(tmp == 2)break;
	        if(tmp == 0)l = mid + 1;
	        else r = mid - 1;
	    }
	}
	int sum = 0;
	for(int i : ans)
		sum += i;
	cout << sum / ans.size();
    return 0;
}