C, Находится ли элемент матрицы (массива) на диагоналях?

В некоторых задачах просят посчитать сумму элементов диагоналей или одной из них, или как-то их особо заполнить, там применяется одна и та же формула, для определения находится ли элемент на диагонали или нет. Но почему бы не сделать функцию или набор функций?

Функция, которая проверяет, находится ли элемент на диагоналях или нет:

bool isOnDiaganal(int i, int j, int size)
{
	return i == j || j==size-i-1 ;
}

Принимает строку, столбец и размер матрицы, а возвращает true или false.

Соответственно, функция для проверки лежит ли элемент на главной диагонали выглядит так:

bool isOnMainDiaganal(int i, int j)
{
	return i == j ;
}

Здесь не нужен даже размер.

А для побочной диагонали:

bool isOnPobDiaganal(int i, int j, int size)
{
	return j==size-i-1 ;
}

Пример решения задачи «След матрицы» с использованием этой функции:

#include <iostream>
#include <ctime>

using namespace std;

bool isOnDiaganal(int i, int j, int size)
{
	return i == j || j==size-i-1 ;
}

int main()
{
	srand(time(NULL));
	int n = 0; 
	int sum = 0;
	cin >> n; // Считываем с клавиатуры n
	int **a = new int* [n]; // Создаем массив указателей
	for (int i = 0; i < n; i++)
	{
		a[i] = new int [n]; // Создаем элементы
	}
	// А дальше работа как с обычным массивом. 
	for (int i = 0; i < n; i++)
	{
		for (int j = 0; j < n; j++)
		{
			a[i][j] = rand() % 10;
			if (isOnDiaganal(i,j,n))
				sum += a[i][j];
			cout << a[i][j] << " "; // Вывести элементы на консольку
		}
		cout << endl; // Двумерный массив. Строка кончилась, переводим строку и на консоли
	}

	cout << "sum = " << sum << endl;

	// Удаление массива
	for (int i = 0; i < n; i++)
	{
		delete[]a[i]; // Удаляем каждый элемент
	}
	delete [] a; // А потом массив
	return 0;
}

1 Response

Добавить комментарий