阅读量:0
题目:
给你一个下标从 0 开始的二维数组 variables ,其中 variables[i] = [ai, bi, ci, mi],以及一个整数 target 。
如果满足以下公式,则下标 i 是 好下标:
返回一个由 好下标 组成的数组,顺序不限 。
解题
- 初始化:创建一个空的 ArrayList 来存储满足条件的索引。
- 遍历数组:使用一个 for 循环遍历 variables 数组。
- 提取元素:在每次循环中,从当前的 variables[i] 中提取 a、b、c 和 m。
- 计算 val1:计算 abmod 10abmod10 的值。这里使用一个内嵌的 for 循环来连续乘以 a,每次乘法后都取模 10,以保持结果在个位数。
- 模运算:将 val1 模 m,得到最终的 val1 值。
- 计算 val2:使用另一个内嵌的 for 循环计算 (val1)cmod m(val1)cmodm 的值。
- 条件判断:检查 val2 是否等于给定的 target。
- 添加索引:如果条件成立,将当前索引 i 添加到结果列表中。
- 返回结果:循环结束后,返回包含所有满足条件索引的列表。
代码
class Solution { public List<Integer> getGoodIndices(int[][] variables, int target) { List<Integer> ans = new ArrayList<>(); int n = variables.length; for (int i = 0; i < n; i++) { int a = variables[i][0] % 10; int b = variables[i][1]; int c = variables[i][2]; int m = variables[i][3]; // 使用快速幂算法计算val1 long val1 = fastPowerMod(a, b, 10); val1 %= m; // 再次使用快速幂算法计算val2 long val2 = fastPowerMod((int)val1, c, m); if (val2 == target) { ans.add(i); } } return ans; } // 快速幂算法,计算base^exp % mod private long fastPowerMod(int base, int exp, int mod) { long result = 1; base %= mod; // 保证base在模m范围内 while (exp > 0) { if ((exp & 1) != 0) { result = (result * base) % mod; } base = (base * base) % mod; exp >>= 1; } return result; } }