D. Same Count One div1+div2

原题

翻译

题意是给你一个矩阵，输出最小操作数以及操作，使得每一行的1的数量都相等，操作是任意两行同一列的数字交换
input

1. 首先输入测试案例数t（1，2e4）
2. 输入n（2,1e5）,m.（2,1e5)
3. 输入n个数组，每个数组长为m，数组中只包含0,1

output

1. 如果无法完成操作输出-1
2. 如果可以完成操作输出操作次数以及操作方法
3. 操作步骤（2 1 1）意为第2行和第1行的第1个元素进行交换

注：下方代码来自三木森sam
我的解题思路与他基本没有区别，只不过在记录操作方式的时候没有找到合适的办法，这里学习原作者，定义mess结构体，然后用vector储存mess，vector.size()即为操作次数。

#pragma GCC optimize(1)
#pragma GCC optimize(2)
#pragma GCC optimize(3,"Ofast","inline")
#define IOS ios::sync_with_stdio(false), cin.tie(0);
#include<iostream>
#include<map>
#include<set> 
#include<cstdio>
#include<cstring>
#include<vector>
#include<stack>
#include<algorithm>
#include<cmath>
#include<queue>
#include<deque>
using namespace std;
#define int long long
typedef long long ll;
typedef pair<int,int> PAII;
const int N=2e6+10,M=5050,INF=1e18,mod=998244353;
struct mess{
	int x,y,z;
};
vector<mess> res;
int c[N];
vector<int> v[N];
signed main(){
    //IOS;
    int T;
    //T=1;
    cin>>T;
    while(T--)
    {
    	res.clear();
		int n,m;
		cin>>n>>m;
		int sum=0;
		for(int i=1;i<=n;i++) v[i].clear();
		for(int i=1;i<=n;i++)
		{
			int cnt=0;
			v[i].push_back(-1);
			for(int j=1;j<=m;j++)
			{
				int x;
				cin>>x;
				v[i].push_back(x);
				if(x==1)
				{
					sum++;
					cnt++;
				}
			}
			c[i]=cnt;
		}
		if(sum%n!=0)
		{
			cout<<"-1\n";
			continue;
		}
		int x=sum/n;
		for(int i=1;i<=m;i++)
		{
			vector<int> a,b;
			for(int j=1;j<=n;j++) //这里开始是我要学习的部分
			{
				if(c[j]>x&&v[j][i]) a.push_back(j);
				if(c[j]<x&&!v[j][i]) b.push_back(j);
			}
			for(int j=0;j<min(a.size(),b.size());j++)
			{
				res.push_back({a[j],b[j],i});
				c[a[j]]--;
				c[b[j]]++;
			}
		}
		cout<<res.size()<<"\n";
		for(int i=0;i<res.size();i++) cout<<res[i].x<<" "<<res[i].y<<" "<<res[i].z<<"\n";
	}
    return 0;
} 
/*
*/ 

• 这是最后使用mess进行操作的示意图
  
  遍历每一列
  如果c[j]大于x并且第j行第i列为1 ， 则把j加入到vector a中，
  如果c[j]小于x并且第j行第i列为0 ， 则把j加入到vector b中，
  同时控制好c[a[j]]或者c[b[j]]
  （由int x=sum/n;代码可以看出x是每一个数组1应该的个数）

最后遍历a,b将{a[j],b[j],i}加入到res中，打印即可。