题目
七夕节因牛郎织女的传说而被扣上了「情人节」的帽子。于是TYVJ今年举办了一次线下七夕祭。Vani同学今年成功邀请到了cl同学陪他来共度七夕,于是他们决定去TYVJ七夕祭游玩。
TYVJ七夕祭和11区的夏祭的形式很像。矩形的祭典会场由N排M列共计N×M个摊点组成。虽然摊点种类繁多,不过cl只对其中的一部分摊点感兴趣,比如章鱼烧、苹果糖、棉花糖、射的屋……什么的。Vani预先联系了七夕祭的负责人zhq,希望能够通过恰当地布置会场,使得各行中cl感兴趣的摊点数一样多,并且各列中cl感兴趣的摊点数也一样多。
不过zhq告诉Vani,摊点已经随意布置完毕了,如果想满足cl的要求,唯一的调整方式就是交换两个相邻的摊点。两个摊点相邻,当且仅当他们处在同一行或者同一列的相邻位置上。由于zhq率领的TYVJ开发小组成功地扭曲了空间,每一行或每一列的第一个位置和最后一个位置也算作相邻。现在Vani想知道他的两个要求最多能满足多少个。在此前提下,至少需要交换多少次摊点。
分析
方法一:
枚举从第 k 个位置开始(1<=k<=n),像均分纸牌的方法一样依次向后推(如果第 i 个位置有
ai 堆牌,那么取 bi 堆牌分给下一堆,若 bi<0 表示从下一堆拿到这一堆)。最后取最小值。
时间复杂度 O(n^2),预计得分 70 分。
方法二:
设 bi 的前缀和为 si。如果从第 k 个位置开始,那么第 i 堆和第 i+1 堆交换的纸牌数就是 |si-sk|。
总代价就是|s1-sk|+|s2-sk|+|s3-sk|+……+|sn-sk|。发现什么了?当 sk 是 s1~sn 中位数的时候,
上式有最小值!所以把 si 排序后,令 sk=s[(n+1)/2],计算代价即可。
时间复杂度 O(nlogn),预计得分 100 分。
程序
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
int n,m,t,r[100001],c[100001],p1[100001],p2[100001];
long long ans;
int main(){
cin>>n>>m>>t;
for (int i=1;i<=t;i++){
int x,y;
cin>>x>>y;
r[x]++;
c[y]++;
}
if ((t%n!=0)&&(t%m!=0)){
cout<<"impossible"<<endl;
return 0;
}
for (int i=1;i<=n;i++) r[i]-=t/n;
for (int j=1;j<=m;j++) c[j]-=t/m;
if (t%n==0){
for (int i=1;i<=n;i++)
p1[i]=p1[i-1]+r[i];
sort(p1+1,p1+1+n);
for (int i=1;i<=n;i++)
ans+=abs(p1[(n+1)/2]-p1[i]);
}
if (t%m==0){
for (int i=1;i<=m;i++)
p2[i]=p2[i-1]+c[i];
sort(p2+1,p2+1+m);
for (int i=1;i<=m;i++)
ans+=abs(p2[(m+1)/2]-p2[i]);
}
if (t%n==0&&t%m==0) cout<<"both ";
else if (t%n==0) cout<<"row ";
else cout<<"column ";
cout<<ans<<endl;
return 0;
}