欢迎来到尧图网

客户服务 关于我们

您的位置:首页 > 科技 > IT业 > AtCoder - arc101_b Median of Medians 分析与解答

AtCoder - arc101_b Median of Medians 分析与解答

2025/2/23 7:13:05 来源:https://blog.csdn.net/codemonkey_66798/article/details/145670626  浏览:    关键词:AtCoder - arc101_b Median of Medians 分析与解答

Median of Medians - AtCoder arc101_b - Virtual Judge

寻找中位数们的中位数,使用二分搜索,在数列a的最小值和最大值之间搜索,

在一列数m1,m2,m3......m_n中,如果x是他们的中位数,应该满足大于等于x的数的数目大于等于

n-[n/2],并且x是满足该条件的最大数

比如1,2,3,4中,x=3是满足两个数(4-[4/2])大于等于x的最大x值

比如 1,2,3,4,5中,x=3是满足三个数(5-[5/2])大于等于x的最大x值

所以可以统计,在所有的中位数中,满足大于等于x的数的个数

如果统计给定一个区间[l,r],快速判断这个区间的中位数是否大于等于x呢:

通过观察,将这个区间的数不降序排列后,如果区间内大于等于x的数的数目(区间中数的个数为n)大于等于(n-[n/2]),则该区间的中位数大于等于x

比如1 2 3 4,x=3,大于等于3的个数是2=4-[4/2]

比如1 2 3 4 5,x=3,大于等于3的个数是3=5-[5/2]

将一个区间中的数分成两类,大于等于x的数令其为1,否则令其为-1,则当这个区间的数的和大于等于0的时候,这个区间的中位数大于等于x

例如1 2 3 4,变为 -1 -1 1 1,和为0

这样处理后,求一个区间的和可以通过前缀和快速计算,给定区间端点l和r,如果pref[r]-pref[l-1]>=0,则这个区间的中位数大于等于x

通过移项得到 pref[r]>=pref[l-1],(r>=l>l-1),也就是在pref数组中出现几个逆序对就有几个区间的中位数不大于等于x

因为l=1时,l-1=0,所以pref数组对于逆序对的计数要从下标0开始,统计逆序对的方法有归并排序和树状数组

复杂度 log(10^9)(二分查找)*Nlog(N)(使用树状数组)

#include<iostream>
#include<cstdio>
#include<climits>
#include<cstring>
using namespace std;#define int long long
#define lowbit(x) (x&(-x))const int maxn=2*1e5+5;int tree[maxn],a[maxn],pref[maxn];
int n;void update(int x,int d){while(x<=maxn){tree[x]+=d;x+=lowbit(x);}
}int query(int x){int res=0;while(x){res+=tree[x];x-=lowbit(x);}return res;
}int satisfy(int x){memset(tree,0,sizeof(tree));int t=n*(n+1)/2;        //总数目int tar=t-t/2;          //顺序对数目应该大于等于这个数int cnt=0;int minpref=0;pref[0]=0;for(int i=1;i<=n;i++){pref[i]=pref[i-1];if(a[i]>=x){pref[i]++;}else {pref[i]--;}minpref=min(minpref,pref[i]);}int base=-minpref+1;for(int i=0;i<=n;i++){pref[i]+=base+1;}for(int i=n;i>=0;i--){cnt+=query(pref[i]-1);update(pref[i],1);}if(t-cnt>=tar){return 1;}else {return 0;}
}int bin_search(int l,int r){int ans=-1;while(l<r){int mid=(l+r)>>1;//printf("l=%lld r=%lld mid=%lld\n",l,r,mid);if(satisfy(mid)){//printf("%lld满足!\n",mid);ans=mid;l=mid+1;}else {r=mid;}}return ans;
}signed main()
{ios::sync_with_stdio(0);cin.tie(0);cin>>n;int maxa=-1,mina=INT_MAX;for(int i=1;i<=n;i++){cin>>a[i];maxa=max(maxa,a[i]);mina=min(mina,a[i]);}cout<<bin_search(mina,maxa+1)<<"\n";return 0;
}

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com

热搜词