【leetcode】3524 求出数组的X值1

题目链接

题目描述

给你一个正整数数组 nums 和一个正整数 k。

你可以对数组执行一次操作：移除不重叠的前缀和后缀（可以为空），留下一个连续非空子数组。
对于每一种留下的子数组，计算：

(该子数组的乘积) % k = x

你需要统计所有可能的结果 x ∈ [0, k-1] 出现的次数，返回一个长度为 k 的数组 result，其中：

result[x] 表示得到 x 作为余数的方案数

解题思路

你可以移除任意前缀和后缀，留下的是一个非空的连续子数组。
也就是说：我们其实是统计所有子数组的乘积模 k 的结果分布。
直接采用动态规划进行转移

class Solution {public long[] resultArray(int[] nums, int k) {long[] ans = new long[k];long[] lurminexod = new long[k];for (int i : nums) {long[] newlurminexod = new long[k];for (int j = 0; j < k; j++) {int index = (int)(((long)j * i) % k);newlurminexod[index] += lurminexod[j];}newlurminexod[i % k]++;lurminexod = newlurminexod;for (int j = 0; j < k; j++) {ans[j] += lurminexod[j];}}return ans;}
}