数组中最大的差值-LintCode

给 m 个数组, 每一个数组均为升序. 现在你可以从两个不同的数组中挑选两个整数(每一个数组选一个)并且计算差值. 我们将两个整数 a 和 b 之间的差定义为它们的绝对差 |a - b|. 你的任务是去找到最大的差值.

注意事项：
每一个给出的数组长度至少为 1. 至少有两个不为空的数组
m 个数组中所有整数的个数和在 [2, 10000]范围内.
m 个数组中所有的整数均将在[-10000, 10000]范围内.

样例：
给一个数组 [[1,2,3], [4,5], [1,2,3]], 返回 4
获得最大差值的一种方式是在第一个数组或第三个数组中取 1, 在第二个数组中取 5.

思路：
由于每个数组都是升序的，只需要选择输出的第一位和最后一位分别作为最小，最大的值。
我们可以求出最小值中的最小值Min和最大值中的最大值Max，之后直接求差值，但是这样在Min和Max存在于同一数组时会出现问题。所以，我们分别维护最大、最小堆，存放元素的值和所在数组的序号。
若堆顶的序号不同，直接返回差值。否则，计算最大堆中的最大值和最小堆中第二小的值之差，以及最小堆的最小值与最大堆中第二大值得差，并取二者最大值。

#ifndef C698_H
#define C698_H
#include<iostream>
#include<vector>
#include<queue>
#include<functional>
#include<utility>
using namespace std;
class Solution {
public:
    /*
    * @param : an array of arrays
    * @return: return the max distance among arrays
    */
    int maxDiff(vector<vector<int>> &arrs) {
        // write your code here
        priority_queue<pair<int, int>> queMax;//维护最大堆
        priority_queue<pair<int, int>, vector<pair<int, int> >, greater<pair<int, int> > > queMin;//维护最小堆
        //堆中存放(元素值,其所在数组的编号)
        for (int i = 0; i < arrs.size();++i)
        {
            if (!arrs[i].empty())
            {
                queMax.push(make_pair(arrs[i].back(), i));
                queMin.push(make_pair(arrs[i][0], i));
            }
        }
        //若堆顶的序号不同，直接返回差值。否则，计算最大堆中的最大值和最小堆中第二小的值之差，以及
        //最小堆的最小值与最大堆中第二大值得差，并取二者最大值
        if (queMax.top().second != queMin.top().second)
        {
            return queMax.top().first - queMin.top().first;
        }
        else
        {
            int maxVal = queMax.top().first;
            queMax.pop();
            int minVal = queMin.top().first;
            queMin.pop();
            return Max(maxVal - queMin.top().first, queMax.top().first - minVal);
        }
    }
    int Max(int a, int b)
    {
        return a > b ? a : b;
    }
};
#endif