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

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

时间限制: 1 Sec 内存限制: 128 MB
提交: 3324 解决: 2101
[状态] [讨论版] [提交] [命题人:admin]
题目描述
给定一个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

#include<bits/stdc++.h>
using namespace std;
int sum(int m,int n)
{
	int s=0,x;
	int *a;
	x=m*n;
	a=(int*)malloc(sizeof(int)*x);//动态内存分配，x是长度 
	for(int i=0;i<m;i++)
	   for(int j=0;j<n;j++)
	   {
	   	cin>>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;
	 cin>>m>>n;
	 s=sum(m,n);
	 cout<<s<<endl;	
 }