leetcode 1497.检查数组对是否可以被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 整除的条件。

提示:

  • arr.length == n
  • 1 <= n <= 105
  • n 为偶数
  • -109 <= arr[i] <= 109
  • 1 <= k <= 105

解题思路:对数组arr第一个元素开始找它的配对元素使其之和能被k整除,找到之后将这两个数组下标记录下来,接下里找第二个元素,当然在找之前要判断数组下标是否已经被记录了。

class Solution {
public:
    bool canArrange(vector<int>& arr, int k) {
        vector<int> s;
         int n = arr.size();
        //  vector<int> s(n,1);      
         for (int i = 0; i < n; ++i)
         {
             if(count(s.begin(),s.end(),i)) continue;
            //  if(s[i] == 0) continue;
            for(int j = i+1; j < n; ++j)
                  {
                      if(count(s.begin(),s.end(),j)) continue;
                    // if(s[j] == 0) continue;
                    if((arr[i] + arr[j])%k == 0)
                           {
                            //    s[i] = 0;
                            //    s[j] = 0;
                               s.push_back(j);
                               s.push_back(i);
                           }
                  }
         }
         //if(count(s.begin(),s.end(),1)) return false;
         if(s.size() == n) return true;
         else return false;
    }
};

发现是有问题的,对于负数无法处理,但是不知道为什么,这里留着疑问。

观察高赞答案:

两个数 x 和 y 的和能被 kkk 整除,当且仅当这两个数对 k 取模的结果 x_kx 和 y_k 的和就能被 k 整除。这里我们规定取模的结果大于等于 0,无论 x 和 y 的正负性。因此,我们将数组 arr 中的每个数 x 对 k 进行取模,并将得到的余数 x_k 进行配对


 

class Solution {
public:
    bool canArrange(vector<int>& arr, int k) {
        vector<int> mod(k);
        for (int num: arr) {
            ++mod[(num % k + k) % k];
        }
        for (int i = 1; i + i < k; ++i) {
            if (mod[i] != mod[k - i]) {
                return false;
            }
        }
        return mod[0] % 2 == 0;
    }
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值