【ACWing】105. 七夕祭

题目地址：

https://www.acwing.com/problem/content/description/107/

七夕节因牛郎织女的传说而被扣上了「情人节」的帽子。于是TYVJ今年举办了一次线下七夕祭。Vani同学今年成功邀请到了cl同学陪他来共度七夕，于是他们决定去TYVJ七夕祭游玩。TYVJ七夕祭和11区的夏祭的形式很像。矩形的祭典会场由$N$排$M$列共计$N\times M$个摊点组成。虽然摊点种类繁多，不过cl只对其中的一部分摊点感兴趣，比如章鱼烧、苹果糖、棉花糖、射的屋……什么的。Vani预先联系了七夕祭的负责人zhq，希望能够通过恰当地布置会场，使得各行中cl感兴趣的摊点数一样多，并且各列中cl感兴趣的摊点数也一样多。不过zhq告诉Vani，摊点已经随意布置完毕了，如果想满足cl的要求，唯一的调整方式就是交换两个相邻的摊点。两个摊点相邻，当且仅当他们处在同一行或者同一列的相邻位置上。由于zhq率领的TYVJ开发小组成功地扭曲了空间，每一行或每一列的第一个位置和最后一个位置也算作相邻。现在Vani想知道他的两个要求最多能满足多少个。在此前提下，至少需要交换多少次摊点。

输入格式：
第一行包含三个整数$N$和$M$和$T$，T$表示cl对多少个摊点感兴趣。接下来$T$行，每行两个整数$x,y$，表示cl对处在第$x$行第$y$列的摊点感兴趣。

输出格式：
首先输出一个字符串。
如果能满足Vani的全部两个要求，输出both；
如果通过调整只能使得各行中cl感兴趣的摊点数一样多，输出row；
如果只能使各列中cl感兴趣的摊点数一样多，输出column；
如果均不能满足，输出impossible。
如果输出的字符串不是impossible，接下来输出最小交换次数，与字符串之间用一个空格隔开。

数据范围：
$1\le N,M\le 100000$
$0\le T\le \min (N\ast M,100000)$
$1\le x\le N$
$1\le y\le M$

首先行之间交换和列之间交换是互相独立的，所以可以分别取最小。当然如果摊点数无法均分，那无解；否则的话问题变为糖果传递问题https://blog.csdn.net/qq_46105170/article/details/135894028，分别在行的方向和列的方向求一下然后加起来即可。代码如下：

#include <algorithm>
#include <iostream>
using namespace std;
using ll = long long;

const int N = 1e5 + 10;
int n, m, k;
int a[N], b[N];

ll calc(int *a, int n) {
  ll res = 0, sum = 0;
  for (int i = 1; i <= n; i++) sum += a[i];
  sum /= n;
  static ll s[N];
  for (int i = 1; i <= n; i++) a[i] -= sum, s[i] = s[i - 1] + a[i];
  sort(s + 1, s + 1 + n);
  for (int i = 1; i <= n; i++) res += abs(s[i] - s[n / 2 + 1]);
  return res;
}

int main() {
  scanf("%d%d%d", &m, &n, &k);
  for (int i = 1; i <= k; i++) {
    int x, y;
    scanf("%d%d", &x, &y);
    a[x]++, b[y]++;
  }
  if (k % m && k % n)
    puts("impossible");
  else if (k % n)
    printf("row %lld\n", calc(a, m));
  else if (k % m)
    printf("column %lld\n", calc(b, n));
  else
    printf("both %lld\n", calc(a, m) + calc(b, n));
}

时间复杂度$O(N\log N+M\log M)$，空间$O(M+N)$。