CF1774D 每日一题 Same Count One

题目链接：Problem - D - Codeforces

input:

3
3 4
1 1 1 0
0 0 1 0
1 0 0 1
4 3
1 0 0
0 1 1
0 0 1
0 0 0
2 2
0 0
0 1

 output:

1
2 1 1
1
4 2 2
-1

题意：

给定 n 行 m 列的01矩阵，给定一种操作，每次选定任意两行的任意一列，使该两行的这一列的数交换。问最少几次操作，使得矩阵满足所有行的1的个数相等，输出操作次数和每次操作的的方式，无解则输出-1.

思路：

暴力模拟即可。

首先判断是否有解：如果矩阵中1的个数不能被 n 整除，则无解。

首先处理出当前每一行的1的个数，随后直接从列入手遍历，分别存储当前位置是1，当前行1的个数大于平均值的行号，以及当前位置是0，且当前行1的个数小于平均值的行号。一列遍历结束后，对上述两种情况的位置尽可能地多交换，并不断更新当前每行的1的个数，并开vector离线存储答案，最后一并输出即可。

Code

#pragma GCC optimize(3)
#include<bits/stdc++.h>
#define endl "\n"
#define pb push_back
using namespace std;
typedef long long ll;
typedef pair<int,int> PII;
const int mod=1e9+7;
const int N=1e6+10;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3f;

int t,n,m;

struct node{
	int x,y,z;
};

int bot[N];
 
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin>>t;
	while(t--)
	{
		cin>>n>>m; 
		vector<vector<int> > v(n,vector<int>(m+1));
		int sum=0,ssum=0;
		for(int i=0;i<n;i++)
		{
			for(int j=0;j<m;j++)
			{
				cin>>v[i][j];
				sum+=v[i][j];
			}
			ssum+=sum;
			bot[i]=sum;
			sum=0;
		}
		if(ssum%n)
		{
			cout<<-1<<endl;
			continue;
		}
		int k=ssum/n;
		vector<node> ans;
		for(int j=0;j<m;j++)
		{
			vector<int> big,small;
			for(int i=0;i<n;i++)
			{
				if(v[i][j]==1&&bot[i]>k)
				{
					big.pb(i);
				}
				if(v[i][j]==0&&bot[i]<k)
				{
					small.pb(i);
				}
			}
			for(int s=0;s<min(big.size(),small.size());s++)
			{
				--bot[big[s]];
				++bot[small[s]];
				ans.pb({big[s]+1,small[s]+1,j+1});
			}
		}
		cout<<ans.size()<<endl;
		for(auto x:ans)
		{
			cout<<x.x<<" "<<x.y<<" "<<x.z<<endl;
		}
	}
	return 0;
}