C++로 행렬의 전체 요소 개수 구하기

이 문제에서는 n개의 행과 m개의 열로 이루어진 행렬이 주어집니다. 우리가 해야 할 일은 이 행렬에 포함된 요소(원소)의 총 개수를 구하고, 그 값을 출력하는 것입니다.

예를 들어 입력이 n = 20, m = 15라면, 행렬에는 20 × 15 = 300개의 요소가 존재하므로 출력 결과는 300이 됩니다.

해결 방법

행렬의 요소 개수는 단순히 행의 개수와 열의 개수를 곱하면 되기 때문에, 이 문제는 매우 간단하게 해결할 수 있습니다.

• 행의 개수 n과 열의 개수 m을 입력받습니다.
• n과 m을 곱한 값을 반환합니다.

return n * m;

구현 예제

아래 코드를 통해 실제 구현 방법을 확인해 보겠습니다.

#include <bits/stdc++.h>
using namespace std;
#define N 100

int solve(int n, int m) {
    return n * m;
}

int main() {
    int n = 20, m = 15;
    cout << solve(n, m);
    return 0;
}

입력

20, 15

출력

300

코드 설명

위 코드에서 solve 함수는 두 개의 정수 매개변수 n(행의 개수)과 m(열의 개수)을 받아 두 값의 곱을 반환합니다. 행렬의 요소 개수는 행 × 열로 계산되기 때문에, 별도의 반복문 없이 O(1)의 시간 복잡도로 즉시 결과를 얻을 수 있습니다. main 함수에서는 n = 20, m = 15를 설정한 뒤 solve 함수를 호출하여 결과인 300을 화면에 출력합니다.