LeetCode-day30-2961. 双模幂运算

题目描述

给你一个下标从 0 开始的二维数组 variables ，其中 variables[i] = [a_i, b_i, c_i, m_i]，以及一个整数 target 。

如果满足以下公式，则下标 i 是 好下标：

• 0 <= i < variables.length
• ((a_i^bi % 10)^ci) % m_i == target

返回一个由 好下标 组成的数组，顺序不限 。

示例

示例1：

输入：variables = [[2,3,3,10],[3,3,3,1],[6,1,1,4]], target = 2
输出：[0,2]
解释：对于 variables 数组中的每个下标 i ：
对于下标 0 ，variables[0] = [2,3,3,10] ，(2³ % 10)³ % 10 = 2 。
对于下标 1 ，variables[1] = [3,3,3,1] ，(3³ % 10)³ % 1 = 0 。
对于下标 2 ，variables[2] = [6,1,1,4] ，(6¹ % 10)¹ % 4 = 2 。
因此，返回 [0,2] 作为答案。

示例2：

输入：variables = [[39,3,1000,1000]], target = 17
输出：[ ]
解释：对于 variables 数组中的每个下标 i ：
对于下标 0 ，variables[0] = [39,3,1000,1000] ，(39³ % 10)¹⁰⁰⁰ % 1000 = 1 。
因此，返回 [ ] 作为答案。

思路

幂次方，暴力循环

代码

class Solution:
    def getGoodIndices(self, variables: List[List[int]], target: int) -> List[int]:
        res = []
        for i,x in enumerate(variables):
            ans =((x[0]**x[1] % 10) ** x[2]) % x[3]
            if ans == target:
                res.append(i)
        return res