AcWing 105. 七夕祭（糖果传递小变种）

题目传送门

七夕节因牛郎织女的传说而被扣上了「情人节」的帽子。
于是 TYVJ 今年举办了一次线下七夕祭。
Vani 同学今年成功邀请到了 cl 同学陪他来共度七夕，于是他们决定去 TYVJ 七夕祭游玩。
TYVJ 七夕祭和 11 区的夏祭的形式很像。
矩形的祭典会场由 N 排 M 列共计 N×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， 接下来输出最小交换次数，与字符串之间用一个空格隔开。
数据范围
1≤N,M≤100000,
0≤T≤min(N∗M,100000),
1≤x≤N,
1≤y≤M
输入样例：
2 3 4
1 3
2 1
2 2
2 3
输出样例：
row 1

题意简单解释一下：
一个N*M的矩阵 然后 要求我们 尽量满足 每行或每列的糖果数量相同，
然后我们可以进行糖果的移动（上下左右 首尾是相连的），每次移动n个糖果的代价就n

如果能满足 Vani 的全部两个要求，输出 both；
如果只能满足行输出 row；
如果只能满足列输出 column；
如果均不能满足，输出 impossible。
如果输出的字符串不是 impossible， 接下来输出最小代价，与字符串之间用一个空格隔开。

对了样例输入的意思是

题解：
试想，当一个讲一颗糖果左右移动时，每行的糖果数量是否变化了？
当一个糖果上下移动时,没列的糖果数量是否变化了？
仔细想过之后是没有的;
既然没有影响，那么横向移动和纵向移动是互相独立的 那么就可以独立出来了。
那么这样就转换成了传递糖果问题了！

每个人初始的糖果是a[i]颗
设x[i]是第i个人给第i+1个人的糖果数量

avg是所有糖果的平均值

那么就有 a[1]+x[n]-x[1]==avg
a[2]+x[1]-x[2]==avg
a[3]+x[2]-x[3]==avg;
…
a[n]+x[n-1]-x[n]==avg

移项后：
x[1]=a[1]-avg+x[n];
x[2]=a[2]+x[1]-avg=a[2]+a[1]-2 * avg+x[n];
x[3]=a[3]+x[2]-avg=a[3]+a[2]+a[1]-3 * avg+x[n];
…
x[n-1]=a[n-1]+…a[1]-(n-2) * avg +x[n-2];

我们要求的答案 就是
|x[1]|+|x[2]|+|x[3]|+|x[4]|+…|x[n]|
设c[i]=(i-1)*avg-(a[1]+…a[i]);

|x[1]|+|x[2]|+|x[3]|+|x[4]|+…|x[n]|
=|c[1]-x[n]|+|c[2]-x[n]|+…|x[n]|

而x[1]为自由变量 要是上式中最小 那么 取c中位数就行了

#include<iostream>
#include<algorithm>
using namespace std;
int n,m;
const int N=100000+100;
const int M=100000+100;
int cols[N];
int rows[M];
long long solve(int a[],int n)
{
    int c[N]={0};
    long long sum=0;
    for(int i=1;i<=n;i++)
    {
        sum+=a[i];
    }
    if(sum%n)return -1;
    int avg = sum/n;
    for(int i=2;i<=n;i++)
    {
        c[i]=c[i-1]+a[i]-avg;
    }
    sort(c+1,c+n+1);
    long long res=0;
    for(int i=1;i<=n;i++)
    {
        res+=abs(c[i]-c[n/2+1]);
    }
    return res;
}
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    cin>>m>>n;
    int t;
    cin>>t;
    while(t--)
    {
        int x,y;
        cin>>x>>y;
        cols[y]++;
        rows[x]++;
    }
    long long fc=0;
    long long fr=0;
    fc = solve(cols,n);
    fr = solve(rows,m);
    if(fc!=-1&&fr!=-1)
    {
        cout<<"both "<<fc+fr<<endl;
    }   
    else if(fc!=-1)
    {
        cout<<"column "<<fc<<endl;
    }
    else if(fr!=-1)
    {
        cout<<"row "<<fr<<endl;
    }
    else cout<<"impossible"<<endl;
    return 0;
}