leetcode1497. 检查数组对是否可以被 k 整除
给你一个整数数组 arr
和一个整数 k
,其中数组长度是偶数,值为 n
。
现在需要把数组恰好分成 n / 2
对,以使每对数字的和都能够被 k
整除。
如果存在这样的分法,请返回 True ;否则,返回 False 。
示例 1:
输入:arr = [1,2,3,4,5,10,6,7,8,9], k = 5
输出:true
解释:划分后的数字对为 (1,9),(2,8),(3,7),(4,6) 以及 (5,10) 。
示例 2:
输入:arr = [1,2,3,4,5,6], k = 7
输出:true
解释:划分后的数字对为 (1,6),(2,5) 以及 (3,4) 。
示例 3:
输入:arr = [1,2,3,4,5,6], k = 10
输出:false
解释:无法在将数组中的数字分为三对的同时满足每对数字和能够被 10 整除的条件。
示例 4:
输入:arr = [-10,10], k = 2
输出:true
示例 5:
输入:arr = [-1,1,-2,2,-3,3,-4,4], k = 3
输出:true
提示:
arr.length == n
1 <= n <= 10^5
n
为偶数-10^9 <= arr[i] <= 10^9
1 <= k <= 10^5
方法:按照整除k余数不同分组
思路:
我们可知,两个数的和能被k整除,那么意味着两个数的和除以k的余数为0。也就意味着两个数分别除以k的余数之和为k或0。
我们使用一个数组residue,residue[i]表示arr中,除以k余数为i的数个数,i的取值为[0,k-1]。
只有两个数对应的余数和为k或0才可以配对,没有其他可能,所以我们只要考虑residue数组即可,分为两种情况,k为奇数或偶数。如下如所示:
代码:
class Solution:
def canArrange(self, arr: List[int], k: int) -> bool:
#数组residue表示除以k余数为i的数的个数
residue = [0 for _ in range(k)]
for num in arr:
temp = num % k
residue[temp] += 1
#对于k是偶数的情况
if k%2 == 0:
for i in range(k//2+1):
if i == 0:
if residue[i] %2 != 0:
return False
elif i == k//2:
if residue[i] %2 != 0:
return False
else:
if residue[i] != residue[k-i]:
return False
return True
#对于k是奇数的情况
else:
for i in range(k//2+1):
if i == 0:
if residue[i] %2 != 0:
return False
else:
if residue[i] != residue[k-i]:
return False
return True