Leetcode 1775. Equal Sum Arrays With Minimum Number of Operations 数学方法,ceil向上取整

  • 刚开始想到堆的方法,见后面
  • 然后发现还有count方法,可以实现O(n)复杂度,这种题目我认为需要有数学的sense,要善于在其中找出规律.这里直接count每个元素出现的次数,可以加快计算
  • (int)ceil(A*1.0/B) 可以起到向上取整的作用
  • min操作很巧妙,一方面如果p2数量足够多,那么就是找能够刚好减完delta的数,如果p2数量不够,那么返回的就是cnt2[p2],因为cnt2[p2] < (int)ceil(1.0*delta/(6-p2))这个向上取整的数

O(n) count

class Solution {
public:
    int minOperations(vector<int>& nums1, vector<int>& nums2) {

        int n1=nums1.size(), n2=nums2.size();
        if(n1*6<n2*1 || n1*1>n2*6) return -1;
        int delta = 0, sum1=0, sum2=0;
        vector<int> cnt1(7,0);
        vector<int> cnt2(7,0);
        for(auto e: nums1){
            sum1+=e;
            cnt1[e]++;
        }
        for(auto e: nums2) {
            sum2+=e;
            cnt2[e]++;
        }
        delta = sum1-sum2;
        if(delta<0) {
            delta=-delta;
            swap(nums1, nums2);
            swap(cnt1, cnt2);
        }

        int p1=6, p2=1;
        int cnt=0;
        while (delta>0 && (p1>1 || p2<6)){
            if(p1<1) {
                cnt+=min(cnt2[p2], (int)ceil(1.0*delta/(6-p2))); //min操作很巧妙,一方面如果p2数量足够多,那么就是找能够刚好减完delta的数,如果p2数量不够,那么返回的就是cnt2[p2],因为cnt2[p2] < (int)ceil(1.0*delta/(6-p2))这个向上取整的数
                delta-=(6-p2)*cnt2[p2];
                p2++;
            }
            else if(p2>6){
                cnt+=min(cnt1[p1], (int)ceil(1.0*delta/(p1-1)));
                delta-=(p1-1)*cnt1[p1];
                p1--;
            }
            else if((p1-1) > (6-p2)){
                cnt+=min(cnt1[p1], (int)ceil(1.0*delta/(p1-1)));
                delta-=(p1-1)*cnt1[p1];
                p1--;
            }
            else{
                cnt+=min(cnt2[p2], (int)ceil(1.0*delta/(6-p2)));
                delta-=(6-p2)*cnt2[p2];
                p2++;
            }
        }
        return delta<=0?cnt:-1;

    }
    void swap(vector<int>& a, vector<int>& b){
        vector<int> t;
        t = a;
        a = b;
        b = t;
    }

};

O(nlogn) 堆的方法

class Solution {
public:
    int minOperations(vector<int>& nums1, vector<int>& nums2) {
        int delta = 0, sum1=0, sum2=0;
        for(auto e: nums1) sum1+=e;
        for(auto e: nums2) sum2+=e;
        delta = sum1-sum2;
        if(delta<0) {
            delta=-delta;
            swap(nums1, nums2);
        }
        
        priority_queue<int> q1;
        priority_queue<int, vector<int>, greater<int>> q2;
        for(auto e: nums1) q1.push(e);
        for(auto e: nums2) q2.push(e);
        int cnt=0;
        while (delta>0 && (!q1.empty() || !q2.empty())){
            if(q1.empty()) {
                delta-=6-q2.top();
                cnt++;
                q2.pop();
            }
            else if(q2.empty()){
                delta-=q1.top()-1;
                cnt++;
                q1.pop();
            }
            else if((q1.top()-1) > (6-q2.top())){
                delta-=q1.top()-1;
                cnt++;
                q1.pop();
            }
            else{
                delta-=6-q2.top();
                cnt++;
                q2.pop();
            }
        }
        return delta<=0?cnt:-1;

    }
    void swap(vector<int>& a, vector<int>& b){
        vector<int> t;
        t = a;
        a = b;
        b = t;
    }

};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值