七夕祭(中位数）

题目描述

题目链接：

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

七夕节因牛郎织女的传说而被扣上了「情人节」的帽子。

于是 TYVJ 今年举办了一次线下七夕祭。

Vani 同学今年成功邀请到了 cl 同学陪他来共度七夕，于是他们决定去 TYVJ 七夕祭游玩。

TYVJ 七夕祭和 11 区的夏祭的形式很像。

矩形的祭典会场由 N 排 M 列共计 N×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≤N,M≤100000,
0≤T≤min(N∗M,100000),
1≤x≤N,
1≤y≤M
输入样例：
2 3 4
1 3
2 1
2 2
2 3
输出样例：
row 1

思考

如果这道题只有一行，且一行的最后和第一不相连的话，就很容易想到 均分纸牌 。所以说，这道题其实就是均分纸牌的进阶版。
但是这道题并不能贪心。毕竟$O(N^2)$还是吃不消的。
如果使用均分纸牌的方法，那么就枚举从第kk个位置开始，像均分纸牌的方法一样依次向后推。最后取最小值。预计得分 70 分。

可以利用前缀和的思想，求出每一行的感兴趣摊点的个数，将这些前缀和排序，再求出中位数，按照|s1−sk|+|s2−sk|+|s3−sk|+……+|sn−sk|公式求出答案即可。

AC代码

#include<iostream>
using namespace std;
#include <algorithm>
#include<stdio.h>
#include<math.h>

int h[100100],l[100100],c[100100];
long long sum[100100];

long long problem(int n,int a[]){
   //前缀和
  for(int i=1;i<=n;i++) sum[i]=sum[i-1]+a[i];
  if(sum[n]%n) return -1;
  long long avg=sum[n]/n;
  //减去平均数，就是需要移动的步骤
  for(int i=1;i<=n;i++) c[i]=sum[i]-i*avg;
  //排序
  sort(c+1,c+1+n);
  long long ans=0;
  for(int i=1;i<=n;i++) ans+=abs(c[i]-c[(n+1)/2]);
  return ans;
}

int main(){
  int n,m,t;
  cin>>n>>m>>t;
  for(int i=1;i<=t;i++){
    int x,y;
    cin>>x>>y;
    h[x]++,l[y]++;
  }
  long long x=problem(n,h);//行
  long long y=problem(m,l);//列
  if(x==-1&&y==-1) cout<<"impossible"<<endl;
  else if(x!=-1&&y==-1) cout<<"row "<<x<<endl;
  else if(x==-1&&y!=-1) cout<<"column "<<y<<endl;
  else cout<<"both "<<x+y<<endl;

}