东华研究生复试------29 二维数组右上左下遍历

问题描述 :

给定一个row行col列的整数数组array，要求从array[0][0]元素开始，依次取右上到左下的对角线上的元素，按从左上到右下的对角线顺序遍历整个数组。

如3行4列的array：

1 2 4 7

3 5 8 10

6 9 11 12

从右上到左下的对角线有6条，第1条包括数字1，第2条包括数字2和3，第3条包括数字4、5和6，第4条包括数字7、8和9…

依次取这些对角线上的元素输出，即可得到1、2、3、4、5、6、7、8、9、10、11、12

输入说明 :

输入的第一行上有两个整数，依次为row和col。

余下有row行，每行包含col个整数，构成一个二维整数数组。

（注：输入的row和col保证0 < row < 100, 0 < col < 100）

输出说明 :

按遍历顺序输出每个整数。所有整数输出在一行上，每个整数后跟一个空格（行尾也有一个空格）。

输入范例：
3 4
1 2 4 7
3 5 8 10
6 9 11 12
输出范例：
1 2 3 4 5 6 7 8 9 10 11 12

运用我的小脑袋，照葫芦画瓢找规律。发现从右上至左下的对角线分区，左侧部分，列的大方向是每次循环递增1，其中每次循环里，输出行从0开始，然后行++，列-- ，注意一次循环结束的条件；右侧部分就需要考虑数组是行数>列数的情况，还是列数>行数的情况，第二种情况是范例给的，（明显看出，列数为0的在刚才已经全部遍历完了）那么第一种情况呢，自己动手画画就明白（列数为0的还存在没遍历的元素）

#include<stdio.h>
#include<stdlib.h>
int main(){
	int n,m,k,i,j;
	int **a;
	scanf("%d%d",&n,&m);
	a=(int**)malloc(sizeof(int*)*n);
	for(i=0;i<n;i++){
		a[i]=(int)malloc(sizeof(int)*m);
	}
	for(i=0;i<n;i++){
		for(j=0;j<m;j++){
			scanf("%d",&a[i][j]);
		}
	}
	for(j=0;j<m;j++){
		k=j;i=0;
		while(k>=0&&i<n){//行、列的下标都必须合法
			printf("%d ",a[i][k]);
			i++;
			k--;
		}
	}
	for(i=1;i<n;i++){
		k=i;j=m-1;
		if(n>m){//行数大于列数的情况
			while(j>=0&&k<n){
			printf("%d ",a[k][j]);
			k++;
			j--;
			}
		}
		else//列数大于行数的情况
			while(j>0&&k<n){
				printf("%d ",a[k][j]);
				k++;
				j--;
			}
	}
	printf("\n");
	return 0;
}