标签:tor less main code psu size 建立 priority mat
给出一个长度为N的非负整数序列Ai?,对于所有1≤k≤(N+1)/2,输出A1,A3,…,A2k−1的中位数。即前1,3,5,…个数的中位数。
第1行为一个正整数N,表示了序列长度。
第2行包含N个非负整数Ai(Ai≤109)。
共(N+1)/2行,第i行为A1,A3,…,A2k−1?的中位数。
首先记录一个变量mid,记录答案(中位数)。建立两个堆,一个大根堆一个小根堆,大根堆存≤mid的数,小根堆存 >mid的的数。
所以我们向堆中加入元素时,就通过与mid的比较,选择加入哪一个堆
但我们在输出答案前需要对mid进行调整,如果小根堆和大根堆内元素相同,就无需处理,此时mid仍然是当前的中位数。
如果两个堆中元素个数不同,那我们就需要进行调整。
1 #include <cstdio> 2 #include <iostream> 3 #include <queue> 4 #include <algorithm> 5 using namespace std; 6 int n; 7 int a[100000]; 8 int mid; 9 priority_queue <int,vector<int>,less<int> >q1; 10 priority_queue <int,vector<int>,greater<int> >q2; 11 int main() 12 { 13 scanf ("%d",&n); 14 scanf ("%d",&a[1]); 15 mid=a[1]; 16 printf ("%d\n",mid); 17 for (int i = 2;i <= n;i++) 18 { 19 scanf ("%d",&a[i]); 20 if (a[i]>mid) q2.push(a[i]); 21 else q1.push(a[i]); 22 if (i%2==1) 23 { 24 while (q1.size()!=q2.size()) 25 { 26 if(q1.size()>q2.size()){ 27 q2.push(mid); 28 mid=q1.top(); 29 q1.pop(); 30 } 31 else{ 32 q1.push(mid); 33 mid=q2.top(); 34 q2.pop(); 35 } 36 } 37 cout<<mid<<endl; 38 } 39 } 40 return 0; 41 }
标签:tor less main code psu size 建立 priority mat
原文地址:https://www.cnblogs.com/very-beginning/p/12203266.html