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)