Careercup - Google面试题 - 6271724635029504

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 = ;
     j = nb - ;

     int sum;
     while (i <= na -  && j >= ) {
         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) ==  && (na >  && nb > )) {
         a.resize(na);
         b.resize(nb);
         for (i = ; i < na; ++i) {
             scanf("%d", &a[i]);
         }
         for (i = ; i < nb; ++i) {
             scanf("%d", &b[i]);
         }
         while (scanf("%d", &target) == ) {
             ia = ib = -;
             if (twoSortedArraySum(a, b, target, ia, ib)) {
                 printf("%d + %d = %d\n", a[ia], b[ib], target);
             } else {
                 printf("Not found.\n");
             }
         }
     }

     return ;
 }