求矩阵中上三角形、下三角形、对角线元素和

#include<iostream>
using namespace std;
int main() {
	int matrix[3][3];
	for (int i = 0; i < 3; i++) {
		for (int j = 0; j < 3; j++) {
			cin >> matrix[i][j];//输入一个3*3的矩阵
		}
	}
	int a = 0, b = 0, c = 0;
	int(*p)[3] = matrix;
	for (int i = 0; i < 3; i++) {
		for (int j = i; j < 3; j++) {
			a += *(*(p + i) + j);//计算上三角形元素和
		}
	}
	for (int i = 0; i < 3; i++) {
		for (int j = 0; j <= i; j++) {
			b += *(*(p + i) + j);//计算下三角形元素和
		}
	}
	for (int i = 0; i < 3; i++) {
		int j = i;
		c += *(*(p + i) + j);//计算对角线元素和
	}
	cout << "上三角形=" << a << "下三角形=" << b << "对角线=" << c;
			return 0;
		}