Leetcode 面试题 17.19. 消失的两个数字【位运算】

最新推荐文章于 2023-06-29 21:30:20 发布

MD_

最新推荐文章于 2023-06-29 21:30:20 发布

阅读量388

点赞数

分类专栏： leetcode 位运算

本文链接：https://blog.csdn.net/qq_27690765/article/details/105870356

版权

leetcode 同时被 2 个专栏收录

189 篇文章 1 订阅

订阅专栏

位运算

5 篇文章 0 订阅

订阅专栏

文章目录

问题描述
解题报告
实现代码
参考资料

问题描述

给定一个数组，包含从 1 到 N 所有的整数，但其中缺了两个数字。你能在 O(N) 时间内只用 O(1) 的空间找到它们吗？

以任意顺序返回这两个数字均可。

解题报告

在原数组的基础上加上序列 $1,2,\cdots,N$ ，那么问题就变成了Leetcode 面试题56-I. 数组中数字出现的次数。解法参考Leetcode 面试题56-I. 数组中数字出现的次数【根据二进制位进行分组异或】。

实现代码

class Solution {
public:
    vector<int> missingTwo(vector<int>& nums) {
        int ret = 0;
        for (int n : nums)
            ret ^= n;
        for(int i=1;i<=nums.size()+2;i++){
            ret^=i;
        }
        int div = ret&(-ret);
        int a = 0, b = 0;
        for (int n : nums)
            if (div & n)
                a ^= n;
            else
                b ^= n;
                
        for(int i=1;i<=nums.size()+2;i++){
            if (div & i)
                a ^= i;
            else
                b ^= i;
        }
        return vector<int>{a, b};
    }
};