小马智行Pony.ai算法实习生笔试题3——求相邻数字经K次交换得到的最大值

版权声明:本文为博主原创文章,遵循 CC 4.0 by-sa 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/shichunxue/article/details/89447050

题目描述:数组中存储了一堆小于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个用例的输出;

每行输出为一个数组,数组元素之间用一个空格隔开,要求每行输出前后均无多余空格。

示例:

#输入
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

#输出
4 3 2 1 5
4 5 2 1 3
5 4 2 1 3
5 4 2 3 1

备注:

60%的数据K<=10的3次方;

70%的数据K<=10的4次方;

80%的数据K<=10的5次方;

90%的数据K<=10的6次方;

100%的数据K<=10的9次方;

100%的数据T<=10的3次方;60%的数据N<=10的3次方;100%的数据N<=5乘上10的3次方。

编程思路:

要将数值较大的数字放到高位上去!如果可以交换K次,那么可以移动的最远的距离是K+1,所以每次执行交换之前,要先从起始位置开始的后面K+1位中去寻找最大值,从而把最大值放到当前的最高位,如果交换次数没有用完,那么继续去确定下一个高位。

步骤:

我们先设置一个起始位B为数组的第一位,循环K次:

   对于每一组测试用例,首先确定判断交换次数K与数组长度N之间的关系:

            K+1<=N,则判断当前起始位B是否是接下来K+1位中的最大值:

                         是最大值,起始位B后移,continue;

                         不是最大值,找到最大值的索引,计算最大值索引与起始位索引的差值C,将最大值经过C次交换换到当前的起始位,剩余的交换次数K更新为K-C;

            K+1>N,则判断当前起始位B是否是接下来N-B位中的最大值:

                         是最大值,起始位B后移,continue;

                         不是最大值,找到最大值的索引,计算最大值索引与起始位索引的差值C,将最大值经过C次交换换到当前的起始位,剩余的交换次数K更新为K-C;

代码如下:

# -*- coding:utf-8 -*-
# @Author:scx
# @time:2019/4/22
# @File:hhh.py
 
##是否是最大值
def isbig(i, array, K):
	# print i
	# print array
	# print K
	if (K+1)<len(array):
		if array[i] == max(array[0:K+1]):
			return True
		else:
			return False
	else:
		if array[i] == max(array[0:]):
			return True
		else:
			return False
 
##从最大值向前面逐个交换
def new_trans_k(array,index_max,k):
    for i in range(k):
        t=array[index_max]
        array[index_max]=array[index_max-1]
        array[index_max - 1]=t
        index_max-=1
        # print i,":",array
    return array

 
#数组转换
def trans(array, K):
    for i in range(K):
		# print 'i=',i
		# print 'K=',K
		if (K+1)<len(array):
			if isbig(i, array, K):
				continue
			else:
				max_num = max(array[i:i+K+1])
				index_max = array.index(max_num)
				c = index_max - i
				K = K - c
				array = new_trans_k(array, index_max, c)
		else:
			if i<len(array):
				if isbig(i, array, K):
					continue
				else:
					max_num = max(array[i:])
					index_max = array.index(max_num)
					c = index_max - i
					K = K - c
					array = new_trans_k(array, index_max, c)
			else:
				break

    return array
 
 
T = int(raw_input())  ##T次测试用例
string_list = []  ##输出列表
 
for i in range(0, T):
    K = int(raw_input())  ###K次变换
    N =  int(raw_input())  ###输入列表array的长度
    array = []
    array = map(int, raw_input().strip().split(" "))
    array = trans(array, K)
    num_list_string = " ".join(map(str,array))
    print num_list_string

参考了博客https://blog.csdn.net/Homewm/article/details/89445391的解题思路,在他的基础上考虑了数组长度与交换次数的关系,对数组长度小于交换次数的情况进行了处理。

展开阅读全文

没有更多推荐了,返回首页