ABC420のB問題のコード修正をお願いします。
ABC420のB問題について教えていただけないでしょうか?
テストケース27個中9個ACで、それ以外WAでした。
#include
using namespace std;
#define rep(i,a,b) for(int i=a;i<b;i++)
using ll = long long; using ld = long double;
int main() {
ll n,m; cin >> n >> m;
vector point(n,0);
vector> masu(n,vector (m));
rep(i,0,n)rep(j,0,m) cin >> masu[i][j];
ll cnt1 = 0, cnt0 = 0;
vector index0,index1;
rep(j,0,m) {
index1.clear();
index0.clear();
cnt0 = 0;
cnt1 = 0;
rep(i,0,n) {
if(masu[i][j] == 0) cnt0++,index0.push_back(i);
else if(masu[i][j] == 1) cnt1++,index1.push_back(i);
}
if(cnt0 == 0 || cnt1 == 0) rep(i,0,point.size()) point[i]++;
else if(cnt0 < cnt1) rep(i,0,index0.size()) point[index0[i]]++;
else if(cnt0 > cnt1) rep(i,0,index1.size()) point[index1[i]]++;
}
//sort(point.begin(),point.end(),greater());
auto& ans = *max_element(point.begin(),point.end());
rep(i,0,point.size()) {
if(point[i] == ans) {
cout << i+1 << " ";
}
}
cout << endl;
return 0;
}