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

题目描述：

解题思路：

要想得到输入的数字列中存在相隔为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)