Python-最大正整数

题目：

数组中存储了一堆小于10的非负整数，整个数组从左到右代表一个正整数（如数组[0, 1, 3, 2]代表正整数132）。现给出一个正整数K，要求经过K次数组相邻位置元素交换（必须完成K次交换），使得这个数组代表的数字最大。

例如，

int array[] = {0, 1, 3, 2}, K = 1，则经过1次交换后，数组所能代表的最大值为1032；

int array[] = {0, 1, 3, 2}, K = 2，则经过2次交换后，数组所能代表的最大值为3012。

输入描述:

首先，输入一个正整数T，表示接收T组测试用例；
此后，输入T组测试用例，其中每组测试用例包含如下内容：
输入1：一个正整数K，表示在当前测试用例中，可以对数组进行K次相邻元素间的位置交换；
输入2：一个正整数N，表示当前用例包含数组的长度；
输入3：N个数组元素，所有元素都是小于10的非负整数；

输出描述:

输出共N行，对应于N个用例的输出：
每行输出为一个数组，数组元素之间用一个空格隔开，要求每行输出前后均无多余空格。

输入例子1:

4
2
5
4 2 1 3 5
3
5
4 2 1 3 5
4
5
4 2 1 3 5
5
5
4 2 1 3 5

输出例子1:

4 3 2 1 5
4 5 2 1 3
5 4 2 1 3
5 4 2 3 1

P.S.题目来源于牛客网

题解：

从`p=A[0]`开始，我们寻找在中比`p`大的数`pm`，并将`pm`对应的值依次交换到`p`对应的值，并更新剩余的变换步数`K-=(pm-p)`,更新`p+=1`.直到`p`达到`A[-1]`或`K=0`。同时注意，题目要求是K最终为0，但有时`p`达到`A[-1]`时，K也没有等于0.这时判断K的奇偶，K为偶数时经过K/2次后两位元素的交换仍为原数，即剩余的K值为偶数时没有影响；而为奇数时，相当于偶数+1，则需要将最后两位进行交换。

 code:

class Solution:
    def __init__(self,A,K,N):
        self.A = A
        self.K = K
        self.N = N

    def swap(self,i,j):
        for k in range(j,i,-1):
            self.A[k-1],self.A[k] = self.A[k],self.A[k-1]

    def step(self,p):
        lower, upper = p + 1, min(p + 1 + self.K, len(self.A))
        pm, max = p, self.A[p]
        for i in range(lower, upper):
            if max < self.A[i]:
                pm, max = i, self.A[i]
        self.swap(p, pm)
        self.K -= (pm - p)

    def solve(self):
        p = 0
        while self.K>0 and p<self.N-1:
            solve.step(p)
            p += 1
        if self.K%2 == 1: self.swap(p-1,p)
        for j in range(self.N):
            print("{} ".format(self.A[j]),end="")
        print("")

if __name__ == '__main__':
    T = int(input())
    K = [0]*T
    N = [0]*T
    A = [[] for _ in range(T)]
    for i in range(T):
        K[i] = int(input())
        N[i] = int(input())
        A[i] = list(map(int,input().split()))
    for i in range(T):
        solve = Solution(A[i],K[i],N[i])
        solve.solve()