题目(复杂)
给定一个长度为n的序列a,试求出对于序列a的每一个前缀的终极数x,使得
最小,试求出终极数t(如若有多个终极数t,只需输出最小的那个)
正解
其实就是求中位数…
输入
第一行一个整数n,第二行n个整数,中间用空格隔开
输出
输出一行一个整数,即表示终极数t
样例输入
5
1 2 3 4 5
Sample Input2
10
1 2 3 4 5 6 7 8 9 10
样例输出
2
Sample Output2
3
解题思路
主要是求中位数的时间问题,这里开两个堆,一个小根堆,一个大根堆。只要保证小根堆里的数都大于大根堆里的数,然后小根堆的堆顶就是中位数。
代码(仅供参考,切勿抄袭)
#include<cstdio>
#include<algorithm>
using namespace std;
int a1[500002],a2[500002],t[1000001],n,x,num1,num2;
void upA(int x)
{int t;while (x>1 && a1[x]<a1[x/2]){swap(a1[x],a1[x/2]);x/=2;}
}//小根堆调整
void upB(int x)
{int t;while (x>1 && a2[x]>a2[x/2]){swap(a2[x],a2[x/2]);x/=2;}
}//大根堆调整
void downA(int x)
{int t,y;while (x*2<=num1 && a1[x]>a1[x*2] || x*2+1<=num1 && a1[x]>a1[x*2+1]){y=x*2;if (y+1<=num1 && a1[y+1]<a1[y]) y++;swap(a1[x],a1[y]);x=y;}
}//小根堆调整
void downB(int x)
{int t,y;while (x*2<=num2 && a2[x]<a2[x*2] || x*2+1<=num2 && a2[x]<a2[x*2+1]){y=x*2;if (y+1<=num2 && a2[y+1]>a2[y]) y++;swap(a2[x],a2[y]);x=y;}
}//大根堆调整
int main()
{//freopen("c.in","r",stdin);//freopen("c.out","w",stdout);scanf("%d",&n);for (int i=1;i<=n;i++){scanf("%d",&x);if (num1<=num2){if (x<a2[1]){a1[++num1]=a2[1];upA(num1);downA(1);a2[1]=x;downB(1);//保证小根堆里的数比大根堆大}else{a1[++num1]=x;upA(num1);downA(1);//放入小根堆}}else{if (x>a1[1]){a2[++num2]=a1[1];upB(num2);downB(1);a1[1]=x;downA(1);//保证大根堆里的数比小根堆里的数小}else{a2[++num2]=x;upB(num2);downB(1);//放入大根堆}}t[i]=a1[1];//记录终极数}sort(t+1,t+1+n);printf("%d",t[n/2]);//计算终极数的终极数
}