PTA 兩個有序序列的中位數（詳解）

問題描述

已知有兩個等長的非降序序列S1, S2, 設計函數求S1與S2並集的中位數。有序序列A0​​,A​1,⋯,A​N−1的中位數指A​(N−1)/2的值,即[(N+1)/2]個數（A​0爲第1個數）。

輸入格式:

輸入分三行。第一行給出序列的公共長度N（0<N≤100000），隨後每行輸入一個序列的信息，即N個非降序排列的整數。數字用空格間隔。

輸出格式:

在一行中輸出兩個輸入序列的並集序列的中位數。

輸入樣例1:

5
1 3 5 7 9
2 3 4 5 6

輸出樣例1:

4

輸入樣例2:

6
-100 -10 1 1 1 1
-50 0 2 3 4 5

輸出樣例2:

1

關鍵

本題關鍵在於題目指出輸入爲整數位非降序排列，就是有可能爲升序，有可能爲等序。

第一種思路（最大N超時）

只使用一個數組，將輸入的兩行整數都存到一個數組裏，再進行排序，最後輸出數組的中位數即可。

#include<stdio.h>
void sort(int data[], int n) {
    int i,j,mix,temp;
    for(i=0;i<n-1;i++) {
        int mix = i;
        for(j=i+1;j<n;j++) {
            if(data[j] < data[mix]) {
                mix = j;
            }
        }
        if(i != mix) {
            temp = data[i];
            data[i] = data[mix];
            data[mix] = temp;
        }
    }
}
void main(){
	int n,i;
	int str[2000001];
	scanf("%d",&n);
	for(i=0;i<n;i++){
		scanf("%d",&str[i]);
	}
	for(i=n;i<2*n;i++){
		scanf("%d",&str[i]);
	}
	sort(str,2*n);
	/*for(i=0;i<2*n;i++){
		printf("%d",str[i]);
	}*/
	int a=(2*n-1)/2;
	printf("%d\n",(str[a]);
	}
}

第二種思路

循環比較兩個數組的中位數的大小，直到兩個數組的起始下標等於結束下標時返回兩個數中小的那個數，就是整個數組的中位數。

#include <stdio.h>
#include <stdlib.h>
 
int Median_Search(int a[] , int b[] , int n){
    int first_a = 0 , first_b = 0 , end_a = n-1 , end_b = n-1 ;
    int mid_a , mid_b ;
    while( first_a != end_a || first_b != end_b ){
        mid_a = (first_a+end_a)/2 ;
        mid_b = (first_b+end_b)/2 ;
        if(a[mid_a]==b[mid_b])
            return a[mid_a];
        if(a[mid_a]<b[mid_b]){
            if( (first_a+end_a)%2 == 0 ){       //元素個數是奇數
                first_a = mid_a ;
                end_b = mid_b ;
            }
            else{
                first_a = mid_a + 1 ;
                end_b = mid_b ;
            }
        }
        else{
            if( (first_a+end_a)%2 == 0 ){
                end_a = mid_a ;
                first_b = mid_b ;
            }
            else{
                end_a = mid_a ;
                first_b = mid_b +1 ;
            }
        }
    }
    return a[first_a]<b[first_b]?a[first_a]:b[first_b];
}
int main()
{
    int n,a[50],b[50];
    scanf("%d",&n);
    for(int i = 0 ; i < n ; i ++){
        scanf("%d",&a[i]);
    }
    for(int i = 0 ; i < n ; i ++){
        scanf("%d",&b[i]);
    }
    printf("%d",Median_Search(a,b,n));
    return 0;
}

第三種思路

從下標爲0開始比較ab兩個數組的大小，如果a大，b的下標向後移一位。如果b大，a的下標向後移一位。知道a,b的下標相加等於n-1時，退出循環。比ab此時下標小的數和比ab下標大的數相等，ab此時的下標對應的兩個數中小的那個數一定是中位數。

#include <stdio.h>
int main(){
	int n, i, j = 0, k = 0;
	scanf("%d", &n);
	int a[n], b[n];
	for(i = 0; i < n; i++)
		scanf("%d", &a[i]);
	for(i = 0; i < n; i++)
		scanf("%d", &b[i]);
	while(j + k < n - 1){
		if(a[j] >= b[k])
			k++;
		else
			j++;
	}
	printf("%d", a[j]>b[k]?b[k]:a[j]);
	return 0;
}

最後

辣雞CSDN，我辛辛苦苦寫了半天的博客突然就沒了，還要重新寫一遍，你們應該殺個程序員謝罪！！！生氣.jpg