LeetCode 1497. 检查数组对是否可以被 k 整除（余数配对）

1. 题目

给你一个整数数组 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/check-if-array-pairs-are-divisible-by-k
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

• 对所有的数求余，如有负数再 +k，再求余
• multiset 不断的删除 一个数 a，和 k-a

class Solution {	//C++
public:
    bool canArrange(vector<int>& arr, int k) {
        multiset<int> s;
        for(auto a : arr)
            s.insert((a%k+k)%k);
        while(!s.empty())
        {
            int a = *s.begin();
            s.erase(s.begin());
            auto it = s.find((k-a)%k);
            if(it == s.end())
                return false;
            else
                s.erase(it);
        }
        return true;
    }
};

804 ms 102.1 MB

• 或者 用数组计数加速

class Solution {	//C++
public:
    bool canArrange(vector<int>& arr, int k) {
        vector<int> freq(k,0);
        for(auto a : arr)
            freq[(a%k+k)%k]++;
        if(freq[0]&1)//直接能被整除的是奇数个，不行
            return false;
        for(int i = 1; i <= k/2; ++i)
            if(freq[i] != freq[k-i])
                return false;
        return true;
    }
};

268 ms 61.8 MB

class Solution:# py3
    def canArrange(self, arr: List[int], k: int) -> bool:
        freq = [0]*k
        for a in arr:
            freq[(a%k+k)%k] += 1
        if freq[0]&1:
            return False
        for i in range(1, k//2+1):
            if freq[i] != freq[k-i]:
                return False
        return True

152 ms 24.9 MB