剑指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[].empty()) return false;
     int m = arr.size(), n = arr[].size();

     int x = , y = n - ;
     while (x < m && y >= ) {
         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[].empty()) return false;
     int m = arr.size(), n = arr[].size();

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

     for (--k; k; --k) {
         Num top = mins.top(); mins.pop();
         if (top.row < n - ) {
             mins.push(Num(top.row + , top.col, arr1[top.row + ] + arr2[top.col]));
         }
     }

     return mins.top().val;
 }

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