初学python记录:力扣2007. 从双倍数组中还原原数组

题目:

一个整数数组 original 可以转变成一个 双倍 数组 changed ,转变方式为将 original 中每个元素 值乘以 2 加入数组中,然后将所有元素 随机打乱

给你一个数组 changed ,如果 change双倍 数组,那么请你返回 original数组,否则请返回空数组。original 的元素可以以 任意 顺序返回。

提示:

  • 1 <= changed.length <= 10^5
  • 0 <= changed[i] <= 10^5

思考:

数组original中每个元素乘以2加入数组中,得到数组changed ---->

  1. 数组changed一定有偶数个元素

  2. 数组changed的元素从小到大遍历,设当前元素为x,那么 x*2 一定另外存在于数组changed中,每次判断完后将当前这两个元素从changed中删去,继续向下判断。

代码如下:

python 复制代码
class Solution(object):
    def findOriginalArray(self, changed):
        """
        :type changed: List[int]
        :rtype: List[int]
        """
        original = []
        n = len(changed)
        i = 0
        if n % 2 == 1:
            return[]

        changed.sort()
        while i < n:
            x = changed[i]
            original.append(x)
            changed.remove(x)
            i -= 1
            n -= 1
            if x * 2 not in changed:
                return []
            else:
                changed.remove(x * 2)
                n -= 1
            i += 1
        return original

提交后卡在例子 176 / 179 超时了:

可以看到这一句判断语句每判断一次都要遍历一次数组,大大增加了耗时:

python 复制代码
if x * 2 not in changed:
                return []

那么直接在最开始统计changed中每个元素的出现次数即可,代码如下:

python 复制代码
from collections import Counter

class Solution(object):
    def findOriginalArray(self, changed):
        """
        :type changed: List[int]
        :rtype: List[int]
        """
        original = []
        n = len(changed)
        i = 0
        if n % 2 == 1:
            return[]

        changed.sort()
        count = Counter(changed)
        for x in changed:
            if count[x] >= 1:
                y = x * 2
                original.append(x)
                count[x] -= 1
                if count[y] < 1:
                    return []
                else:
                    count[y] -= 1
        return original

提交通过:

相关推荐
沈浩(种子思维作者)2 分钟前
真的能精准医疗吗?癌症能提前发现吗?
人工智能·python·网络安全·健康医疗·量子计算
MM_MS12 分钟前
Halcon变量控制类型、数据类型转换、字符串格式化、元组操作
开发语言·人工智能·深度学习·算法·目标检测·计算机视觉·视觉检测
独自破碎E37 分钟前
【二分法】寻找峰值
算法
njsgcs1 小时前
ue python二次开发启动教程+ 导入fbx到指定文件夹
开发语言·python·unreal engine·ue
io_T_T1 小时前
迭代器 iteration、iter 与 多线程 concurrent 交叉实践(详细)
python
mit6.8241 小时前
位运算|拆分贪心
算法
华研前沿标杆游学1 小时前
2026年走进洛阳格力工厂参观游学
python
Carl_奕然1 小时前
【数据挖掘】数据挖掘必会技能之:A/B测试
人工智能·python·数据挖掘·数据分析
ghie90901 小时前
基于MATLAB的TLBO算法优化实现与改进
开发语言·算法·matlab
恋爱绝缘体11 小时前
2020重学C++重构你的C++知识体系
java·开发语言·c++·算法·junit