c_cpp GFG 2个排序数组的中位数

Posted 2021-05-14

#include <iostream>
#include <bits/stdc++.h>
using namespace std;

// #Sorting #Math #Problem #GFG
// https://www.geeksforgeeks.org/median-of-two-sorted-arrays/

int main() {
	int t;
	cin>>t;
	while(t--){
	    int n;
	    cin>>n;
	    std::vector<int> a(n) ;
	    vector<int> b(n);
	    for(int i=0;i<n;i++){
	        cin>>a[i];
	    }
	    for(int i=0;i<n;i++){
	        cin>>b[i];
	    }
	    vector<int> c(2*n);
	    int i=0;
	    int j=0;
	    int k=0;
	    while(i<n && j<n ){
	     if(a[i]<=b[j]){
	         c[k]=a[i];
	         i++;
	     }else{
	         c[k]=b[j];
	         j++;
	     }
	     k++;
	    }
	    while(i<n){
	        c[k]=a[i];
	        i++;
	        k++;
	    }
	    while(j<n){
	        c[k]=b[j];
	        j++;
	        k++;
	    }
	    for(int x=0;x<2*n;x++){
	        cout<<c[x]<<" ";
	    }
	    cout<<endl;
	    if((n+n)%2!=0){
            return c[(n+n)/2];
        }else{
            n=(n+n)/2;
            double a1=c[n-1];
            double a2=c[n];
	        return ((a1+a2)/2);   
        }
	}
	return 0;
}