AcWing_105 七夕祭

七夕祭

题目链接

AcWing_105 七夕祭

题目描述

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

于是 $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$， 接下来输出最小交换次数，与字符串之间用一个空格隔开。

数据范围

$$\begin{gathered} 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 \end{gathered}$$

输入样例：

2 3 4
1 3
2 1
2 2
2 3

输出样例：

row 1

思路

由于交换同一列相邻两行的摊位只会改变相关行的“感兴趣摊点数”，交换同一行相邻两列的摊位只会改变相关列的“感兴趣摊点数”，所以可以将问题分成“各列相邻两行的交换”和“各行相邻两列的交换”两部分来解。

以“各行相邻两列的交换”为例。

设 $a_i$ 为每一列的“感兴趣摊点数”，其中 $1\le i\le m$ ，若问题有解，那么交换完成后，每个 $a_i^{\prime }$ 一定都是相等的，即 $t$ 可以被 $m$ 整除。这样问题就变成了“均分纸牌”问题。

最优的结果一定是只与右侧（或左侧）的摊位交换的。

设 $x_i$ 为第 $i$ 列与 第 $i-1$ 列交换的次数（可以为负），$\bar{a}=i\ast \frac{t}{m}$，那么我们有：
$$\begin{gathered} x_1=x_1 \\ {x}_2=a_2+x_1-\bar{a} \\ {x}_3=a_3+x_2-\bar{a} \\ {x}_4=a_4+x_3-\bar{a} \\ {x}_i=a_i+x_{i-1}-\bar{a} \end{gathered}$$

如果把所有式子都用 $x_1$ 来表示，则有：
$$\begin{gathered} x_1=x_1-0 \\ {x}_2=x_1-(\bar{a}-a_2) \\ {x}_3=x_1-(2\times \bar{a}-a_2-a_3) \\ {x}_4=x_1-(3\times \bar{a}-a_2-a_3-a_4) \\ {x}_i=x_1-(i\times \bar{a}-\sum _{j=1}^i{a}_i) \end{gathered}$$

于是问题就变成了，求一个最优的 $x_k$，使得 ${\sum }_{i=1}^n|x_k-(i\times \bar{a}-{\sum }_{j=1}^i{a}_i)|$ 最小。这实际上就是“货仓选址”问题，其中 $|x_k-(i\times \bar{a}-{\sum }_{j=1}^i{a}_i)|$ 就相当于“货仓选址”问题中每个货仓的地址。很显然，只有 $x_k$ 为所有 $x_i$ 的中位数时，问题的答案才有最小值，详情请见。

由于：$$x_k=\sum _{i=1}^n|x_k-(i\times \bar{a}-\sum _{j=1}^i{a}_i)|$$
所以答案的式子还可以表示成：$$\sum _{i=1}^n|(i\times \bar{a}-\sum _{j=1}^k{a}_j)-(i\times \bar{a}-\sum _{j=1}^i{a}_i)|$$

若设 $s_i=i\times \bar{a}-{\sum }_{j=1}^i{a}_i$，那么最后答案的式子就变成了：
$$\sum _{i=1}^n|s_k-s_i|$$

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<vector>
#include<algorithm>
using namespace std;
long long n,m,t,x,y;
long long row[100001],col[100001];
long long solve(long long a[100001],long long x)
{
	int f[100001],s[100001],ans,k;
	for(int i=1;i<=x;i++)
	{
		f[i]=a[i]-t/x;
		s[i]=s[i-1]+f[i];
	}
	sort(s+1,s+x+1);
	if(n&1)//求中位数 
		k=(x+1)/2;
	else
		k=x/2;
	ans=0;
	for(int i=1;i<=x;i++)
		ans+=abs(s[i]-s[k]);
	return ans;
}
int main()
{
	scanf("%lld%lld%lld",&n,&m,&t);
	for(int i=1;i<=t;i++)
	{
		scanf("%lld%lld",&x,&y);
		row[x]++,col[y]++;
	}
	if(t%n==0&&t%m==0)
		printf("both %lld",solve(row,n)+solve(col,m));
	else if(t%m==0)
		printf("column %lld",solve(col,m));
	else if(t%n==0)
		printf("row %lld",solve(row,n));
	else
		printf("impossible");
	return 0;
}