题目地址:
https://www.acwing.com/problem/content/description/107/
七夕节因牛郎织女的传说而被扣上了「情人节」的帽子。于是TYVJ今年举办了一次线下七夕祭。Vani同学今年成功邀请到了cl同学陪他来共度七夕,于是他们决定去TYVJ七夕祭游玩。TYVJ七夕祭和11区的夏祭的形式很像。矩形的祭典会场由 N N N排 M M M列共计 N × M N×M N×M个摊点组成。虽然摊点种类繁多,不过cl只对其中的一部分摊点感兴趣,比如章鱼烧、苹果糖、棉花糖、射的屋……什么的。Vani预先联系了七夕祭的负责人zhq,希望能够通过恰当地布置会场,使得各行中cl感兴趣的摊点数一样多,并且各列中cl感兴趣的摊点数也一样多。不过zhq告诉Vani,摊点已经随意布置完毕了,如果想满足cl的要求,唯一的调整方式就是交换两个相邻的摊点。两个摊点相邻,当且仅当他们处在同一行或者同一列的相邻位置上。由于zhq率领的TYVJ开发小组成功地扭曲了空间,每一行或每一列的第一个位置和最后一个位置也算作相邻。现在Vani想知道他的两个要求最多能满足多少个。在此前提下,至少需要交换多少次摊点。
输入格式:
第一行包含三个整数
N
N
N和
M
M
M和
T
T
T,T
表示
c
l
对多少个摊点感兴趣。接下来
表示cl对多少个摊点感兴趣。 接下来
表示cl对多少个摊点感兴趣。接下来T
行,每行两个整数
行,每行两个整数
行,每行两个整数x,y
,表示
c
l
对处在第
,表示cl对处在第
,表示cl对处在第x
行第
行第
行第y$列的摊点感兴趣。
输出格式:
首先输出一个字符串。
如果能满足Vani的全部两个要求,输出both
;
如果通过调整只能使得各行中cl感兴趣的摊点数一样多,输出row
;
如果只能使各列中cl感兴趣的摊点数一样多,输出column
;
如果均不能满足,输出impossible
。
如果输出的字符串不是impossible
,接下来输出最小交换次数,与字符串之间用一个空格隔开。
数据范围:
1
≤
N
,
M
≤
100000
1≤N,M≤100000
1≤N,M≤100000
0
≤
T
≤
min
(
N
∗
M
,
100000
)
0≤T≤\min(N∗M,100000)
0≤T≤min(N∗M,100000)
1
≤
x
≤
N
1≤x≤N
1≤x≤N
1
≤
y
≤
M
1≤y≤M
1≤y≤M
首先行之间交换和列之间交换是互相独立的,所以可以分别取最小。当然如果摊点数无法均分,那无解;否则的话问题变为糖果传递问题https://blog.csdn.net/qq_46105170/article/details/135894028,分别在行的方向和列的方向求一下然后加起来即可。代码如下:
#include <algorithm>
#include <iostream>
using namespace std;
using ll = long long;
const int N = 1e5 + 10;
int n, m, k;
int a[N], b[N];
ll calc(int *a, int n) {
ll res = 0, sum = 0;
for (int i = 1; i <= n; i++) sum += a[i];
sum /= n;
static ll s[N];
for (int i = 1; i <= n; i++) a[i] -= sum, s[i] = s[i - 1] + a[i];
sort(s + 1, s + 1 + n);
for (int i = 1; i <= n; i++) res += abs(s[i] - s[n / 2 + 1]);
return res;
}
int main() {
scanf("%d%d%d", &m, &n, &k);
for (int i = 1; i <= k; i++) {
int x, y;
scanf("%d%d", &x, &y);
a[x]++, b[y]++;
}
if (k % m && k % n)
puts("impossible");
else if (k % n)
printf("row %lld\n", calc(a, m));
else if (k % m)
printf("column %lld\n", calc(b, n));
else
printf("both %lld\n", calc(a, m) + calc(b, n));
}
时间复杂度 O ( N log N + M log M ) O(N\log N+M\log M) O(NlogN+MlogM),空间 O ( M + N ) O(M+N) O(M+N)。