题目链接🔗:2049. 奶牛摄影 - AcWing题库
解题思路:
一个很关键的点就在于每个奶牛只会移动一次
所以每次只考虑两个奶牛A<B的话;
我们移动其他奶牛是不会改变AB的前后关系的
只有把B移到A前面,或者把A移到B后面才会改变AB前后关系(最多两种情况)
还有其他3次均是A<B
所以次数可以判断两个数,哪个在前,哪个在后
思路有了 怎么实现排序呢?
如果只用python自带的sort函数 是无法实现这个排序的 因为我们判断大小的条件是:
如果A在B之前的这种情况出现了3次或以上,才说明A<B,反之A>B
所以 这里我们需要引入一个库:functools 并使用其中的cmp_to_key函数
这个函数可以根据我们自定义的函数来进行排序 例如 传入两个参数A,B 如果满足某个条件,返回值为1,则代表A>B / 若返回值是-1 则代表A<B
import functools
N = int(input())
lst = [[0 for i in range(N+1)]for j in range(6)]
for i in range(1,6) :
for j in range(1,N+1) :
x = int(input())
lst[i][x] = j # 注意这里的储存方式:按照读入的顺序,代表的是在第i组中,x是第j个被读入的数字
queue = [i for i in range(N,0,-1)] # 初始化答案队列
def check(a,b) :
global lst
cnt = 0
for i in range(1,6) :
if lst[i][a] < lst[i][b] : cnt += 1 # 如果在第i组中,A的位置比B的位置靠前(小),则cnt加1
if cnt >= 3 : return -1 # 如果cnt大于等于3,则说明A的原位置比B的原位置靠前(小),返回-1 => A<B
else : return 1 # 反之 若小于3,则说明A的原位置比B的原位置靠后(大),返回1 => A>B
queue.sort(key = functools.cmp_to_key(check))
for i in queue :
print(i)