2021校招 | 交通银行-IT类编程题

绝地灬酷狼 2022-12-11 07:49 148阅读 0赞

86题：给定N个随机正整数，将其中为素数的整数输出。

# -*- coding: utf-8 -*-
    # 86题：给定N个随机正整数，将其中为素数的整数输出。
    # 例如：
    # 输入：[3,5,11,12]
    # 输出：[3,5,11]
    # 注意：1、输出数组剩余元素先后顺序需要与原数组保持一致，否则不得分
    #      2、给出数组中不存在重复元素，则无需去重
    class Solution:
        def find_prime(self, nums):
            """
            找到给出正整数中的素数
            :param nums: 给定的正整数列表
            :return: 给定数组只含素数的列表
            """
            flag = self.need_dup(nums)
            if flag:  # 需要去重
                list_1 = []
                for i in nums:
                    if i not in list_1:
                        list_1.append(i)
                result = []
                for i in list_1:
                    if self.is_prime(i):
                        result.append(i)
                return result
            else:
                result = []
                for i in nums:
                    if self.is_prime(i):
                        result.append(i)
                return result
    
        def is_prime(self, x):
            """
            判断该数是不是素数【如果因数超过2个，则不是素数】
            :param x: 待判定的数
            :return: 真假
            """
            count = 0  # 该数的因子
            for i in range(1, x + 1):
                if x % i == 0:
                    count += 1
            if count > 2:
                return False
            else:
                return True
    
        def need_dup(self, arr):
            """
            判断是否需要去重【如果有重复，则去重后的长度和去重前的长度不相等】
            :param arr: 列表
            :return: True/False是否需要去重
            """
            len_1 = len(arr)
            len_2 = len(list(set(arr)))
            if len_1 == len_2:
                return False
            else:
                return True
    
    
    if __name__ == '__main__':
        input_str = input()
        x = input_str.replace('[', '')#处理左方括号
        x = x.replace(']', '')#处理右方括号
        x = x.split(',')#按照逗号分隔
        x = [int(i) for i in x]  # 将字符串转换为整型
        s = Solution()
        output = s.find_prime(x)
        print(output)

![20200929122213590.png][]

![20200929122250302.png][]

也不知道做对了没有，脑阔痛！

[20200929122213590.png]: /images/20221123/c4b3c4e9c3fa473bbc585fe9d20b3cf0.png
[20200929122250302.png]: /images/20221123/1ed77a8b407a4259a488c3f1af4512e4.png