Careercup

2014-05-02 00:30

题目链接

原题：

Given two arrays of sorted integers, merge them keeping in mind that there might be common elements in the arrays and that common elements must only appear once in the merged array.

题目：归并两个已排序的数组，重复元素只能在归并的结果里出现一次。

解法：题意很清晰，只是没有说两个数组里本身是否有重复元素，依题意写代码就可以了。

代码：

// http://www.careercup.com/question?id=4713484755402752
#include <vector>
using namespace std;

class Solution {
public:
    void mergeSortedArray(vector<int> &a, vector<int> &b, vector<int> &c) {
        int na, nb;
        
        na = (int)a.size();
        nb = (int)b.size();
        if (na == 0) {
            c = b;
            return;
        } else if (nb == 0) {
            c = a;
            return;
        }
        
        int i, j;
        i = j = 0;
        while (i < na && j < nb) {
            if (a[i] < b[j])
                c.push_back(a[i++]);
            } else if (a[i] > b[j]) {
                c.push_back(b[j++]);
            } else {
                c.push_back((a[i++], b[j++]));
            }
        }
        while (i < na) {
            c.push_back(a[i++]);
        }
        while (j < nb) {
            c.push_back(b[j++]);
        }
    }
};