CH 0502 七夕祭 (进阶指南，前缀和，中位数)

算法经竞赛进阶指南，34页， 中位数, 前缀和

本题要点：
1、 左右交换，行的摊点数不变，列的摊点数改变。同理，上下交换， 列的摊点数不变，行的摊点数改变。
因此，行的摊点和列的摊点， 可以看做是独立的两个事件；
2、这里以行的摊点为例子；
row[i] 表示第i行有多少摊点；
sum_x 一共有多少个摊点，一共n行。如果，sum_x % n == 0, 那么上下交换，一定可以使得，每一行的摊点数一样。
每一行有 t = sum_x / n 个摊点；

3、 求解过程，类似于 “均分纸牌”。详细看书：
数组a[MaxN] a[i] = row[i] - t;
数组a的前缀和 s[MaxN];
答案 = sum{|s[1] - s[k]|, |s[2] - s[k]|, …, |s[n] - s[k]|}
1 <= k <= n;
根据绝对值的几何意义， s[k] 是数组 s[MaxN] 的中位数时， 和最大。
4、 注意使用 long long
5、 计算绝对值的时候， 先排序， 然后 最大减最小，次大减次小，依次类推。
因为， s[k] 处于中位， 左边都比 s[k] 小，右边都比s[k] 大。展开绝对值式子的时候，抵消掉了。
答案 = sum{|s[1] - s[k]|, |s[2] - s[k]|, …, |s[n] - s[k]|}
= s[k] - s[1] + s[k] - s[2] + … + s[n - 1] - s[k] + s[n] - s[k]
= -s[1] - s[2] - … + s[n - 1] + s[n]

#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int MaxN = 100010;
long long a[MaxN], s[MaxN];
long long row[MaxN], col[MaxN];//row[i] 表示第i行有多少， col[i] 表示第i列有多少个摊点
int n, m, t; 
long long sum_x, sum_y;

int calc(int num, long long sum, long long A[])
{
	s[0] = 0;
	for(int i = 1; i <= num; ++i)
	{
		a[i] = A[i] - sum / num;	
		s[i] = s[i - 1] + a[i];
	}
	sort(s + 1, s + num + 1);
	long long cnt = 0;
	for(int i = 1; i <= num / 2; ++i)
	{
		cnt += s[num + 1 - i] - s[i];	
	}
	return cnt;
}

void solve()
{
	bool r = false, c = false;
	long long cnt_x = 0, cnt_y = 0;
	if(sum_x % n == 0)
	{
		r = true;
		cnt_x = calc(n, sum_x, row);
	}
	if(sum_y % m == 0)
	{
		c = true;
		cnt_y = calc(m, sum_y, col);
	}
	if(r && c)
	{
		printf("both %lld\n", cnt_x + cnt_y);
	}else if(r){
		printf("row %lld\n", cnt_x);
	}else if(c){
		printf("column %lld\n", cnt_y);
	}else{
		printf("impossible\n");
	}
}

int main()
{
	int x, y;
	while(scanf("%d%d", &n, &m) != EOF)
	{
		sum_x = 0, sum_y = 0;
		memset(row, 0, sizeof(row));
		memset(col, 0, sizeof(col));
		scanf("%d", &t);
		for(int i = 0; i < t; ++i)
		{
			scanf("%d%d", &x, &y);
			row[x]++, sum_x++;
			col[y]++, sum_y++;
		}
		solve();
	}
	return 0;
}

/*
2 3 4
1 3
2 1
2 2
2 3
*/

/*
row 1
*/

/*
3 3 3
1 3
2 2
2 3
*/

/*
both 2
*/


``