1306.Sequence Median（堆排序）

1306

URAL真是没水题 以为简单的排序就好了 ME  内存限制很紧 堆排序 或者 STL

用堆排序做的 正好复习一下 都忘了

#include <iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<stdlib.h>
using namespace std;
#define N 1250010
int a[N];
void adjust(int i,int n)
{
    int j;
    a[0] = a[i];
    j = 2*i;
    while(j<=n)
    {
        if(j<n&&a[j+1]>a[j])
        j++;
        if(a[0]<a[j])
        {
            a[i] = a[j];
            i = j;
            j = 2*i;
        }
        else
        break;
    }
    a[i] = a[0];
}
int main()
{
    int i,j,n,t,x;
    scanf("%d",&n);
    int k = n/2+1;
    for(i = 1; i <= k ; i++)
    scanf("%d",&a[i]);
    if(n==2)
    {
        printf("%.1lf
",(a[2]/2.0+a[1]/2.0));
        return 0;
    }
    for(i = k/2 ; i>= 1 ; i--)
    adjust(i,k);
    for(i = k+1 ; i <= n ; i++)
    {
        scanf("%d",&x);
        if(x<a[1])
        {
            a[1] = x;
            adjust(1,k);
        }
    }
    for(i = k ; i >1 ; i--)
    {
        t = a[1];
        a[1] = a[i];
        a[i] = t;
        adjust(1,i-1);
    }
    if(n%2!=0)
    printf("%.1lf
",1.0*a[k]);
    else
    printf("%.1lf
",a[k-1]/2.0+a[k]/2.0);
    return 0;
}