奇安信21届秋招算法岗笔试题 亲7数的个数

奇安信21秋招算法岗笔试题

题目描述

如果一个正整数可以被7整除，我们称之为亲7数，对于给出的一组个位数字，请找出使用所有的数字排列的数中的亲7数的个数。
其中给出的个位数字数组中每一个都是不相关的，即使有重复的数字，如{1,1,2}排列出的整数位{112,121,112,121,211,211}，亲7数为{112,112}共2个。

输入：个位数字数组，数组有m个元素{x1,x2,x3…xm}（0<=xi<=9）
输出：亲7的个数

测试样例

输入
[1,1,2]

输出
2

解题思路

方法1：

利用 itertools 迭代器包，可以很方便的实现这个功能，对于给定的个位数字数组会自动生成一个排列好的集合迭代器，输出的结果中每个元素是他们组合完成的元组。

res_list = []
for one in  itertools.permutations([1,1,2]):
    res_list.append(one)
res_list
>> [(1, 1, 2), (1, 2, 1), (1, 1, 2), (1, 2, 1), (2, 1, 1), (2, 1, 1)]

我们可以用这个包很轻松的实现个位数字的排列，先定义一个函数辅助函数helper，用来生成个位数字数组的排列，并放入列表输出，通过itertools迭代所有排列，然后遍历生成的迭代对象，把其中单个元组中每个元素组合起来，再添加进空列表 alist 并返回。
在主函数中，调用辅助函数 helper； 然后遍历所有生成个位数字数组的排列列表，如果其中的元素能整除7就让count += 1，最后输出count即可。

import itertools

class Solution:
    def helper(self , num_list):        
        tmp_list = itertools.permutations(num_list)
        res_list=[]
        alist = []
        for one in  tmp_list:
            res_list.append(one)
        for i in res_list:
            x = ''.join(tuple(map(str,i)))
            alist.append(x)
        return alist
    def reletive_7(self , digit):
        # write code here
        count = 0
        alist = self.helper(digit)
        for i in range(len(alist)):
            if int(alist[i])%7 == 0:
                count += 1
        return count
    
s = Solution()
s.reletive_7([1,1,2])
>> 2

方法2

def permutation(digit):
    s= []
    for i in digit:
        s.append(str(i))
        c, res = list(s), []
    def dfs(x):
        if x == len(c) - 1:
            res.append(''.join(c)) # 添加排列方案
            return
        dic = []
        for i in range(x, len(c)):
            #if c[i] in dic: continue # 重复，因此剪枝
            dic.append(c[i])
            c[i], c[x] = c[x], c[i] # 交换，将 c[i] 固定在第 x 位
            dfs(x + 1) # 开启固定第 x + 1 位字符
            c[i], c[x] = c[x], c[i] # 恢复交换
    dfs(0)
    result = []
    for i in res:
        if int(i) % 7 == 0:
            result.append((int(i)))

    return len(result)
permutation([1,1,2])
>> 2