交换的最小次数

题：D. Same Count One
题意：给出$n$个长度为$m$并且由$0$和$1$组成的数组，可以进行的操作为任意选择两个数组，并且两个数组中选择相同的位置进行交换，问最少交换多少次可以让所有数组中$1$的个数都相同。
参考文章：点我
一些想法：其实对于不在正确位置上的坏元素，最终都是要换掉的。可以这样想，现在把每个数组中的坏元素拿出来然后放在一起，有一个神秘人物暗中操作一下，所有数组中的元素就都是正确的了。这其实可以用环(?)（也许吧，反正我一直是这么想的）来理解，$1$需要换到$4$位置上，$2$需要换到$1$位置上，$3$需要换到$2$位置上，$4$需要换到$3$位置上，求最小交换次数。这里面，$1、2、3、4$都需要出来，那就先让$1$和$4$换，$4$和$3$换，$3$和$2$换，$2$再和$1$换。形成的环就是$1\to 4\to 3\to 2\to 1$，$1$首尾成环，并且是最优的。毕竟最小就需要换这些，那只要这些能满足就是最优的了。
解法：那么对于这一题来说，反正$1$多的要和$1$少的换，那就把$1$的个数排序，然后首尾指两个指针，依次交换。这里要让超过平均值，和没有超过平均值的换。并且只要有一个换到平均值，就不用再换了。这样做是否一定可行？或者一定能找到可以交换的列？先设$1$多的行（超过平均值的）为$q$,$1$少的为$p$。最不好的情况就是在$p$由$1$的位置，$q$全都有，但是$q$其他位置的$1$是可以换给$p$的。
后记（小声$bb$)这个题的范围是$[1,n]$，用$[0,n)$可能会有奇怪的问题（重写一下就行了）
代码：

#include<iostream>
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
const int length = 1e5 + 5;
bool comp(pair<int, int> a, pair<int, int> b)
{
	return a.second < b.second;
}
int main(void)
{
	int t;
	scanf_s("%d", &t);
	for (int i = 0; i < t; i++)
	{
		int n, m;
		scanf_s("%d%d", &n, &m);
		vector<vector<int>> graph(n + 5);
		vector<pair<int, int>> res;
		int sum = 0;
		for (int i = 1; i <= n; i++)
		{
			vector<int> tmp(m + 5);
			graph[i]=tmp;
			int yh = 0;
			for (int j = 1; j <= m; j++)
			{
				scanf_s("%d", &graph[i][j]);
				if (graph[i][j] == 1)
				{
					yh++;
				}
			}
			res.push_back({ i,yh });
			sum += yh;
		}
		if (sum%n)
		{
			printf("-1\n");
			continue;
		}
		sort(res.begin(), res.end(), comp);
		int k = sum / n;
		int l = 0;
		int r = n-1;
		vector<pair<pair<int, int>, int>> edge;
		while (l < r)
		{
			int i = 1;
			while (res[l].second<k&&i <= m && res[r].second>k)
			{
				int idl = res[l].first;
				int idr = res[r].first;
				if (graph[idl][i] == 0 && graph[idr][i] == 1)
				{
					swap(graph[idl][i], graph[idr][i]);
					res[l].second++;
					res[r].second--;
					edge.push_back({ {idl,idr},i });
				}
				i++;
			}
			if (res[l].second == k)l++;
			if (res[r].second == k)r--;
		}
		printf("%d\n", edge.size());
		for (auto u : edge)
			printf("%d %d %d\n", u.first.first, u.first.second, u.second);
	}
}