小红书2024算法实习-编程题，K排序

题目描述:
    在算法中，有各种各样的排序算法，例如归并排序，冒泡排序，快速排序等等。本题中，我们会使用-种新的排序算法:K排序。
    K排序算法描述如下:
    首先，算法需要按照某种规则选择该数列上至多K个位置，将其对应的数抽出来，其他的数都往左对齐，之后这K个数排好序之后依次放在原数列末尾。以上过程算作一次操作。
    例如，对于数列[1,3,5,4,2]，当K=2时可以选择数字5和4，之后数列变成[1,3,2,4,5]。
    你的任务是:对于给定的数列，你需要计算出最少需要多少次上述操作，使得整个数列从小到大排好序?
输入描述：
    第一行一个正整数T，表示有T组数据。
    对于每一组数据，第一行输入两个正整数n,k;
    第二行输入n个数a1,...,an。该序列是一个1~n的数列。
输入：
    2
    5 1
    1 2 3 4 5
    5 2
    1 3 5 4 2
输出：
    0
    2

def K(arr,n,k):
    count = 0
    index = 0
    while index < n:
        nums = []
        i = 0
        while i < k:
            index_m = 0
            while index_m < len(arr)-1:
                if arr[index_m] > arr[index_m + 1]:
                    num_pop = arr.pop(index_m)
                    nums.append(num_pop)
                index_m += 1
                continue
            i += 1
            continue
        if nums:
            nums_new = sorted(nums)
            for num in nums_new:
                arr.append(num)
            count += 1
        index += 1
    print(count)


def main():
    T = int(input())
    i = 0
    while i < T:
        n, k = input().split()
        n = int(n)
        k = int(k)

        arr = list(input())
        for num in arr:
            if num == " ":
                arr.remove(" ")
        K(arr,n,k)
        i += 1
if __name__ == '__main__':
    main()