七夕祭 C组模拟赛

题目大意：

有一个N*M的矩阵，给出T个特别点，问多少次移动使得各行各列都有数量相同的特别点(每次移动只能与相邻的点进行交换，简单来说就是向上向下向左向右)，每行的第一个点和第N个点相邻，每列的第一个点和第M个点相邻

解题思路：

我们发现行的移动是不会影响列的移动，于是我们可以分开做，把每行 1 的个数看做 n 堆纸牌、每列 1 的个数
看做 m 堆纸牌，题目实际上就是两个环形的均分纸牌问题。
什么？你不知道均分纸牌问题！
下面来说环形均分纸牌问题怎么做：
设有 n 堆纸牌，每堆有 ai 张，所有堆一共有 s 张，那么最终每堆应该有 s / n 张。因此如果 s
mod ai≠0，显然是无解的。接下来构造数列 bi=ai-s/n。
方法一：
枚举从第 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<cstdio>
#include<cctype>
#include<algorithm>
#define gc getchar()
#define ll long long
#define N 100005
using namespace std;
int n,m,K,x,y;
ll ans,p,row[N],col[N],mr[N],mc[N];
void read(int &f)
{
    int d=0; f=1; char c=gc;
    while(!isdigit(c)){if (c=='-') f=-1;c=gc;}
    while(isdigit(c)) {d=d*10+(int)c-48;c=gc;}
    f*=d; return;
}
void write(int x)
{if(x>9) write(x/10);putchar(x%10+48);return;}
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]);
    return;
}
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]);
    return;
}
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);
    return;
}
int main()
{
    read(n);read(m);read(K);
    for (int i=1;i<=K;i++) read(x),read(y),row[x]++,col[y]++;
    work();
    return 0;
}