1. 问题描述:
七夕节因牛郎织女的传说而被扣上了「情人节」的帽子。于是 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
来源:https://www.acwing.com/problem/content/description/107/
2. 思路分析:
首先直接做其实是不好做的,我们需要先看一下题目有什么性质,我们可以先处理行可以发现对列是没有什么影响的,先处理列对行也是没有什么影响的,所以这两个问题是相互独立的,所以可以对这两个问题分开处理,并且由于第一个位置是与最后一个位置相连的所以其实对于行与列的操作其实是一个环形,我们可以使用row,col分别记录每一行与每一列的摊点的数量,可以先操作行然后再操作列,将其转化为下面的环形操作,最终分别转换为数学中的绝对值不等式的问题,先对行进行操作然后再对列表进行操作,先判断是否有解,有解得时候再求解出对应的操作次数即可。
3. 代码如下:
from typing import List
class Solution:
def work(self, n: int, a: List[int]):
s = [0]
# 预处理前缀和
for i in range(1, n + 1):
s.append(s[-1] + a[i])
# 如果不能够整除说明是不满足条件的
if s[n] % n != 0:
return -1
avg = s[n] // n
# 千万要注意长度为n + 1因为python在排序的时候是对整个列表进行排序的需要设置n + 1的长度即可
c = [0] * (n + 1)
res = 0
for i in range(2, n + 1):
c[i] = s[i - 1] - (i - 1) * avg
# 弹出第一个元素防止第一个元素加入排序的过程, 最后将再插进来
c.pop(0)
c.sort()
c.insert(0, 0)
# 当且仅当位置x1位于中点的时候等号成立那么计算|x1-ci|的和
for i in range(1, n + 1):
res += abs(c[i] - c[(n + 1) // 2])
return res
# 环形纸牌问题
def process(self):
n, m, t = map(int, input().split())
row, col = [0] * (n + 1), [0] * (m + 1)
for i in range(t):
x, y = map(int, input().split())
# 统计每一行的与每一列的数量, 下标从1开始
row[x] += 1
col[y] += 1
r = self.work(n, row)
c = self.work(m, col)
# 使用format函数格式化输出
if r != -1 and c != -1:
print("{:} {:}".format("both", r + c))
elif r != -1:
print("{:} {:}".format("row", r))
elif c != -1:
print("{:} {:}".format("column", c))
else: print("impossible")
if __name__ == "__main__":
Solution().process()