LintCode 387. The Smallest Difference

#include <iostream>
#include <vector>
#include <climits>
#include <algorithm>
using namespace std;

/*
Given two array of integer (A and B). Now we are going to find a element in A and another
element in B, so that the difference between A[i] and B[j] |A[i] - B[j]| is as small as
possible, return their smallest difference.
*/

int smallestDifference(vector<int>& A, vector<int>& B) {
sort(A.begin(), A.end());
sort(B.begin(), B.end());
int res = INT_MAX;
int i = 0, j = 0;
while(i < A.size() && j < B.size()) {
int diff = (A[i] - B[j]);
if(diff == 0) return 0;
res = min(res, abs(diff));
if(diff < 0) i ++;
else j++;
}
while(i < A.size()) {
res = min(res, abs(A[i] - B[j - 1]));
i++;
}
while(j < B.size()) {
res = min(res, abs(A[i - 1] - B[j]));
j++;
}
return res;
}

int main(void) {
vector<int> A{3, 6, 7, 4};
vector<int> B{2, 8, 9, 3};
cout << smallestDifference(A, B) << endl;
}