网易2020笔试真题 序列交换

题目描述:
小易给你一个包含n个数字的数组a1,a2,….an。你可以对这个数组执行任意以下交换操作:
对于数组中的两个下标i,j(1<=I,j<=n),如果ai+aj为奇数,就可以交换ai和aj
现在允许你使用操作次数不限,小易希望你能求出在所有通过若干次操作可以得到的数组中,字典序最小的一个是什么
输入描述:
第一行一个整数n,
第二行n个整数a1,a2,an,表示数组,每两个数字之间用一个空格分隔。
输出描述:
n个整数,每两个整数之间用一个空格分隔,表示得到的字典序最小的数组

思路:有奇偶数的时候,都是可以很快排序,只是需要考虑全偶数和全奇数的情况。题目很简单,只是当时完全没有想到。

import sys
if __name__ == "__main__":
    n = int(sys.stdin.readline().strip())    
    line = sys.stdin.readline().strip()    
    values = map(int, line.split())    
    result = list()    
    for v in values:        
        result.append(v)    
    temp1 = [x for x in result if x % 2 == 0]    
    temp2 = [x for x in result if x %2 != 0]    
    if temp1 != [] and temp2  != []:        
        result.sort()    
    result = [str(x) for x in result]    
    print((' '.join(result)))

你可能感兴趣的:(算法工程师)