原始问题:
小红操作数组n-1次后剩余数为0至9的方案数统计
这个问题可以使用动态规划来解决。我们可以定义一个二维数组dp,其中dpi表示在进行i次操作后,数组中最后一个数为j的方案数。
首先,我们初始化dp数组为0。然后,我们可以根据题目中的两种操作来更新dp数组。
对于第一种操作,我们可以将数组的最后两个数x和y的个位数放回数组的最后。因此,我们可以得到状态转移方程:
dpi += dpi-1
dpi += dpi-1
对于第二种操作,我们可以将数组的最后两个数x和y的乘积的个位数放回数组的最后。因此,我们可以得到状态转移方程:
dpi += dpi-1
最后,我们可以根据题目中的操作次数n-1来计算dp数组。最终的答案就是dpn-1到dpn-1的和。
下面是使用Python代码实现上述思路:
def countNumbers(nums, n):
dp = [[0] * 10 for _ in range(n)]
dp[0][nums[0]] = 1
for i in range(1, n):
for j in range(10):
for x in range(10):
for y in range(10):
if (x + y) % 10 == j:
dp[i][j] += dp[i-1][x] * dp[i-1][y]
if (x * y) % 10 == j:
dp[i][j] += dp[i-1][x] * dp[i-1][y]
ans = sum(dp[n-1])
return ans
nums = [1, 2, 3, 4]
n = len(nums)
result = countNumbers(nums, n)
print(result)这样,我们就可以得到数组中最后一个数等于0, 1, ..., 9的方案数分别为多少。
Prev:小红棋盘选四棋子构成正方形的方案数计算