Leetcode.2197 替换数组中的非互质数

最新推荐文章于 2024-09-14 20:30:05 发布

感觉画质不如…原神

最新推荐文章于 2024-09-14 20:30:05 发布

阅读量62

点赞数 1

分类专栏： Leetcode 文章标签：算法 leetcode 栈

本文链接：https://blog.csdn.net/m0_74396439/article/details/130221505

版权

Leetcode 专栏收录该内容

441 篇文章 2 订阅

订阅专栏

题目链接

Leetcode.2197 替换数组中的非互质数 Rating ： 2057

题目描述

给你一个整数数组 nums。请你对数组执行下述操作：

从 nums中找出任意两个相邻的 非互质 数。
如果不存在这样的数，终止这一过程。
否则，删除这两个数，并替换为它们的最小公倍数（Least Common Multiple，LCM）。
只要还能找出两个相邻的非互质数就继续重复这一过程。

返回修改后得到的最终数组。可以证明的是，以任意顺序替换相邻的非互质数都可以得到相同的结果。

生成的测试用例可以保证最终数组中的值小于或者等于 $10^8$ 。

两个数字 x 和 y 满足非互质数的条件是： $GC D (x, y) > 1$ ，其中 $GC D (x, y)$ 是 x 和 y 的 最大公约数 。

示例 1 ：

输入：nums = [6,4,3,2,7,6,2]
输出：[12,7,6]
解释：

(6, 4) 是一组非互质数，且 LCM(6, 4) = 12 。得到 nums = [12,3,2,7,6,2] 。
(12, 3) 是一组非互质数，且 LCM(12, 3) = 12 。得到 nums = [12,2,7,6,2] 。
(12, 2) 是一组非互质数，且 LCM(12, 2) = 12 。得到 nums = [12,7,6,2] 。
(6, 2) 是一组非互质数，且 LCM(6, 2) = 6 。得到 nums = [12,7,6] 。现在，nums 中不存在相邻的非互质数。因此，修改后得到的最终数组是 [12,7,6] 。注意，存在其他方法可以获得相同的最终数组。

示例 2 :

输入：nums = [2,2,1,1,3,3,3]
输出：[2,1,1,3]
解释：

(3, 3) 是一组非互质数，且 LCM(3, 3) = 3 。得到 nums = [2,2,1,1,3,3] 。
(3, 3) 是一组非互质数，且 LCM(3, 3) = 3 。得到 nums = [2,2,1,1,3] 。
(2, 2) 是一组非互质数，且 LCM(2, 2) = 2 。得到 nums = [2,1,1,3] 。现在，nums 中不存在相邻的非互质数。因此，修改后得到的最终数组是 [2,1,1,3] 。注意，存在其他方法可以获得相同的最终数组。

提示:

$1 <= nums.length <= 10^5$
$1 <= nums[i] <= 10^5$
生成的测试用例可以保证最终数组中的值小于或者等于 $10^8$ 。

解法：栈模拟

对于 $n u m s$ 中的元素 $n u m s [i]$ ，我们每次都先将其入栈 $s t a c k$ 。

在 $\geq 2$ 的前提下，取栈顶的两个元素 $a$ 和 $b$ ， $g$ 是 $a 和 b$ 的最大公约数，即 $g = g c d (a, b)$ 。

如果 $g = 1$ ，说明 $a 和 b$ 互质，就不进行操作；
否则将 $c = l c m (a, b) = a * b / g$ （ $a 和 b$ 的最小公倍数）代替 $a 和 b$ 入栈。

循环结束，栈中的元素就是我们的答案。

时间复杂度： $O (n l o g C)$

C++代码：

class Solution {
public:
    vector<int> replaceNonCoprimes(vector<int>& nums) {
        vector<int> ans;
        for(auto x:nums){
            ans.push_back(x);
            while(ans.size() > 1){
                int b = ans.back();
                int a = ans[ans.size() - 2];

                int g = gcd(a , b);
                if(g == 1) break;
                ans.pop_back();
                ans.back() *= b / g;
            }
        }

        return ans;
    }
};