七夕祭

题目

七夕节因牛郎织女的传说而被扣上了「情人节」的帽子。于是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;
}