Careercup

2014-05-06 13:23

题目链接

原题：

Finding a pair of elements from two sorted lists(or array) for which the sum of the elements is a certain value. Anyway solution that can do better than O(a.length + b.length)?

题目：给定两个有序的数组，如何从两数组中各选出一个元素使得两元素加起来等于某个目标值。

解法：又是这个“Guy”出的题目，此人想要追求优于O(n + m)的算法。这人代码水平不高，我想他对于算法复杂度的上下界也不知道怎么估计吧。我个人认为不太可能更优化了，因为你找的不是一个元素，而是一对。我的解法，是使用两个iterator，一个在A数组头部，一个在B数组尾部。通过A数组后移，B数组前移来调整相加的结果。这样的算法，复杂度就是O(n + m)的。

代码：

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

bool twoSortedArraySum(vector<int> &a, vector<int> &b, int target, int &ia, int &ib)
{
    int i, j;
    int na, nb;

    na = (int)a.size();
    nb = (int)b.size();

    i = 0;
    j = nb - 1;

    int sum;
    while (i <= na - 1 && j >= 0) {
        sum = a[i] + b[j];
        if (sum > target) {
            --j;
        } else if (sum < target) {
            ++i;
        } else {
            ia = i;
            ib = j;
            return true;
        }
    }
    return false;
}

int main()
{
    vector<int> a, b;
    int na, nb;
    int i;
    int ia, ib;
    int target;
    
    while (scanf("%d%d", &na, &nb) == 2 && (na > 0 && nb > 0)) {
        a.resize(na);
        b.resize(nb);
        for (i = 0; i < na; ++i) {
            scanf("%d", &a[i]);
        }
        for (i = 0; i < nb; ++i) {
            scanf("%d", &b[i]);
        }
        while (scanf("%d", &target) == 1) {
            ia = ib = -1;
            if (twoSortedArraySum(a, b, target, ia, ib)) {
                printf("%d + %d = %d
", a[ia], b[ib], target);
            } else {
                printf("Not found.
");
            }
        }
    }
    
    return 0;
}