c++实现下三角矩阵的压缩存储

下三角矩阵中任一元素在一维数组中的下标k与i，j的对应关系为:当i>=j,k=i*(i+1)/2+j；当i<j,k=n*(n+1)/2

IDE为vs2013.

// 2.5.4.1symmetry.cpp : 定义控制台应用程序的入口点。
//下三角矩阵的压缩
//下三角矩阵中任一元素在一维数组中的下标k与i，j的对应关系为:当i>=j,k=i*(i+1)/2+j；当i<j,k=n*(n+1)/2
#include "stdafx.h"
#include<iostream>
using namespace std;
const int N = 5;
int _tmain(int argc, _TCHAR* argv[])
{
	int a[N][N] = { { 3, 0, 0, 0, 0 }, 
							{ 6, 2, 0, 0, 0 },
							{ 4, 8, 1, 0, 0 }, 
							{ 7, 4, 6, 0, 0 }, 
							{ 8, 2, 9, 5, 7 } };
	int array[N * (N + 1) / 2+1] = { 0 };//将n*n个元素存储到n*(n+1)/2+1个长度的一维数组中
	int i, j;
	for (i = 0; i < N; i++)//输出矩阵
	{
		for (j = 0; j < N; j++)
		{
			cout << a[i][j] << " ";
		}
		cout << endl;
	}
	for (i = 0; i < N; i++)
	{
		for (j = 0; j <= i; j++)
		{
			array[i * (i + 1) / 2 + j] = a[i][j]; //存储下三角中的元素
		}
	}
	array[N*(N + 1) / 2 ] = a[0][4];//存储对角线上方的常数
	cout<<"请输入行号和列号：";
	cin>>i>>j;
	cout<<i<<"行"<<j<<"列的元素值是：";
	if (i >= j)
	{
		cout<<array[i * (i + 1)/2 + j]<<endl;//输出下三角的元素
	}
	else
	{
		cout << array[N*(N + 1) / 2] << endl;//输出对角线上方的常量
	}
	system("pause");
	return 0;
}