进阶实验1-3.1 两个有序序列的中位数 (25 分)

已知有两个等长的非降序序列S1, S2, 设计函数求S1与S2并集的中位数。有序序列A0, A1,…, AN-1 的中位数指 A(N−1)/2 的值,即第 ⌊(N+1)/2⌋ 个数（A0为第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

---

AC:

• V e c t o r

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

int main()
{
    std::ios_base::sync_with_stdio(false);
    int n;
    cin >> n;
    vector<int> s;
    for (int i = 0; i < 2 * n;i++)
    {
        int a;
        cin >> a;
        s.push_back(a);
    }
    sort(s.begin(), s.end());
    cout << s[(s.size() - 1) / 2] << endl;
    return 0;
}

• 这里提供下实验书上的一个思路：
  由于原序列有序，可以借鉴二分法。
  假设 S1 为 A0, A1, …, AN-1 ,
  S2 为 B0, B1, …, BN-1,
  其中 S1中位数为 A(N-1)/2 ,S2中位数为B(N-1)/2 .
• 若记最终并集的中位数为 M，
  则可证明 min(A(N-1)/2,B(N-1)/2)<=M<=max(A(N-1)/2,B(N-1)/2).
  因此，当 N为1时，中位数 M=min(A0,B0)。
  一般情况下，若两个中位数相等，即 A(N-1)/2==B(N-1)/2,
  则 M=A(N-1)/2 ;
  若不相等，即 A(N-1)/2 != B(N-1)/2 ,
  不妨假设 A(N-1)/2 < B(N-1)/2,
  可证明，序列 S1从 A(N-1)/2 左面去除 k个数，
  同时序列 S2从 B(N-1)/2 右面去除 k个数后，
  新序列并集的中位数仍然为 M。
  因此，可转换为求两个新序列的中位数。

---

WA：

 #include <bits/stdc++.h>
using namespace std;
int main()
{
    std::ios_base::sync_with_stdio(false);
    int n;
    cin >> n;
    int *A = new int();
    int *B = new int();
    for (int i = 0; i < n; i++)
        cin >> A[i];
    for (int j = 0; j < n; j++)
        cin >> B[j];
    int as = 0, ae = n - 1;
    int bs = 0, be = n - 1;
    int am, bm, k;
    //s表示起点start，e表示终点end
    while (n > 1)
    {
        am = (as + ae) / 2;
        bm = (bs + be) / 2;
        //寻找中位数
        if (n != 2)
            k = (n - 1) / 2;
        else
            k = n / 2;
        if (A[am] == B[bm])
        {
            cout << A[am] << endl;
            return 0;
        }
        else if (A[am] < B[bm])
        {
            as += k;
            be -= k;
        }
        else
        {
            ae -= k;
            bs += k;
        }
        n = n - k;
        //迭代求 A，B数组中剩下的最后一位数
    }
    if(A[ae] < B[be])
        cout << A[ae] << endl;
    if(A[ae] > B[be])
        cout << B[be] << endl;
    delete A;
    delete B;
    return 0;
}

感觉也没什么问题…

• 结果
  
  改了半天，还是不对！
  难道是c++ 中的 new 与c 中 的 malloc 区别？
  试了一下，果真 ！

---

AC：

#include <bits/stdc++.h>
using namespace std;
int main()
{
    std::ios_base::sync_with_stdio(false);
    int n;
    cin >> n;
    int* A = (int*)malloc(n * sizeof(int)); 
    int* B = (int*)malloc(n * sizeof(int));
    for (int i = 0; i < n; i++)
        cin >> A[i];
    for (int j = 0; j < n; j++)
        cin >> B[j];
    int as = 0, ae = n - 1;
    int bs = 0, be = n - 1;
    int am, bm, k;
    //s表示起点start，e表示终点end
    while (n > 1)
    {
        am = (as + ae) / 2;
        bm = (bs + be) / 2;
        //寻找中位数
        if (n != 2)
            k = (n - 1) / 2;
        else
            k = n / 2;
        if (A[am] == B[bm])
        {
            cout << A[am] << endl;
            return 0;
        }
        else if (A[am] < B[bm])
        {
            as += k;
            be -= k;
        }
        else
        {
            ae -= k;
            bs += k;
        }
        n = n - k;
        //迭代求 A，B数组中剩下的最后一位数
    }
    if(A[ae] < B[be])
        cout << A[ae] << endl;
    if(A[ae] > B[be])
        cout << B[be] << endl;
    free(A);
    free(B);
    return 0;
}

---

听闻 c++ 中有个迭代器。
不会用！ಥ_ಥ