TYVJ P1924 - [Poetize II]七夕祭

背景 Background

　　七夕节因牛郎织女的传说而被扣上了「情人节」的帽子。于是TYVJ今年举办了一次线下七夕祭。Vani同学今年成功邀请到了cl同学陪他来共度七夕，于是他们决定去TYVJ七夕祭游玩。

描述 Description

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

输入格式 InputFormat

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

输出格式 OutputFormat

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

样例输入 SampleInput [复制数据]

样例输入1
2 3 4
1 3
2 1
2 2
2 3

样例输入2
3 3 3
1 3
2 2
2 3

样例输出 SampleOutput [复制数据]

样例输出1
row 1

样例输出2
both 2

数据范围和注释 Hint

　　对于30% 的数据，N, M≤100。
　　对于70% 的数据，N, M≤1000。
　　对于100% 的数据，1≤N, M≤100000，0≤T≤min(NM, 100000)，1≤x≤N，1≤y≤M。

题解

环形分金币。白书第四页有详解。唉！少看了column那一行wa了将近十次，所以代码改得很丑。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<cmath>
#include<algorithm>
#define ll long long
using namespace std;
int n,m,K;
ll ans,p,row[100002],col[100002],mr[100002],mc[100002];
void init()
{
	scanf("%d%d%d",&n,&m,&K);
	int i,x,y;
	for(i=1;i<=K;i++)
	   {scanf("%d%d",&x,&y);
	    row[x]++; col[y]++;
	   }
}
void dorow()
{
	p=(ll)K/n;
	int i;ll x;
	for(i=1;i<n;i++)
	   mr[i]=(ll)mr[i-1]+row[i]-p;
	sort(mr,mr+n);
	x=mr[n>>1];
	for(i=0;i<n;i++) ans+=(ll)abs(x-mr[i]);
}
void docol()
{
	p=(ll)K/m;
	int i; ll x;
	for(i=1;i<m;i++)
	   mc[i]=(ll)mc[i-1]+col[i]-p;
	sort(mc,mc+m);
	x=mc[m>>1];
	for(i=0;i<m;i++) ans+=(ll)abs(x-mc[i]);
}
void work()
{
	if(K%n!=0&&K%m!=0) {printf("impossible\n"); return;}
	else if(K%n==0&&K%m==0) printf("both ");
	else if(K%n==0) printf("row ");
	else printf("column ");
	if(K%n==0) dorow();
	if(K%m==0) docol();
	printf("%lld\n",ans);
}
int main()
{
	init(); work();
	return 0;
}