查找任何行的总和是否等于具有 SUM、ANY、SIZE 的任何列的总和

Question

导师给了我一个任务，检查二维矩阵中是否有任何行的总和等于任何列的总和。只允许使用 **1 loop** 和 sum,any,size。

a = [2 3 4; 5 5 2; 1 3 3; 1 1 1]; --- % Just an example %

rows = size(a,1);
cols = size(a,2);
x = [rows, cols];
tA = a'; i = 1;
RowSum = zeros(rows,1)';
ColSum = zeros(1,cols);

while (i<=max(x))

    if (size(a,1)>=i)
        RowSum(i) = sum(a(i,:));
    end
    if (size(tA,1)>=i)
        ColSum(i) = sum(tA(i,:));
    end
  i=i+1;

end

我知道这可能有点乱，但它给了我这份工作。现在我不知道如何检查RowSum 和ColSum 中是否有任何匹配值。 （不能使用intersect）；有什么想法吗？

Answer 1

解决方案及代码讨论

您可以采用此三步流程并通过 sum、size 和 @987654325 实施解决方案@ 和 without using any loop！！

这种实现要遵循的步骤可以分为三个步骤-

• 获取输入矩阵沿行和列的总和。

• 计算沿行总和的所有元素与沿列总和的距离矩阵。 本质上是这些步骤中最重要的一步，否则您至少需要一个循环或bsxfun（用于内部复制）来执行此操作。 现在，这里的距离矩阵计算是用矩阵乘法进行的，是this solution to Speed-efficient classification in Matlab的简化情况。

• 对于所有距离矩阵元素，查看是否有任何元素为零。

实现上述步骤的代码如下所示 -

%// Get the sum of rows and columns into column vectors - A and B respectively
A = sum(M,2)
B = sum(M,1)' %//'

%// Calculate the distance matrix between A and B
nA = size(A,1);
nB = size(B,1);

A_ext(nA,3) = 0;
A_ext(:,1) = 1;
A_ext(:,2) = -2*A;
A_ext(:,3) = A.^2;

B_ext(nB,3) = 0;
B_ext(:,3) = 1;
B_ext(:,1) = B.^2;
B_ext(:,2) = B;

distmat = A_ext * B_ext.'; %//'

%// For all distance matrix values check if any is zero for the final output
out = any(distmat(:)==0)

样品运行和验证

>> M (Sample input matrix)
M =
     2    14     3
     5     6     7
     1     5     5
     3     1     2

可以看出，沿第三行的和为11，与沿第一列的和相同。因此，在距离矩阵中，至少有一个元素必须是zero。

>> A (Sum along rows)
A =
    19
    18
    11
     6
>> B (Sum along columns)
B =
    11
    26
    17
>> distmat (distance matrix between A and B)
distmat =
    64    49     4
    49    64     1
     0   225    36
    25   400   121

如前所述，(3,1) 处有一个零，因此any(distmat(:)==0) 的最终输出将是1，这是预期的最终结果。

Answer 2

这就是我所做的：

ColSum = sum(a,1);
RowSum = sum(a,2)';
r = 0;
for i=1:length(ColSum)
    if (any(RowSum==ColSum(i)))
        r = 1;
    end
end
disp(r);

现在正在考虑如何在没有循环的情况下处理它..

Answer 3

在你用一个循环解决它之后，sum，any，size，我建议你阅读bsxfun并找到一个没有循环的单行解决方案bsxfun，@987654327 @,any（鼠标悬停查看）：

any(any(bsxfun(@eq, sum(a,1), sum(a,2))))