剑指offer？

1. 在一个m*n二维数组中，每一行都按照从左到右的递增顺排序，每一列都按照从上到下的顺序排序，请完成一个函数，输入这样一个二维数组和一个整数，判断数组中是否含有该整数。
1 2 8 9
2 4 9 12
4 7 9 12
6 8 11 15

杨氏矩阵。这题有更好的做法吗？O(m+n)是最好的吗？网上说的分治法只对m=n的情况吧。

bool find(vector<vector<int> > &arr, int t) {
    if (arr.empty() || arr[0].empty()) return false;
    int m = arr.size(), n = arr[0].size();

    int x = 0, y = n - 1;
    while (x < m && y >= 0) {
        if (t == arr[x][y]) return true;
        else if (t > arr[x][y]) x++;
        else y--;
    }      
    return true;
}

变种1：

求第k个数，我觉得能想出来用堆就不错了。n路归并，O(klgn)，就是维护一个堆。

struct Num {
    int val, row, col;
    Num(int r, int c, int v) : val(v), row(r), col(c) {}
};

class Comparator {
public:
    bool operator() (Num &n1, Num &n2) {
        return n1.val > n2.val;
    }
};

int findKthSmallest(vector<vector<int> > &arr, int k) {
    if (arr.empty() || arr[0].empty()) return false;
    int m = arr.size(), n = arr[0].size();

    priority_queue<Num, vector<Num>, Comparator> mins;
    for (int i = 0; i < k && i < n; ++i) {
        mins.push(Num(0, i, arr[0][i]));
    }
    for (--k; k; --k) {
        Num top = mins.top(); mins.pop(); 
        if (top.row < n - 1) {
            mins.push(Num(top.row + 1, top.col, arr[top.row + 1][top.col]));
        }     
    }
    return mins.top().val;
}

这问题线性的比较难了。http://www.quora.com/How-can-you-efficiently-determine-the-k-th-maximum-element-in-a-MxN-sorted-matrix

可以用一个k大小的堆，O(klgk)。每次pop掉一个数，比如arr[i][j]，然后把arr[i+1][j]和arr[i][j+1]进堆，代价是要有一个数组维护这个数是不是已经在堆里了。

变种2：

给定两个有序数组，长度分别为m和n，分别取出一个数再求和，共有m*n个数，求第k个和（从小到大）。

还是组成一个杨氏矩阵。

int findKthSmallest2(vector<int> &arr1, vector<int> &arr2, int k) {
    if (arr1.empty() || arr2.empty()) return false;
    int m = arr1.size(), n = arr2.size();

    priority_queue<Num, vector<Num>, Comparator> mins;
    for (int i = 0; i < k && i < n; ++i) {
        mins.push(Num(0, i, arr1[0] + arr2[i]));
    }
    
    for (--k; k; --k) {
        Num top = mins.top(); mins.pop(); 
        if (top.row < n - 1) {
            mins.push(Num(top.row + 1, top.col, arr1[top.row + 1] + arr2[top.col]));
        }     
    }

    return mins.top().val;
}

2 输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建二叉树并输出它的后序遍历序列。 （本题可以直接输出来，不用先还原出二叉树)