2018春招-今日头条笔试题-第一题（python）

题目描述：2018春招-今日头条笔试题5题（后附大佬答案-c++版）

解题思路：

要想得到输入的数字列中存在相隔为k的数，可以将输入的数字加上k，然后判断其在不在输入的数字列中即可。

#-*- coding:utf-8 -*-
class Fun: def __init__(self,string,k): self.string=string self.k=int(k) self._set=set() def fun(self): l=set() #用于存储，已经比较的数，防止重复比较
        for i in self.string: if l not in l:  #只比较未比较的
                if str(int(i)+self.k) in self.string: self._set.add(i + str(int(i)+self.k)) #将符合的数字对以字符串的形式存入集合
                l.add(i)    #已经的比较的放入集合
        return len(self._set)  #返回集合的长度
 n_k=raw_input().split(' ') k=n_k[1] string=raw_input().split(' ') print Fun(string,k).fun()

大佬的做法：将输入的数字数列去重，全部加k，生成一个新的list后，比较这里两个list的相同数的个数

#-*- coding:utf-8 -*-
def diff_list(lst, k): #将原始字符串转化为int类型加k，生成新的list
    newlst = [str(int(i) + int(k)) for i in lst] #将list转为集合（去重）,利用&操作得到两个set集合的相同数的集合
    print len(set(lst) & set(newlst)) n_k=raw_input().split(' ') n=n_k[0] k=n_k[1] string=raw_input().split(' ') diff_list(string,k)