leetcode 5499 检查数组对是否可以被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) 。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/check-if-array-pairs-are-divisible-by-k

分析

所有余数都将在0到k之间。而k-i和i这两个数的个数，决定了这一对是否能k整除。两者相等则配对求和不会有剩下的。 最后mod[0]则是所有为k的整数倍的数个数，如果他们不为偶数，则同样意味着配对时候，有一个会被剩下来，与其他0到k-1的数据配对，这样是不满足题目的。

解法

func canArrange(arr []int, k int) bool {
    mod := make([]int, k)
    for _, num := range arr {
        mod[(num%k+k)%k]++
    }

    for i:= 1; i < k/2+1; i++ {
        if mod[i] != mod[k-i] {
            return false
        }
    }

    return mod[0] % 2 == 0
}