算法竞赛进阶指南 七夕祭

七夕节因牛郎织女的传说而被扣上了「情人节」的帽子。

于是 TYVJ 今年举办了一次线下七夕祭。

Vani 同学今年成功邀请到了 cl 同学陪他来共度七夕，于是他们决定去 TYVJ 七夕祭游玩。

TYVJ 七夕祭和 11 区的夏祭的形式很像。

矩形的祭典会场由 NN 排 MM 列共计 N×MN×M 个摊点组成。

虽然摊点种类繁多，不过 cl 只对其中的一部分摊点感兴趣，比如章鱼烧、苹果糖、棉花糖、射的屋……什么的。

Vani 预先联系了七夕祭的负责人 zhq，希望能够通过恰当地布置会场，使得各行中 cl 感兴趣的摊点数一样多，并且各列中 cl 感兴趣的摊点数也一样多。

不过 zhq 告诉 Vani，摊点已经随意布置完毕了，如果想满足 cl 的要求，唯一的调整方式就是交换两个相邻的摊点。

两个摊点相邻，当且仅当他们处在同一行或者同一列的相邻位置上。

由于 zhq 率领的 TYVJ 开发小组成功地扭曲了空间，每一行或每一列的第一个位置和最后一个位置也算作相邻。

现在 Vani 想知道他的两个要求最多能满足多少个。

在此前提下，至少需要交换多少次摊点。

输入格式

第一行包含三个整数 NN 和 MM 和 TT，TT 表示 cl 对多少个摊点感兴趣。

接下来 TT 行，每行两个整数 x,yx,y，表示 cl 对处在第 xx 行第 yy 列的摊点感兴趣。

输出格式

首先输出一个字符串。

如果能满足 Vani 的全部两个要求，输出 both；

如果通过调整只能使得各行中 cl 感兴趣的摊点数一样多，输出 row；

如果只能使各列中 cl 感兴趣的摊点数一样多，输出 column；

如果均不能满足，输出 impossible。

如果输出的字符串不是 impossible， 接下来输出最小交换次数，与字符串之间用一个空格隔开。

数据范围

1≤N,M≤1000001≤N,M≤100000,
0≤T≤min(N∗M,100000)0≤T≤min(N∗M,100000),
1≤x≤N1≤x≤N,
1≤y≤M1≤y≤M

输入样例：

2 3 4
1 3
2 1
2 2
2 3

输出样例：

row 1

解题思路：

 

 

 

解题代码：

#include <cmath>
#include <cstdio>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 100010;

typedef long long LL;

int n, m, cnt;
int c[N], s[N];//c[N]为题解中的c[N], s[N]为前缀和
int row[N], col[N];

LL work(int n, int a[])
{
    for (int i = 1; i <= n; i ++ ) s[i] = s[i - 1] + a[i];
    if (s[n] % n) return -1;//若是不能平均分配则无解
    
    int avg = s[n] / n;
    
    c[1] = 0;
    for (int i = 2; i <= n; i ++ ) c[i] = s[i - 1] - (i - 1) * avg;
    
    LL res = 0;
    sort(c + 1, c + n + 1);//将c[N]从小到大排序,若是不理解可以去学习贪心
    for (int i = 1; i <= n; i ++ ) res += abs(c[i] - c[(n + 1) / 2]);
    
    return res;
}

int main()
{
    cin >> n >> m >> cnt;
    while (cnt -- )
    {
        int x, y;
        scanf("%d %d", &x, &y);
        row[x] ++ , col[y] ++ ;
    }
    
    LL rw = work(n, row);//这里会爆int因为当所有糖果在同一行时所需移动步数为1 + 2 + 3 + ...+ n = n * (n - 1) / 2;
    LL cl = work(m, col);
    
    if (rw != -1 && cl != -1) printf("both %lld\n", rw + cl);
    else if (rw != -1) printf("row %lld\n", rw);
    else if (cl != -1) printf("column %lld\n", cl);
    else puts("impossible");
    
    return 0;
}