LintCode 解题记录 Matrix专题

LintCode Matrix Zig Traversal

按照之字形遍历输出矩阵。跟之前按照螺旋稳的方式遍历输出矩阵的题目很相似。
但是自己做起来就觉得总是这考虑不全那儿考虑不全，改来改去才能AC。
看了网上的解法，觉得比自己的要好，于是记录一下。
思路：
首先要明白之字形是怎么样的一种走法，可以概括为如下方式:
先沿斜右上方走到头，然后选择向右移动一位或者是像下移动一位。
然后沿斜左下方走到头，然后选择向右移动一位或者是像下移动一位。
然后重复上述过程，直到填满res数组。
代码:

    vector<int> printZMatrix(vector<vector<int> > &matrix) {
        // write your code here
        vector<int> res;
        if (matrix.size() == 0 || matrix[0].size() == 0) return res;
        int m = matrix.size(), n = matrix[0].size(), cnt = m * n;
        res = vector<int>(cnt, 0);
        res[0] = matrix[0][0];
        for (int i = 1, r = 0, c = 0; i < cnt;) {
            while (r > 0 && c+1 < n) res[i++] = matrix[--r][++c]; //斜右上方
            if (c+1 < n) res[i++] = matrix[r][++c];
            else if (r+1 < m) res[i++] = matrix[++r][c];
            while (c > 0 && r+1 < m) res[i++] = matrix[++r][--c]; //斜左下方
            if (r+1 < m) res[i++] = matrix[++r][c];
            else if (c+1 < n) res[i++] = matrix[r][++c];
        }
        return res;
    }

LintCode Search a 2D Matrix

给定一个mxn矩阵，每一行都按照从小到大排列，下一行的第一个元素总比上一行的最后一个元素大，要求查找一个数target，时间复杂度是O(logn+ logm)。
思路:
典型的两次二分查找。简要回忆一下三种二分查找的形式。
1）查找给定的数，找不到则返回-1。
2）查找序列中第一个不小于目标值的数，返回其下标。即C++STL之lower_bound。同理于最后一个小于目标值的数。
3）查找序列中第一个大于目标值的数，返回其下标。即C++STL之upper_bound。同理于最后一个小于等于目标值的数。
那么此题就是现在第一列中找到最后一个大于目标值的数，然后在此行再用一次二分查找查找该数，找到则返回true，找不到则返回false。

    bool searchMatrix(vector<vector<int> > &matrix, int target) {
        // write your code here
        if (matrix.size() == 0 || matrix[0].size() == 0) return false;
        int row = matrix.size(), col = matrix[0].size();
        int l = 0, h = row, mid = 0;
        while (l < h) {
            mid = (l+h) >> 1;
            if (matrix[mid][0] <= target) l = mid+1; //小于等于目标值就跳过
            else h = mid;
        }
        int currh = h-1;
        if (currh < 0) return false;
        l =