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

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

题目描述

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

C

#include<stdio.h>
#include<stdlib.h>
int sum(int m,int n)
{
	int s=0,i,j,x;
	int *a;
	x=m*n;
	a=(int*)malloc(sizeof(int)*x);
	for(i=0;i<m;i++)
		for(j=0;j<n;j++)
		{
			scanf("%d",&a[i*n+j]);
			if(i==0 || i==m-1)
				s=s+a[i*n+j];
			else if(j==0 || j==n-1)
				s=s+a[i*n+j];
		}
	free(a);
	return s;
}
int main()
{
	int m,n,s;
	scanf("%d%d",&m,&n);
	s=sum(m,n);
	printf("%d\n",s);
	return 0;
}