对角线遍历MxN数组

给你一个大小为 m x n 的矩阵 mat ，请以对角线遍历的顺序，用一个数组返回这个矩阵中的所有元素。

示例 1：
输入：mat = [[1,2,3],[4,5,6],[7,8,9]]
输出：[1,2,4,7,5,3,6,8,9]
示例 2：
输入：mat = [[1,2],[3,4]]
输出：[1,2,3,4]

作者：力扣 (LeetCode)
链接：https://leetcode-cn.com/leetbook/read/array-and-string/cuxq3/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

对角线分为奇偶代表两个遍历方向，分别对两个方向的初始值进行指定，然后按照对应的方向循环更新坐标值，到达边界时循环结束。
需要注意行或列到达边界后初始值的计算。row = i - col; col = i - row;
这种方法不用关心遍历时方向的切换等复杂逻辑，只需要判断到达边界退出循环即可。

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* findDiagonalOrder(int** mat, int matSize, int* matColSize, int* returnSize){

    int col_size = *matColSize;
    int *dst = malloc(matSize * col_size * sizeof(int));
    *returnSize=matSize*(*matColSize);

    int row = 0;
    int col = 0;
    int index = 0;
    for (int i = 0; i < (matSize + col_size - 1); i++)
    {
        if(i % 2)
        {
            col = (i < col_size) ? i : col_size - 1;
            row = i - col;
            while(col >= 0 && row < matSize)
            {
                dst[index] = mat[row][col];
                index++;
                row = row + 1;
                col = col - 1;
            }
        }
        else
        {
            row = (i < matSize) ? i : matSize - 1;
            col = i - row;
            while(col < col_size && row >= 0)
            {
                dst[index] = mat[row][col];
                index++;
                row = row -1;
                col = col + 1;
            }
                
        }    
    }
    return dst;
}