矩阵行和列不相等检查检查列是否只有第一

Question

我必须检查矩阵中每列等于 1 的元素是否有行和列不相等。当我尝试所有解决方案时，它适用于行 == 列。我尝试对每一列进行求和，然后检查它是否等于 6 行，因为我希望列中的所有元素都等于 1。

这是我的输入和输出任务的屏幕截图： task is here as picture

任务描述

在火车时刻表中，火车的路线由一系列位描述（1 - 如果火车停在车站，0 - 如果不是）。
编写一个程序，给出位序列，其中只有那些位为 1，所有火车都停止。

输入

标准输入的第一行 包含列车数量 (1≤T≤100) 和站数（1≤S≤100）。 接下来的 T 行 包含代表每列火车路线的序列： S 位以空格分隔， 其中 1 表示火车（行）停在给定的车站（列）， 0 表示火车不停。

输出

标准输出的第一行应该包含 由空格分隔的 S 位序列， 其中 1 表示所有火车都停站， 0 表示并非所有列车都停止。

示例

输入

6 8
1 1 1 1 1 1 0 1 
1 0 0 0 1 1 0 1 
1 0 0 0 1 0 0 0 
1 1 1 0 1 0 0 0 
1 1 0 1 1 0 0 1 
1 1 1 0 1 0 0 1

输出

1 0 0 0 1 0 0 0

这是我的 C++ 代码：

#include <iostream>
#include <climits>

using namespace std;


int main()
{
    int n,m;
    cin>>n>>m;
    int trainStops[n][m];
    for(int i=0; i<n; i++)
    {
        for(int j=0; j<m; j++)
        {
            cin>>trainStops[i][j];
        }
    }
    int sum =0;
    int output[m]={0};
    for (int i=0; i<n; i++)
    {
        for(int j=0; j<m; j++)
        {
            if(trainStops[j][i]==1){
                output[j]+=trainStops[j][i];
            }
        }
    }
cout<<"==============="<<endl;
    for(int i=0; i<m; i++)
    {
        cout<<output[i]<<" ";
    }

    return 0;
}

Answer 1

如果我有兴趣知道另一种方法，这就是我解决它的方法，提前谢谢你。

#include <iostream>
#include <climits>

using namespace std;


int main()
{
    int n,m;
    cin>>n>>m;
    int trainStops[n][m];
    for(int i=0; i<n; i++)
    {
        for(int j=0; j<m; j++)
        {
            cin>>trainStops[i][j];
        }
    }
    int sum =0;
    int output[m]={0};
    for (int i=0; i<m; i++)
    {
        for(int j=0; j<n; j++)
        {
                output[i]+=trainStops[j][i];

        }
    }

    for(int i=0; i<m; i++)
    {
        if(output[i]==n){
            cout<<1<<" ";
        }
        else{
            cout<<0<<" ";
        }
    }
    cout<<endl;

    return 0;
}