题解 | #牛的品种排序I#

# 因为只有2个元素，所以可以不用排序算法，直接安排另一个数组，遍历当前数组 
# 如果数字是0，则插到另一个数组的队首，否则插到队尾，遍历结束返回新数组
# 时间复杂度O(n)，空间复杂度O(n)
#
# 
# @param cows int整型一维数组 
# @return int整型一维数组
#
class Solution:
    def sortCows(self , cows: List[int]) -> List[int]:
        # write code here
        sorted_cows = []
        for cow in cows:
            if cow == 1:
                sorted_cows.append(cow)
            else:
                sorted_cows.insert(0, cow)
        return sorted_cows