CH0502 七夕祭

题意

给定一个$n\ast m$的矩阵，其中第一行和最后一行相邻，第一列和最后一列相邻。已知$t$个标记点坐标，每个标记点可以上下左右移动。有两个条件：

1. 每行标记点数相同
2. 每列标记点数相同

问最多能满足多少个条件，并求出最少移动步数。

思路

上下移动只会影响条件一，左右移动只会影响条件二，因此可以把行列分开处理，下面只考虑行，该题转化为环形“均分纸牌”，可通过枚举分割点将环形“均分纸牌”转化为线性“均分纸牌”。下面考虑分割位置为$1$和$n$之间的情况：
设$c_i$为第i位的纸牌数，答案为$ans$；
经推导易得为
$ans=\sum _{i=1}^n|i\ast p-\sum _{j=1}^i{c}_j|=\sum _{i=1}^n|\sum _{j=1}^{i}p-\sum _{j=1}^i{c}_j|=\sum _{i=1}^n|\sum _{j=1}^i(p-c_j)|$，其中$p=\frac{\sum _{i=1}^n{c}_i}{n}$($c_i$的平均数)；
设$s_i=\sum _{j=1}^i(p-c_j)$
特别地 $s_n=\sum _{i=1}^{n}p-\sum _{i=1}^n{c}_i=n\ast \frac{\sum _{i=1}^n{c}_i}{n}-\sum _{i=1}^n{c}_i=0$
则
$ans=\sum _{i=1}^n|s_i|$
若分割位置改为$k$和$k+1$之间，则
$$s_i^{\prime }=\{\begin{array}{ll}{s}_i+s_n-s_k& 1\le i\le k\\ s_i-s_k& k<i\le n\end{array}$$
又因为$s_n=0$
所以$s_i^{\prime }=s_i-s_k$
$an{s}^{\prime }=\sum _{i=1}^n|s_i^{\prime }|=\sum _{i=1}^n|s_i-s_k|$
问题成功转化为“货仓选址”问题，当$s_k$为 { s i } \{s_i\} {si​}中位数时$ans$有最小值。

代码

#include <algorithm>
#include <cstdlib>
#include <cstring>
#include <cctype>
#include <cstdio>
#include <cmath>
#include <iostream>
using namespace std;

#define fo(i, x, y) for (int i = x; i <= y; ++i)
#define fd(i, x, y) for (int i = x; i >= y; --i)

typedef long long ll;

const int maxn = 1e5 + 5;

struct point{int x, y;}p[maxn];
int n, m, t;
ll a[maxn], s[maxn];

int getint()
{
	char ch;
	int res = 0, p;
	while (!isdigit(ch = getchar()) && (ch ^ '-'));
	p = ch == '-'? ch = getchar(), -1 : 1;
	while (isdigit(ch))
		res = (res << 3) + (res << 1) + (ch ^ 48), ch = getchar();
	return res * p;
}

ll getans(ll *a, int n)
{
	ll sum = 0;
	fo(i, 1, n) sum += a[i];
	if (sum % n) return -1;
	fo(i, 1, n) a[i] -= sum / n;
	fo(i, 1, n) s[i] = s[i - 1] + a[i];
	sort(s + 1, s + 1 + n);
	ll ans = 0;
	fo(i, 1, n) ans += abs(s[i] - s[(n + 1) / 2]);
	return ans;
}

int main()
{
	n = getint(); m = getint(); t = getint();
	fo(i, 1, t)
	{
		p[i].x = getint(); p[i].y = getint();
	}
	
	fo(i, 1, t) a[p[i].x]++;
	ll ansx = getans(a, n);
	memset(a, 0, sizeof(a));
	fo(i, 1, t) a[p[i].y]++;
	ll ansy = getans(a, m);
	
	if ((~ansx) && (~ansy)) printf("both %lld\n", ansx + ansy);
		else if (~ansx) printf("row %lld\n", ansx);
			else if (~ansy) printf("column %lld\n", ansy);
				else printf("impossible\n");
	return 0;
}