- Python算法指南:程序员经典算法分析与实现
- 李永华
- 264字
- 2021-03-28 12:47:44
例71 数组中的k-diff对的数量
1. 问题描述
给定一个整数数组和一个整数k,这里k-diff对被定义为整数对(i,j),其中i和j都是数组中的数字,它们的绝对差是k,本例将找到数组中k-diff对的数量。
2. 问题示例
输入[3,1,4,1,5],k=2,输出2,数组中有两个2-diff对,(1,3)和(3,5)。虽然在输入中有两个1,但只返回唯一对的数量。
输入[1,2,3,4,5],k=1,输出4,数组中有四个1-diff对,(1,2),(2,3),(3,4)和(4,5)。
输入[1,3,1,5,4],k=0,输出1,数组中有一个0-diff对,(1,1)。
3. 代码实现
4. 运行结果
输入:[6,3,4,2,5,1] 2
输出:4