七夕祭
题目描述
题意解释
题目的大体意思就是:通过调整,使得每一行中cl感兴趣的摊点数量相同,每一列中cl感兴趣的摊点数量相同。要注意题目中说的相邻的含义,每一行或每一列的第一个位置和最后一个位置也是相邻的。这就可以理解为环。
核心思路
经过分析,发现了这道题目有2个重要的性质:
- 交换左右相邻的摊点只会改变某两列中cl感兴趣的摊点个数,并不会改变每行中cl感兴趣的摊点个数。
- 交换上下相邻的摊点只会改变某两行中cl感兴趣的摊点个数,并不会改变每列中cl感兴趣的摊点个数。
因此,我们可以把本题分成独立的两部分进行计算:
- 通过最少次数的左右交换使得每列中cl感兴趣的摊点个数都是相同的。
- 通过最少次数的上下交换使得每行中cl感兴趣的摊点个数都是相同的。
问题:怎样才能用最少的步数使得每行中/每列中cl感兴趣的摊点个数相同呢?
其实这个问题就类似于均分纸牌或者糖果传递。分析如下
设有 n n n行 m m m列。则每一列中一共含有n个格子,每个格子里面有很多摊位(其中或许包含了cl感兴趣的摊位),假设第一个格子有 a 1 a_1 a1个cl感兴趣的摊点,第二个格子有 a 2 a_2 a2个cl感兴趣的摊点, ⋯ \cdots ⋯,第n个格子有 a n a_n an个cl感兴趣的摊点。那么对这n个格子均分后,设每个格子都含有 a v e ave ave个al感兴趣的摊点。则 a v e = a 1 + a 2 + ⋯ + a n n ave=\dfrac{a_1+a_2+\cdots +a_n}{n} ave=na1+a2+⋯+an。
于是有以下式子
- a 1 − x 1 + x 2 = a v e a_1-x_1+x_2=ave a1−x1+x2=ave
- a 2 − x 2 + x 3 = a v e a_2-x_2+x_3=ave a2−x2+x3=ave
- ⋯ \cdots ⋯
- ⋯ \cdots ⋯
- a n − 1 − x n − 1 + x n = a v e a_{n-1}-x_{n-1}+x_n=ave an−1−xn−1+xn=ave
- a n − x n + x 1 = a v e a_n-x_n+x_1=ave an−xn+x1=ave
将含有 x x x的项都留在等式左边,其他常数移动等式右边,得到以下式子:
- x 1 − x 2 = a 1 − a v e x_1-x_2=a_1-ave x1−x2=a1−ave
- x 2 − x 3 = a 2 − a v e x_2-x_3=a_2-ave x2−x3=a2−ave
- ⋯ \cdots ⋯
- ⋯ \cdots ⋯
- x n − 1 − x n = a n − 1 − a v e x_{n-1}-x_n=a_{n-1}-ave xn−1−xn=an−1−ave
- x n − x 1 = a n − a v e x_n-x_1=a_n-ave xn−x1=an−ave
解出 x i x_i xi的式子,如下
-
x 1 = x 1 − 0 x_1=x_1-0 x1=x1−0,设 c [ 1 ] = 0 c[1]=0 c[1]=0
-
x 2 = x 1 − ( a 1 − a v e ) x_2=x_1-(a_1-ave) x2=x1−(a1−ave),设 c [ 2 ] = a 1 − a v e c[2]=a_1-ave c[2]=a1−ave
-
x 3 = x 1 − ( a 1 + a 2 − 2 a v e ) x_3=x_1-(a_1+a_2-2ave) x3=x1−(a1+a2−2ave),设 c [ 3 ] = a 1 + a 2 − 2 a v e c[3]=a_1+a_2-2ave c[3]=a1+a2−2ave
-
⋯ \cdots ⋯
-
⋯ \cdots ⋯
-
x n = x 1 − ( a 1 + a 2 + a 3 + ⋯ + a n − 1 − ( n − 1 ) a v e ) x_{n}=x_1-(a_1+a_2+a_3+\cdots +a_{n-1}-(n-1)ave) xn=x1−(a1+a2+a3+⋯+an−1−(n−1)ave),设 c [ n ] = a 1 + a 2 + a 3 + ⋯ + a n − 1 − ( n − 1 ) a v e c[n]=a_1+a_2+a_3+\cdots +a_{n-1}-(n-1)ave c[n]=a1+a2+a3+⋯+an−1−(n−1)ave
-
由推导可知, c i c_i ci的递推式为: c [ i ] = s [ i − 1 ] − ( n − 1 ) ∗ a v e c[i]=s[i-1]-(n-1)*ave c[i]=s[i−1]−(n−1)∗ave。
因此,上述式子就可以转换为
- x 1 = x 1 − c [ 1 ] x_1=x_1-c[1] x1=x1−c[1]
- x 2 = x 1 − c [ 2 ] x_2=x_1-c[2] x2=x1−c[2]
- x 3 = x 1 − c [ 3 ] x_3=x_1-c[3] x3=x1−c[3]
- $\cdots $
- $\cdots $
- x n = x 1 − c [ n ] x_n=x_1-c[n] xn=x1−c[n]
最小交换次数= ∣ x 1 ∣ + ∣ x 2 ∣ + ⋯ + ∣ x n ∣ |x_1|+|x_2|+\cdots +|x_n| ∣x1∣+∣x2∣+⋯+∣xn∣。而由上述式子可知,最小交换次数= ∣ x 1 − c [ 1 ] ∣ + ∣ x 1 − c [ 2 ] ∣ + ⋯ + ∣ x 1 − c [ n ] ∣ |x_1-c[1]|+|x_1-c[2]|+\cdots +|x_1-c[n]| ∣x1−c[1]∣+∣x1−c[2]∣+⋯+∣x1−c[n]∣。可以把 c [ 1 ] , c [ 2 ] , ⋯ , c [ n ] c[1],c[2],\cdots ,c[n] c[1],c[2],⋯,c[n]按从小到大排好序后放到数轴上,那么 ∣ x 1 − c [ 1 ] ∣ + ∣ x 1 − c [ 2 ] ∣ + ⋯ + ∣ x 1 − c [ n ] ∣ |x_1-c[1]|+|x_1-c[2]|+\cdots +|x_1-c[n]| ∣x1−c[1]∣+∣x1−c[2]∣+⋯+∣x1−c[n]∣的几何意义就是在数轴上找到一个点 x 1 x_1 x1,使得该点 x 1 x_1 x1到其他点 c 1 , c 2 , ⋯ , c n c_1,c_2,\cdots ,c_n c1,c2,⋯,cn的距离最小。这就和货舱选址一题完全相同了。由此可知,点 x 1 x1 x1应该取 c 1 , c 2 , ⋯ , c n c1,c2,\cdots ,c_n c1,c2,⋯,cn的中位数。当 n n n是奇数时,在 c [ ( n + 1 ) / 2 ] c[(n+1)/2] c[(n+1)/2]处最优;当 n n n为偶数时,在 c [ n / 2 + 1 ] c[n/2+1] c[n/2+1]处最优。
总的最小交换次数=行的最小交换次数+列的最小交换次数
代码
#include<iostream>
#include<algorithm>
using namespace std;
const int N=100010;
typedef long long LL;
int c[N],s[N],row[N],col[N];
int n,m,T;
LL work(int n,int a[])
{
//先预处理出前缀和
for(int i=1;i<=n;i++)
s[i]=s[i-1]+a[i];
//假设有n行,如果总的摊点数s[n]不能够整除n,那么就说明不能均分
if(s[n]%n)
return -1;
//求出这一行或列中,这n个小格子里面均分到的cl感兴趣的摊点个数,设为ave
int ave=s[n]/n;
c[1]=0;
//按照递推式,依次求出c[2]~c[n]
for(int i=2;i<=n;i++)
c[i]=s[i-1]-(i-1)*ave;
//给c[1],c[2]..c[n]在数轴上排序
sort(c+1,c+1+n);
LL res=0;
//int center=n/2+1; //如果n是偶数则写成这种形式
int center=(n+1)/2; //如果n是奇数则写成这种形式
for(int i=1;i<=n;i++)
res+=abs(c[i]-c[center]);
return res;
}
int main()
{
scanf("%d%d%d",&n,&m,&T);
while(T--)
{
int x,y;
scanf("%d%d",&x,&y);
row[x]++; //cl对第x行的中的某个摊点感兴趣的个数+1
col[y]++; //cl对第y列中的某个摊点感兴趣的个数+1
}
LL row_num=work(n,row); //row_num是行的最少交换次数
LL col_num=work(m,col); //col_num是列的最少交换次数
//通过调整,可以同时满足两个要求,那么总的最少交换次数=行的最少交换次数+列的最少交换次数
if(row_num!=-1&&col_num!=-1)
printf("both %lld\n",row_num+col_num);
//通过调整,只能使得各行中cl感兴趣的摊点数一样多,输出行的最少交换次数即可
else if(row_num!=-1)
printf("row %lld\n",row_num);
//通过调整,使得各列中cl感兴趣的摊点数一样多,输出列的最少交换次数即可
else if(col_num!=-1)
printf("column %lld\n",col_num);
//通过调整,都不能满足两个要求
else
puts("impossible");
return 0;
}