题目:
给你一个下标从 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;
}
}