使用 C++ 统计列和大于行和的配对数量
给定一个 NXN 大小的矩阵。目标是找到所有有效索引对 (i,j) 的数量,使得列 j 的元素之和大于行 i 的元素之和。
我们将通过遍历矩阵并计算每一行和每一列的元素之和来实现这一点。
将每一行的元素之和存储在 rowsum[N] 中,将每一列的元素之和存储在 colsum[N] 中。
现在将 rowsum[i] 和 colsum[j] 配对,并检查 colsum[j] > rowsum[i] 是否成立。如果成立,则为这样的配对增加计数。
让我们通过示例来理解。
Input-: matrix= { { 1,2,0,1}, { 3,3,0,2}, { 1,3,0,2}, { 3,0,0,2} };
**输出** - 有效配对数量 - 9
**解释** -
Rowsum[0]= 1+2+0+1=5 Colsum[0]= 1+3+1+3=8 Rowsum[1]=3+3+0+2=8 Colsum[1]= 2+3+3+0=8 Rowsum[2]= 1+3+0+2=6 Colsum[2]= 0+0+0+0=0 Rowsum[3]=3+0+0+2=5 Colsum[3]= 1+2+2+2=7 Pairs of (i,j) such that rowsum[i] < colsum[j]. (0,0), (0,1), (0,3), (2,0), (2,1), (2,3), (3,0) (3,1), (3,3)
**输入** -
Arr[]= { {1,1,1}, {1,1,1}, {1,1,1} } N=3
**输出** - 有效配对数量 - 0
**解释** -
Rowsum[0]= 1+1+1=3 Colsum[0]= 1+1+1=3 Rowsum[1]= 1+1+1=3 Colsum[1]= 1+1+1=3 Rowsum[2]= 1+1+1=3 Colsum[2]= 1+1+1=3 No pairs possible where rowsum[i]<colsum[j]
下面程序中使用的方法如下
我们使用一个用随机数初始化的整数数组 Arr[]。
使用一个变量 n 来存储 Arr[] 的长度。
函数 countPairs(int arr[], int n) 以数组及其长度作为输入,并返回满足所需条件的有效配对。
我们使用两个数组 rowsum[n] 和 colsum[n]。
遍历矩阵并将 arr[i][j] 加到 rowsum[i] 和 colsum[j] 中,以计算行 i 和列 j 的和。
现在使用两个 for 循环遍历数组 colsum[] 和 rowsum[]。
如果任何 colsum[j] > rowsum[i]。递增计数。
返回计数作为结果。
示例
#include <bits/stdc++.h> using namespace std; int countPairs(int arr[][3], int n){ // Count of pairs int count = 0; int rowsum[n]={0}; int colsum[n]={0}; int i,j; for (i = 0; i < n; i++){ for (j = 0; j < n; j++){ rowsum[i]+=arr[i][j]; colsum[j]+=arr[i][j]; } } for(i=0;i<n;i++){ for(j=0;j<n;j++) if(colsum[j]>rowsum[i]) { count++; } } return count; } int main(){ int Arr[][3] = { {1,3,5},{2,4,6},{3,5,7} }; int side=3; cout <<endl<<"Count of number of pairs : "<< countPairs(Arr, side); return 0; }
输出
如果我们运行上述代码,它将生成以下输出:
Count of number of pairs : 4
广告