nums, and a positive integer k.
You are allowed to perform an operation once on nums, where in each operation you can remove any non-overlapping prefix and suffix from nums such that nums remains non-empty.
You need to find the x-value of nums, which is the number of ways to perform this operation so that the product of the remaining elements leaves a remainder of x when divided by k.
Return an array result of size k where result[x] is the x-value of nums for 0 <= x <= k - 1.
A prefix of an array is a subarray that starts from the beginning of the array and extends to any point within it.
A suffix of an array is a subarray that starts at any point within the array and extends to the end of the array.
Note that the prefix and suffix to be chosen for the operation can be empty.
Example 1:
Output: [9,2,4]
Explanation:
x = 0, the possible operations include all possible ways to remove non-overlapping prefix/suffix that do not remove nums[2] == 3.x = 1, the possible operations are:
[2, 3, 4, 5]. nums becomes [1].[1, 2, 3] and the suffix [5]. nums becomes [4].x = 2, the possible operations are:
[3, 4, 5]. nums becomes [1, 2].[1] and the suffix [3, 4, 5]. nums becomes [2].[1, 2, 3] and the empty suffix. nums becomes [4, 5].[1, 2, 3, 4] and the empty suffix. nums becomes [5].Example 2:
Output: [18,1,2,0]
Explanation:
x = 0, the only operations that do not result in x = 0 are:
<ul>
<li>Remove the empty prefix and the suffix <code>[4, 8, 16, 32]</code>. <code>nums</code> becomes <code>[1, 2]</code>.</li>
<li>Remove the empty prefix and the suffix <code>[2, 4, 8, 16, 32]</code>. <code>nums</code> becomes <code>[1]</code>.</li>
<li>Remove the prefix <code>[1]</code> and the suffix <code>[4, 8, 16, 32]</code>. <code>nums</code> becomes <code>[2]</code>.</li>
</ul>
</li>
<li>For <code>x = 1</code>, the only possible operation is:
<ul>
<li>Remove the empty prefix and the suffix <code>[2, 4, 8, 16, 32]</code>. <code>nums</code> becomes <code>[1]</code>.</li>
</ul>
</li>
<li>For <code>x = 2</code>, the possible operations are:
<ul>
<li>Remove the empty prefix and the suffix <code>[4, 8, 16, 32]</code>. <code>nums</code> becomes <code>[1, 2]</code>.</li>
<li>Remove the prefix <code>[1]</code> and the suffix <code>[4, 8, 16, 32]</code>. <code>nums</code> becomes <code>[2]</code>.</li>
</ul>
</li>
<li>For <code>x = 3</code>, there is no possible way to perform the operation.</li>
Example 3:
Output: [9,6]
Constraints:
1 <= nums[i] <= 1091 <= nums.length <= 1051 <= k <= 5배열, 수학, 동적 계획법
AI 풀이 설명 생성 중 오류가 발생했습니다. 나중에 다시 시도해주세요.
class Solution:
def resultArray(self, nums: list[int], k: int) -> list[int]:
answer = [0] * k
dp = [0] * k
for x in nums:
v = x % k
n = [0] * k
n[v] += 1
for i in range(k):
if dp[i]:
n[(i * v) % k] += dp[i]
dp = n
for i in range(k):
answer[i] += dp[i]
return answer