程序员面试金典 - 面试题 17.19. 消失的两个数字(数学/位运算)

1. 题目

给定一个数组,包含从 1 到 N 所有的整数,但其中缺了两个数字。

你能在 O(N) 时间内只用 O(1) 的空间找到它们吗?

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

示例 1:
输入: [1]
输出: [2,3]

示例 2:
输入: [2,3]
输出: [1,4]

提示:
nums.length <= 30000

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/missing-two-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 解题

2.1 数学公式

1 2 + 2 2 + 3 2 + 4 2 + . . . . + n 2 = n ∗ ( n + 1 ) ∗ ( 2 n + 1 ) / 6 1^2+2^2+3^2+4^2+....+n^2 = n*(n+1)*(2n+1)/6 12+22+32+42+....+n2=n(n+1)(2n+1)/6
1 + 2 + . . . + n = n ∗ ( 1 + n ) / 2 1+2+...+n = n*(1+n)/2 1+2+...+n=n(1+n)/2

class Solution {
public:
    vector<int> missingTwo(vector<int>& nums) {
        int n = nums.size()+2, a, b;
        long sum = 0, squareSum = 0;
        for(int i = 0; i < nums.size(); ++i)
        {
            sum += nums[i];
            squareSum += nums[i]*nums[i];
        }
        squareSum = long(n)*(n+1)*(2*n+1)/6 - squareSum;
        sum = n*(n+1)/2 - sum;
        for(a = 1; a <= n; ++a)
        {
            if(2*a*(sum-a) == sum*sum - squareSum)
                break;
        }
        b = sum-a;
        return {a,b};
    }
};

72 ms 22.5 MB

2.2 常规解法

class Solution {
public:
    vector<int> missingTwo(vector<int>& nums) {
        sort(nums.begin(), nums.end());
        vector<int> ans;
        int i, j;
        for(i = 1, j = 0; i <= nums.size()+2 && j < nums.size(); ++i)
        {
            if(nums[j] != i)
                ans.push_back(i);
            else
                j++;
            if(ans.size()==2)
                return ans;
        }
        if(ans.size()==1)
            ans.push_back(i);
        else if(ans.size()==0)
        {
            ans.push_back(i);
            ans.push_back(i+1);
        }
        return ans;
    }
};

116 ms 22.4 MB

2.3 位运算

  • 0 异或所有数,以及1-n的所有数,只出现1次的两个数的异或值得到了
  • 把上面异或值不为0的二进制位找到
  • 用这个位把 1到n 和数组里的数(2n-2 个),按照上面找出的二进制位,分成2组
  • 即得到出现1次的两个数
class Solution {
public:
    vector<int> missingTwo(vector<int>& nums) {
        int XOR = 0, a = 0, b = 0, i;
        for(auto& n : nums) XOR ^= n;
        for(i = 1; i <= nums.size()+2; ++i) XOR ^= i;
        for(i = 0; i < 32; ++i)
            if(XOR&(1<<i))//a、b二进制不同的位
                break;// i为 a,b 不相同的二进制位
        for(auto& n : nums)
        {
            if((n>>i)&1)//分组
                a ^= n;
            else
                b ^= n;
        }
        for(int j = 1; j <= nums.size()+2; ++j)
        {
            if((j>>i)&1)
                a ^= j;
            else
                b ^= j;
        }
        return {a,b};
    }
};

80 ms 22.3 MB

已标记关键词 清除标记
课程目标 从零开始掌握Premiere影视剪辑的基础知识,学会视频剪辑+影视特效+视频调色+字幕制作 适用人群 PR零基础小白,在校大学生,职场新人,想成为影视剪辑、影视后期、短视频制作、自媒体等高手的朋友。 课程简介 课程以新版PR 2020讲解,可使用PR任意版本学习,绝大多数功能兼容旧版,但强烈建议使用新版本。 【Adobe认证专家讲师精耕细作精品教程,非学院派照本宣科软件操作教程,以任务为导向,面向实际应用场景,每一章都能学会实打实的高手技能,讲解细致,小白也能轻松入门成大神!】 课程好不好,看过就知道,前面的免费章节欢迎试看。 本课程学习不需要任何PR基础,只需要电脑操作基础即可。兼容Windows和Mac操作系统,同时讲解两种系统下的快捷键操作,不用担心操作上的障碍问题。 课程特色 1、以实际PR影视编辑与特效的流程为导向,绝大多数内容都是为了完成某个具体任务,而不是为了讲解某个软件操作而凑数。 2、不同于国内多数教程和书籍,每个知识点务求讲精、讲透,帮助你掌握PR的精髓,而非软件操作上的皮毛,让你真正学到PR的本质,一次学习,终身受用,少走弯路,节约生命。 3、课程会随PR新版本的推出持续更新,不必担心有新功能却不知道怎么用。 4、充足的练习题和作业题,让你在不断的练习和挑战中提升PR技能。
相关推荐
©️2020 CSDN 皮肤主题: Age of Ai 设计师:meimeiellie 返回首页