1172: 矩阵边界和（指针专题）

题目描述

给定一个m行n列的二维矩阵，求其四周边元素和。1<=m、n<=100000，可能是1行100000列，也可能是10000行50列，但保证矩阵元素不多于500000。你可能不能预定义数组的大小了，你要学会使用动态内存分配哦。你可以动态申请m*n个内存单元，然后用一维数组来存储二维数组，二维数组元素a[i][j]对应一维数组a[i*n+j]，i、j均从0开始。

输入

输入第一行是m和n，然后是一个m行n列的矩阵。

输出

输出一个整数，表示矩阵所有边界元素的和。

样例输入 Copy

3 4
1 2 3 4 
5 6 7 8 
9 5 4 6 

样例输出 Copy

47

#include <stdio.h>//未知内存大小的矩阵边界的和
#include <stdlib.h>

int main(){
	int m,n,i,j;
	int *p=NULL;
	scanf("%d%d",&m,&n);
	p = (int*)malloc(sizeof(int)*m*n);
	
	for(i=0; i<m*n; i++)
		scanf("%d",(p+i));
		
	int sum=0;
	for(i=0; i<m; i++)
		for(j=0; j<n; j++){
			if(i==0||j==0||i==m-1||j==n-1)
				sum+=p[i*n+j];
		}
	
//	if(m==1){
//		for(i=0; i<n; i++)
//			sum+=p[i];	
//	}
//	else{
//		for(i=0; i<n; i++)
//		sum+=p[i]+p[(m-1)*n+i];
//	}
//	
//	for(i=1;i<m-1; i++)
//		for(j=0;j<n;j++)
//			if(j==0||j==n-1)
//				sum+=p[i*n+j];

	printf("%d\n",sum);
	
	free(p);
		
	return 0;
}