数论-因数 LeetCode-3266. K 次乘运算后的最终数组 II 2509分

黄昏月亮很忙

于 2024-10-10 09:43:08 发布

阅读量42

点赞数 1

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

本文链接：https://blog.csdn.net/2402_84970476/article/details/142813199

版权

LeetCode 专栏收录该内容

15 篇文章 0 订阅

订阅专栏

class Solution:
    def getFinalState(self, nums: List[int], k: int, multiplier: int) -> List[int]:
        if multiplier == 1:
            return nums
        
        from sortedcontainers import SortedList
        MOD = 10 ** 9 + 7
        sl = SortedList((v, i) for i, v in enumerate(nums))
        n = len(nums)
        times = [0] * n

        while k:
            v, i = sl[0]
            if v * multiplier > sl[-1][0]:
                break
            sl.pop(0)
            v *= multiplier
            times[i] += 1
            sl.add((v, i))
            k -= 1

        div, mod = divmod(k, n)
        for _, i in sl:
            times[i] += div
            if mod:
                times[i] += 1
                mod -= 1

        return [(v * pow(multiplier, t, MOD)) % MOD for v, t in zip(nums, times)]