补题:leetcode第382场周赛 3022. 给定操作次数内使剩余元素的或值最小

3022. 给定操作次数内使剩余元素的或值最小 - 力扣(LeetCode)
补题:leetcode第382场周赛 3022. 给定操作次数内使剩余元素的或值最小_第1张图片
拆位
n个数进行或运算的结果最小,每次操作可以对相邻的两个数进行与运算,至多进行k次操作
n个数进行或运算,可以对每个数进行拆解,拆解成最小的bit后再进行或运算
比如,2,4,3进行或运算
2:0 1 0
4:1 0 0
3:0 1 1
本来是010 | 100 | 011,拆解后:(0 | 1 | 0) + (1 | 0 | 1) + (0 | 1 | 1)
从高到低对每个数bitwei进行或运算

回到题目,要使最后的运算结果最小,就要从高到低尽可能地使每个bit位为0
从高到低的过程中,若确定了某一位的运算结果能为0,之后的考虑便要带上可能为0的这一位

思路就是这样,具体实现比较难,有些考验代码能力

class Solution {
public:
    int minOrAfterOperations(vector<int>& nums, int k) {
        int n = nums.size();
        vector<int> a(n);
        int ans = 0, mask = 0;
        for (int i = 31; i >= 0; -- i)
        {
            for (int j = 0; j < n; ++ j)
                a[j] = (nums[j] & mask) | (nums[j] & (1 << i));
            bool zero = false, flag = true;
            int cnt = 0;
            for (int j = 0; j < n; ++ j) 
            {
                int t = 0, cur = a[j];
                while (j < n && (cur &= a[j])) j ++ , t ++ ;
                if (j == n && t && zero == false) flag = false;
                else cnt += t;
                zero = true;
            }
            if (!flag || cnt > k) ans |= (1 << i);
            else mask |= (1 << i);
        }
        return ans;
    }
};

你可能感兴趣的:(练习赛补题,leetcode,算法,数据结构)