185 矩阵的之字形遍历

原题网址：http://www.lintcode.com/zh-cn/problem/matrix-zigzag-traversal/

给你一个包含 m x n 个元素的矩阵 (m 行, n 列), 求该矩阵的之字型遍历。

您在真实的面试中是否遇到过这个题？ 

Yes

样例

对于如下矩阵：

[
  [1, 2,  3,  4],
  [5, 6,  7,  8],
  [9,10, 11, 12]
]

返回 [1, 2, 5, 9, 6, 3, 4, 7, 10, 11, 8, 12]

#include <iostream>
#include <vector>
#include <math.h>
#include <string>
#include <algorithm>
using namespace std;

 vector<int> printZMatrix(vector<vector<int>> &matrix)
 { 
     int row=matrix.size();
     int col=matrix[0].size();
     int total=row*col;
     vector<int> result;
     int i=0;
     int j=0;
     int count=0;

     result.push_back(matrix[i][j]);
     count++;
     
     bool up=true; 

     while(count<total)
     {

         if (up)
         {
             //向上走;
             if ((i-1)>=0&&(j+1)<col)
             {
                 result.push_back(matrix[i-1][j+1]);
                 count++;
                 i=i-1;
                 j=j+1;     
             }
             else 
             {        
                 if ((j+1)<col)
                 {
                     result.push_back(matrix[i][j+1]);
                     count++;
                     j=j+1;
                 }
                 else  //error:写成if((j+i)>=col),因为上一个if中j+1可能发生变化，应该用else直接分成两种情况;
                 {
                     result.push_back(matrix[i+1][j]);
                     count++;
                     i=i+1;
                 }
                 
                  up=false;
             }
         }
         else  //向下走;
         {
             
           if ((i+1)<row&&(j-1)>=0)
         {
             result.push_back(matrix[i+1][j-1]);
             count++;
             i=i+1;
             j=j-1;
            
         }
         else
         {
            
             if ((i+1)<row)
             {
                 result.push_back(matrix[i+1][j]);
                 count++;
                 i=i+1;
             }
             else
             {
                 result.push_back(matrix[i][j+1]);
                 count++;
                 j=j+1;
             }
             
              up=true;
         }
         }     
         
     }
     return result;
 }

参考：

1 https://blog.csdn.net/guoziqing506/article/details/51602261