Acwing：奶牛摄影（思维+排序深探好题 Python）

题目链接🔗：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)