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;elses.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:# py3def canArrange(self, arr: List[int], k: int) -> bool:freq = [0]*kfor a in arr:freq[(a%k+k)%k] += 1if freq[0]&1:return Falsefor i in range(1, k//2+1):if freq[i] != freq[k-i]:return Falsereturn True
152 ms 24.9 MB