题目:
数组中存储了一堆小于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
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()